app.py

import cv2
from telegram.ext.callbackcontext import CallbackContext
from telegram.ext.commandhandler import CommandHandler
from telegram.ext.filters import Filters
from telegram.ext.messagehandler import MessageHandler
from telegram.ext.updater import Updater
from telegram.update import Update

from merch_generator import MerchGenerator, ImgProcessor, TextProcessor


class TelegramBot:
    def __init__(self, bot_url: str,):
        self.updater = Updater(bot_url, use_context=True)
        self.updater.dispatcher.add_handler(CommandHandler('start', self.start))
        self.updater.dispatcher.add_handler(CommandHandler('help', self.help))
        self.updater.dispatcher.add_handler(MessageHandler(Filters.text, self.text_message_handler))
        self.updater.dispatcher.add_handler(MessageHandler(Filters.command, self.unknown))
        self.merch_generator = MerchGenerator(img_processor=ImgProcessor(),
                                              text_processor=TextProcessor())

    def start(self, update: Update, context: CallbackContext):
        update.message.reply_text(
            "Привет! Это телеграм-бот `@merch_generator_bot`, генерирующий мерч "
            "(а именно футболку) с рисунком по текстовому описанию"
        )
        update.message.reply_text("Для генерация футболки с рисунком:\n"
                                  "1) Введите текстовое описание футболки, указав\n"
                                  "   - Название бренда, логотип которого вы хотите видеть на футболке\n"
                                  "   - Основной цвет футболки\n"
                                  "  - Для какого мероприятия предназначена футболка\n"
                                  "2) Дождитесь, пока бот сгенерирует изображение:)")


    def help(self, update: Update, context: CallbackContext):
        update.message.reply_text("Для генерация футболки с рисунком:\n"
                                  "1) Введите текстовое описание футболки, указав\n"
                                  "   - Название бренда, логотип которого вы хотите видеть на футболке\n"
                                  "   - Основной цвет футболки\n"
                                  "  - Для какого мероприятия предназначена футболка\n"                                
                                  "2) Дождитесь, пока бот сгенерирует изображение:)")

    def unknown(self, update: Update, context: CallbackContext):
        with open('results/output.png', 'rb') as file:
            update.message.reply_photo(file)
        update.message.reply_text(
            "Sorry '%s' is not a valid command" % update.message.text)

    def text_message_handler(self, update: Update, context: CallbackContext):
        # (update.message.text)
        img = self.merch_generator.generate(update.message.text)
        cv2.imwrite("results/full_bot_result.png", img)
        with open('results/full_bot_result.png', 'rb') as file:
            update.message.reply_photo(file)

    def run(self):
        self.updater.start_polling()