FROM python:latest


  #  Установка необходимых зависимостей
  RUN pip install telebot
  RUN pip install git+https://github.com/Uberi/speech_recognition.git
  RUN pip install pydub
  RUN pip install requests

  #  Обновление системы и установка необходимых пакетов
  RUN apt update
  RUN apt upgrade -y
  RUN apt install -y ffmpeg


  # Копировать файл скрипта из текущего каталога, внутрь контейнера в корень
  COPY audio_to_text.py .

  # Точкой входа в контейнер, установим запуск этого скрипта 
  ENTRYPOINT ["python", "audio_to_text.py"]

  # и укажем опцию перезапуска контейнера внутри образа, в случае остановки
  CMD ["--restart=always"]

  

import os
import telebot
import speech_recognition
from pydub import AudioSegment
import requests

# при запуске контейнера вставить свой токен, который дал BotFather в Telegram при регистрации нашего бота
#Поэтому при создании контейнера нужно передавать в параметрах контейнеру Docker, используя переменные среды.
#docker run -e TOKEN='ваш токен вашего бота полученного у @botfather в телеграмм' my_image

#здесь мы берём эту переменную из среды и помещаем в локальную переменную скрипта
token = os.environ['TOKEN']

bot = telebot.TeleBot(token)

welcome_message = 'Привет, если ты отправишь сюда голосовое, я переведу его в текст и выведу в этот чат'

# Add a handler for the /start command
@bot.message_handler(commands=['start'])
def send_welcome(message):
    # Send the welcome message along with the menu options
    bot.reply_to(message, welcome_message)


def oga2wav(filename):
    # Конвертация формата файлов
    new_filename = filename.replace('.oga', '.wav')
    audio = AudioSegment.from_file(filename)
    audio.export(new_filename, format='wav')
    return new_filename


def recognize_speech(oga_filename):
    # Перевод голоса в текст + удаление использованных файлов
    wav_filename = oga2wav(oga_filename)
    recognizer = speech_recognition.Recognizer()

    with speech_recognition.WavFile(wav_filename) as source:
        wav_audio = recognizer.record(source)

    text = recognizer.recognize_google(wav_audio, language='ru')

    if os.path.exists(oga_filename):
        os.remove(oga_filename)

    if os.path.exists(wav_filename):
        os.remove(wav_filename)

    return text


def download_file(bot, file_id):
    # Скачивание файла, который прислал пользователь
    file_info = bot.get_file(file_id)
    downloaded_file = bot.download_file(file_info.file_path)
    filename = file_id + file_info.file_path
    filename = filename.replace('/', '_')
    with open(filename, 'wb') as f:
        f.write(downloaded_file)
    return filename

@bot.message_handler(content_types=['voice'])
def transcript(message):
    # Функция, отправляющая текст в ответ на голосовое
    filename = download_file(bot, message.voice.file_id)
    text = recognize_speech(filename)
    bot.send_message(message.chat.id, text)


# Запускаем бота.
bot.polling()