diff options
Diffstat (limited to 'callbacks.py')
| -rw-r--r-- | callbacks.py | 66 |
1 files changed, 49 insertions, 17 deletions
diff --git a/callbacks.py b/callbacks.py index c411d75..432c0af 100644 --- a/callbacks.py +++ b/callbacks.py @@ -26,7 +26,10 @@ async def select_provider_callback(callback: CallbackQuery): buttons = [] + i = 0 + for model in MODELS.get(provider): + i += 1 if f"{provider}*{model}" == user_model: text = f"{model} ✅" else: @@ -34,14 +37,21 @@ async def select_provider_callback(callback: CallbackQuery): if MODELS[provider][model].get("alias"): model = MODELS[provider][model].get("alias") - - buttons.append( - [ + + if not i % 2: + buttons[len(buttons)-1] +=[ InlineKeyboardButton( text=text, callback_data=f"set_model%{user_id}%{provider}*{model}" ) ] - ) + else: + buttons.append( + [ + InlineKeyboardButton( + text=text, callback_data=f"set_model%{user_id}%{provider}*{model}" + ) + ] + ) if not buttons: buttons = [ @@ -53,7 +63,7 @@ async def select_provider_callback(callback: CallbackQuery): ] buttons.append( - [InlineKeyboardButton(text="назад", callback_data=f"models%{user_id}")] + [InlineKeyboardButton(text="⬅️ назад", callback_data=f"models%{user_id}")] ) await callback.message.edit_text( @@ -72,16 +82,27 @@ async def models_callback(callback: CallbackQuery): config = get_user_config(user_id) buttons = [] + + i = 0 for provider in MODELS: - buttons.append( - [ + i += 1 + if not i % 2: + buttons[len(buttons)-1] +=[ InlineKeyboardButton( - text=PROVIDERS.get(provider).get("name"), - callback_data=f"prov%{user_id}%{provider}", + text=PROVIDERS.get(provider).get("name"), + callback_data=f"prov%{user_id}%{provider}" ) ] - ) + else: + buttons.append( + [ + InlineKeyboardButton( + text=PROVIDERS.get(provider).get("name"), + callback_data=f"prov%{user_id}%{provider}", + ) + ] + ) if not buttons: buttons.append( @@ -93,7 +114,7 @@ async def models_callback(callback: CallbackQuery): ) buttons.append( - [InlineKeyboardButton(text="назад", callback_data=f"settings%{user_id}")] + [InlineKeyboardButton(text="⬅️ назад", callback_data=f"settings%{user_id}")] ) await callback.message.edit_text( @@ -147,19 +168,30 @@ async def prompts_callback(callback: CallbackQuery): prompt = config.get("prompt") buttons = [] + i = 0 for name in SYSTEM_PROMPTS: + i += 1 text = name if prompt == name: text += " ✅" - buttons.append( - [ + + if not i % 2: + buttons[len(buttons)-1] +=[ InlineKeyboardButton( - text=text, - callback_data=f"set_prompt%{user_id}%{name}", + text=text, + callback_data=f"set_prompt%{user_id}%{name}" ) ] - ) + else: + buttons.append( + [ + InlineKeyboardButton( + text=text, + callback_data=f"set_prompt%{user_id}%{name}", + ) + ] + ) if not buttons: buttons.append( @@ -171,7 +203,7 @@ async def prompts_callback(callback: CallbackQuery): ) buttons.append( - [InlineKeyboardButton(text="назад", callback_data=f"settings%{user_id}")] + [InlineKeyboardButton(text="⬅️ назад", callback_data=f"settings%{user_id}")] ) await callback.message.edit_text( |