Update chatbot/plugins/keyboards_gemini.py

chatbot/plugins/keyboards_gemini.py

@@ -18,9 +18,9 @@ def get_gemini_keyboard():
     ])
     return InlineKeyboardMarkup(keyboard)
 
-@Client.on_callback_query(filters.regex(r"^models_$"))
-async def set_modelok(client: Client, callback_query: CallbackQuery):
-    model = callback_query.data.split("_")[1]
+@Client.on_callback_query(filters.regex(r"^models_(\w+)$"))
+async def xset_modelok(client: Client, callback_query: CallbackQuery):
+    model = callback.matches[0].group(1)
     user_id = callback_query.from_user.id
     await users_collection.update_one(
         {"user_id": user_id},
@@ -34,7 +34,7 @@ async def set_modelok(client: Client, callback_query: CallbackQuery):
     )
 
 @Client.on_message(filters.command("setmodel"))
-async def setmodel_cmd(client: Client, message: Message):
+async def xsetmodel_cmd(client: Client, message: Message):
     await message.reply(
         "Please select your preferred Gemini Model:",
         reply_markup=get_gemini_keyboard()