randydev commited on
Commit
6bc40c3
·
verified ·
1 Parent(s): 08fc023
Files changed (1) hide show
  1. chatbot/plugins/keyboards_gemini.py +0 -41
chatbot/plugins/keyboards_gemini.py DELETED
@@ -1,41 +0,0 @@
1
- from pyrogram import Client, filters
2
- from pyrogram.types import *
3
- from database import users_collection
4
-
5
- def get_gemini_keyboard():
6
- keyboard = []
7
- keyboard.append([
8
- InlineKeyboardButton("Gemini Flash EXP", callback_data="models_gemini-2.0-flash-exp"),
9
- InlineKeyboardButton("Gemini EXP", callback_data="models_gemini-exp-1206")
10
- ])
11
- keyboard.append([
12
- InlineKeyboardButton("Gemini Flash 8B", callback_data="models_gemini-1.5-flash-8b"),
13
- InlineKeyboardButton("Gemini 2.0 Flash ", callback_data="models_gemini-2.0-flash-001")
14
- ])
15
- keyboard.append([
16
- InlineKeyboardButton("Gemini PRO", callback_data="models_gemini-1.5-pro"),
17
- InlineKeyboardButton("Gemini Flash", callback_data="models_gemini-1.5-flash"),
18
- ])
19
- return InlineKeyboardMarkup(keyboard)
20
-
21
- @Client.on_callback_query(filters.regex(r"^models_(\w+)$"))
22
- async def xset_modelok(client: Client, callback_query: CallbackQuery):
23
- model = callback.matches[0].group(1)
24
- user_id = callback_query.from_user.id
25
- await users_collection.update_one(
26
- {"user_id": user_id},
27
- {"$set": {"model": model}},
28
- upsert=True
29
- )
30
- await callback_query.answer(f"Model set to {model}")
31
- await callback_query.message.edit_text(
32
- f"Your Model has been set to **{model}**.",
33
- reply_markup=None
34
- )
35
-
36
- @Client.on_message(filters.command("setmodel"))
37
- async def xsetmodel_cmd(client: Client, message: Message):
38
- await message.reply(
39
- "Please select your preferred Gemini Model:",
40
- reply_markup=get_gemini_keyboard()
41
- )