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 |
-
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|