Files
TeleDiffusionBot/bot/handlers/txt2img/set_model.py
2023-05-03 21:35:23 +03:00

20 lines
858 B
Python

from aiogram import types
from bot.db import db, DBTables, decrypt
from bot.utils.cooldown import throttle
from bot.keyboards.set_model import get_set_model_keyboard
from bot.modules.api.models import get_models
from bot.utils.errorable_command import wrap_exception
@wrap_exception()
@throttle(cooldown=30*60, admin_ids=db[DBTables.config].get('admins'), by_id=False)
async def set_model_command(message: types.Message):
models = await get_models()
if models is not None and len(models) > 0:
db[DBTables.config]['models'] = models
else:
await message.reply('❌ No models available')
return
await message.reply("Examples of models (with additional info): https://telegra.ph/Opisanie-raboty-modelej-05-03")
await message.reply("🪄 You can choose model from available:", reply_markup=get_set_model_keyboard(0))