from aiogram import types
from bot.db import db, DBTables
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=5*60, admin_ids=db[DBTables.config].get('admins', []), by_id=False)
async def set_model_command(message: types.Message):
    if db[DBTables.config].get('whitelist') and \
            (message.chat.id not in db[DBTables.config]['whitelist'] and
             message.from_id not in db[DBTables.config]['whitelist']):
        await message.reply('❌ You are not on the white list, access denied. Contact admin @kilisauros for details')
        return
    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("🪄 You can choose model from available:", reply_markup=get_set_model_keyboard(0))