main.py

import logging
import os

from telegram import Audio, Document, Message, Update, Video, VideoNote, Voice
from telegram.constants import ChatAction, ParseMode
from telegram.ext import ApplicationBuilder, CommandHandler, ContextTypes, MessageHandler

from filter_allowed_chats import FilterAllowedChats
from message_transcriber import AudioMessageTranscriber

LOG_LEVEL = logging.DEBUG if os.environ.get('LOG_LEVEL', 'INFO') == 'DEBUG' else logging.INFO

allowed_chat_ids = os.environ.get('ALLOWED_CHAT_IDS', default='').split(',')

logging.basicConfig(level=LOG_LEVEL,
                    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')

logger = logging.getLogger(__name__)


async def transcribe(audio, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    logger.info('Transcribing Audio message')
    await context.bot.send_chat_action(chat_id=update.effective_chat.id, action=ChatAction.TYPING)
    text, processing_time = await AudioMessageTranscriber.transcribe(context, audio)
    markdown_text = AudioMessageTranscriber.to_markdown(text, processing_time)
    await update.message.reply_text(markdown_text, parse_mode=ParseMode.MARKDOWN)


def _get_audio_from_message(message: Message) -> Voice | Audio | Video | VideoNote | Document | None:
    try:
        return (
            message.voice
            or message.audio
            or message.video
            or message.video_note
            or None
        )
    except AttributeError:
        return None


async def prepare_to_transcribe(update, context, message):
    audio = _get_audio_from_message(message)
    if audio is None:
        logger.info('Message is not a video, not an audio or not a voice.')
        return
    await transcribe(audio, update, context)


async def transcribe_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    await prepare_to_transcribe(update, context, update.message.reply_to_message)


async def transcribe_message(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    await prepare_to_transcribe(update, context, update.message)


def main():
    bot_token = os.environ.get('BOT_TOKEN')
    app = ApplicationBuilder().token(bot_token).build()
    # Filters
    filter_allowed_chats = FilterAllowedChats(allowed_chat_ids)
    # Handlers
    app.add_handler(CommandHandler('transcribe', transcribe_command, filter_allowed_chats))
    app.add_handler(MessageHandler(filter_allowed_chats, transcribe_message))
    app.run_polling()


if __name__ == '__main__':
    main()