Create keyboards_gemini.py

chatbot/plugins/keyboards_gemini.py

@@ -0,0 +1,33 @@
+from pyrogram import Client
+from pyrogram.types import *
+from database import users_collection
+
+def get_gemini_keyboard():
+    keyboard = []
+    keyboard.append([
+        InlineKeyboardButton("Gemini Flash EXP", callback_data="model_gemini-2.0-flash-exp"),
+        InlineKeyboardButton("Gemini EXP", callback_data="model_gemini-exp-1206")
+    ])
+    keyboard.append([
+        InlineKeyboardButton("Gemini Flash 8B", callback_data="model_gemini-1.5-flash-8b")
+    ])
+    keyboard.append([
+        InlineKeyboardButton("Gemini PRO", callback_data="model_gemini-1.5-pro"),
+        InlineKeyboardButton("Gemini Flash", callback_data="model_gemini-1.5-flash"),
+    ])
+    return InlineKeyboardMarkup(keyboard)
+
+@app.on_callback_query(filters.regex(r"^model_"))
+async def set_model(client: Client, callback_query: CallbackQuery):
+    model = callback_query.data.split("_")[1]
+    user_id = callback_query.from_user.id
+    await users_collection.update_one(
+        {"user_id": user_id},
+        {"$set": {"model": model}},
+        upsert=True
+    )
+    await callback_query.answer(f"Model set to {model}")
+    await callback_query.message.edit_text(
+        f"Your Model has been set to **{model}**.",
+        reply_markup=None
+    )