Spaces:
Running
Running
| from flask import Flask, render_template, request | |
| from io import BytesIO | |
| from client import client | |
| import telebot | |
| import logging | |
| import os | |
| import warnings | |
| from client import client | |
| from io import BytesIO | |
| import pydub | |
| warnings.simplefilter('ignore') | |
| TOKEN = os.environ['TOKEN'] | |
| if not TOKEN: | |
| print('You must set the TOKEN environment variable') | |
| exit(1) | |
| START_MSG = '''Вітання! | |
| Цей бот створений для тестування перекладу українських аудіозаписів в текст. | |
| Група для обговорення: https://t.me/speech_recognition_uk''' | |
| FIRST_STEP = '''Використовувати бота просто: надішліть аудіоповідомлення і чекайте відповіді''' | |
| bot = telebot.TeleBot(TOKEN, parse_mode=None, threaded=False) | |
| app = Flask(__name__,) | |
| def index(): | |
| bot.remove_webhook() | |
| bot.set_webhook( | |
| url='https://voice-recognition-ua.herokuapp.com/' + TOKEN) | |
| return render_template('hello.html') | |
| def recognize(): | |
| file = request.files['file'] | |
| lang = request.form["lang"] | |
| audio = BytesIO() | |
| file.save(audio) | |
| audio.seek(0) | |
| result = client(audio, lang) | |
| return result | |
| def getMessage(): | |
| bot.process_new_updates( | |
| [telebot.types.Update.de_json(request.stream.read().decode("utf-8"))]) | |
| return "!", 200 | |
| def send_welcome(message): | |
| bot.reply_to(message, START_MSG) | |
| bot.reply_to(message, FIRST_STEP) | |
| def process_voice_message(message): | |
| # download the recording | |
| file_info = bot.get_file(message.voice.file_id) | |
| downloaded_file = bot.download_file(file_info.file_path) | |
| # create in-memory representation of files | |
| source_audio = BytesIO() | |
| source_audio.write(downloaded_file) | |
| source_audio.seek(0) | |
| output_audio = BytesIO() | |
| ogg_file = pydub.AudioSegment.from_ogg( | |
| source_audio) | |
| # convert ogg to wav | |
| ogg_file.set_frame_rate(16000).set_channels( | |
| 1).export(output_audio, "wav", codec="pcm_s16le") | |
| output_audio.seek(0) | |
| # do the recognition | |
| # get the recognized text | |
| try: | |
| text = client(output_audio) | |
| # no results | |
| if not text: | |
| bot.reply_to(message, 'Я не зміг розпізнати 😢') | |
| else: | |
| # send the recognized text | |
| bot.reply_to(message, text) | |
| except Exception as e: | |
| logging.log(logging.ERROR, str(e)) | |
| bot.reply_to(message, 'Трапилась помилка 😢') | |
| if __name__ == '__main__': | |
| app.run(host='0.0.0.0') | |