from bot.Handler import Handler from telegram.constants import ParseMode from telegram.ext import MessageHandler, filters, CallbackQueryHandler from dotenv import dotenv_values from ai_bot.states.KlingStates import KLING_SET_PROMPT, KLING_SET_TEXT_PROMPT config = dotenv_values(".env") class KlingHandler(Handler): @staticmethod def set_prompt(update, context): @staticmethod def get_states(data=None) -> dict: return { KLING_SET_TEXT_PROMPT: [ MessageHandler(filters.TEXT, KlingHandler.set_prompt) ], }