tgbots/TeleDiffusionBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed changing models. Changed cooldown for changing models to 30m

Browse Source
This commit is contained in:
seel2304
2023-05-03 17:09:25 +00:00
parent 2dda780d3f
commit 75f6404e94
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bot/handlers/txt2img/set_model.py
View File

@@ -1,5 +1,5 @@
from aiogram import types from aiogram import types
from bot.db import db, DBTables from bot.db import db, DBTables, decrypt
from bot.utils.cooldown import throttle from bot.utils.cooldown import throttle
from bot.keyboards.set_model import get_set_model_keyboard from bot.keyboards.set_model import get_set_model_keyboard
from bot.modules.api.models import get_models from bot.modules.api.models import get_models
@@ -7,7 +7,7 @@ from bot.utils.errorable_command import wrap_exception
@wrap_exception() @wrap_exception()
@throttle(cooldown=60*60, admin_ids=db[DBTables.config].get('admins'), by_id=False) @throttle(cooldown=30*60, admin_ids=db[DBTables.config].get('admins'), by_id=False)
async def set_model_command(message: types.Message): async def set_model_command(message: types.Message):
models = await get_models() models = await get_models()
if models is not None and len(models) > 0: if models is not None and len(models) > 0:

6
bot/modules/api/models.py
View File

@@ -1,14 +1,16 @@
import aiohttp import aiohttp
import json
from bot.db import db, DBTables, decrypt from bot.db import db, DBTables, decrypt
async def get_models() -> list | None: async def get_models() -> list | None:
endpoint = decrypt(db[DBTables.config].get('endpoint')) endpoint = decrypt(db[DBTables.config].get('endpoint'))
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
r = await session.get(endpoint + "/sdapi/v1/sd-models") r = await session.get(endpoint + '/sdapi/v1/sd-models')
if r.status != 200: if r.status != 200:
return None return None
return [x["title"] for x in await r.json()] json_data = await r.json()
return [x["title"] for x in json_data]
async def set_model(model_name: str): async def set_model(model_name: str):