24 lines
583 B
Python
24 lines
583 B
Python
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)
|
|
],
|
|
|
|
} |