Upload chat.py

chatbot/plugins/chat.py

@@ -199,85 +199,8 @@ Type anything to begin.
 
 gen = genai.Client(api_key=GOOGLE_API_KEY)
 
-
-async def progress(current, total, message, start, type_of_ps, file_name=None):
-    """Progress Bar For Showing Progress While Uploading / Downloading File - Normal"""
-    now = time.time()
-    diff = now - start
-    buttons = [
-        [
-            InlineKeyboardButton(
-                text="Cancel",
-                callback_data="cancels"
-            )
-        ],
-    ]
-    if round(diff % 10.00) == 0 or current == total:
-        percentage = current * 100 / total
-        speed = current / diff
-        elapsed_time = round(diff) * 1000
-        if elapsed_time == 0:
-            return
-        time_to_completion = round((total - current) / speed) * 1000
-        estimated_total_time = elapsed_time + time_to_completion
-        progress_str = f"{''.join(['▰' for i in range(math.floor(percentage / 10))])}"
-        progress_str += (
-            f"{''.join(['▱' for i in range(10 - math.floor(percentage / 10))])}"
-            )
-        progress_str += f"{round(percentage, 2)}%\n"
-        tmp = f"{progress_str}{humanbytes(current)} of {humanbytes(total)}\n"
-        tmp += f"ETA: {time_formatter(estimated_total_time)}"
-        if file_name:
-            try:
-                await message.edit(
-                    f"{type_of_ps}\n**File Name:** `{file_name}`\n{tmp}"
-                    )
-            except FloodWait as e:
-                await asyncio.sleep(e.x)
-            except MessageNotModified:
-                pass
-        else:
-            try:
-                await message.edit_text(
-                    f"{type_of_ps}\n{tmp}",
-                    parse_mode=enums.ParseMode.MARKDOWN,
-                    reply_markup=InlineKeyboardMarkup(buttons)
-                )
-            except FloodWait as e:
-                await asyncio.sleep(e.x)
-            except MessageNotModified:
-                pass
-
-def humanbytes(size):
-    """Convert Bytes To Bytes So That Human Can Read It"""
-    if not size:
-        return ""
-    power = 2**10
-    raised_to_pow = 0
-    dict_power_n = {0: "", 1: "Ki", 2: "Mi", 3: "Gi", 4: "Ti"}
-    while size > power:
-        size /= power
-        raised_to_pow += 1
-    return str(round(size, 2)) + " " + dict_power_n[raised_to_pow] + "B"
-
-def time_formatter(milliseconds: int) -> str:
-    """Time Formatter"""
-    seconds, milliseconds = divmod(int(milliseconds), 1000)
-    minutes, seconds = divmod(seconds, 60)
-    hours, minutes = divmod(minutes, 60)
-    days, hours = divmod(hours, 24)
-    tmp = (
-        ((str(days) + " day(s), ") if days else "")
-        + ((str(hours) + " hour(s), ") if hours else "")
-        + ((str(minutes) + " minute(s), ") if minutes else "")
-        + ((str(seconds) + " second(s), ") if seconds else "")
-        + ((str(milliseconds) + " millisecond(s), ") if milliseconds else "")
-    )
-    return tmp[:-2]
-
 # Original: @GeminiAIDev_Bot
 # Code by @xtdevs
-
 def replace_with_markdown_list(text, pattern=r"^\s*\*\s*(.*)$"):
     def replace(match):
         text = match.group(1)
@@ -304,14 +227,56 @@ def split_message(text: str) -> List[str]:
             text = ""
     return parts
 
-@Client.on_callback_query(filters.regex("^cancels"))
-async def cancel_(client, callback_query):
-    chat_id = callback_query.message.chat.id
-    if await task_manager.cancel_task(chat_id):
-        spam_chats.remove(chat_id)
-        await callback_query.edit_message_text("✅ Processing canceled.")
-    else:
-        await callback_query.edit_message_text("⚠️ No active task to cancel.")
+@Client.on_callback_query(filters.regex("^closedd$"))
+async def closeed(client, callback):
+    await callback.message.delete()
+
+@Client.on_callback_query(filters.regex("^modelmenu$"))
+async def modelgeminimenu(client, callback):
+    keyboard.append([
+        InlineKeyboardButton("Gemini Flash EXP", callback_data="modelset:gemini-2.0-flash-exp"),
+        InlineKeyboardButton("Gemini EXP", callback_data="modelset:gemini-exp-1206")
+    ])
+    keyboard.append([
+        InlineKeyboardButton("Gemini Flash 8B", callback_data="modelset:gemini-1.5-flash-8b"),
+        InlineKeyboardButton("Gemini 2.0 Flash ", callback_data="modelset:gemini-2.0-flash-001")
+    ])
+    keyboard.append([
+        InlineKeyboardButton("Gemini PRO", callback_data="modelset:gemini-1.5-pro"),
+        InlineKeyboardButton("Gemini Flash", callback_data="modelset:gemini-1.5-flash"),
+    ])
+    keyboard.append([
+        InlineKeyboardButton("Close", callback_data="close")
+    ])
+    await callback.edit_message_reply_markup(reply_markup=InlineKeyboardMarkup(keyboard))
+
+@Client.on_callback_query(filters.regex(r"^modelset:(.*)$"))
+async def gemini_setmodel(client, callback):
+    model = callback.matches[0].group(1)
+    user_id = callback.from_user.id
+    await users_collection.update_one(
+        {"user_id": user_id},
+        {"$set": {"model": model}},
+        upsert=True
+    )
+    await callback.answer(f"Model set to {model}")
+    keyboard = []
+    keyboard.append([
+        InlineKeyboardButton("Gemini Flash EXP", callback_data="modelset:gemini-2.0-flash-exp"),
+        InlineKeyboardButton("Gemini EXP", callback_data="modelset:gemini-exp-1206")
+    ])
+    keyboard.append([
+        InlineKeyboardButton("Gemini Flash 8B", callback_data="modelset:gemini-1.5-flash-8b"),
+        InlineKeyboardButton("Gemini 2.0 Flash ", callback_data="modelset:gemini-2.0-flash-001")
+    ])
+    keyboard.append([
+        InlineKeyboardButton("Gemini PRO", callback_data="modelset:gemini-1.5-pro"),
+        InlineKeyboardButton("Gemini Flash", callback_data="modelset:gemini-1.5-flash"),
+    ])
+    keyboard.append([
+        InlineKeyboardButton("Close", callback_data="close")
+    ])
+    await callback.edit_message_reply_markup(reply_markup=InlineKeyboardMarkup(keyboard))
 
 @Client.on_callback_query(filters.regex("^deletemydatagm$"))
 async def deletemydata(client, callback):
@@ -550,11 +515,6 @@ async def remover_bg(client, callback):
             reply_markup=None
         )
 
-
-@Client.on_callback_query(filters.regex("^closedd$"))
-async def closeed(client, callback):
-    await callback.message.delete()
-
 @Client.on_callback_query(filters.regex("^memory_(\d+)_([a-f0-9]{8})$"))
 async def memory_updated(client, callback):
     user_id, uuidstr = callback.matches[0].groups()
@@ -734,6 +694,7 @@ def create_keyboard(
                 [
                     InlineKeyboardButton(f"♾️", callback_data=f"trmulti_{user_id}_{uuid_str}"),
                     InlineKeyboardButton(f"🔒", callback_data=f"showchat_{user_id}"),
+                    InlineKeyboardButton(f"⛏️", callback_data=f"modelmenu"),
                     InlineKeyboardButton(f"💾", callback_data=f"memory_{user_id}_{uuid_str}")
                 ],
                 [
@@ -1540,4 +1501,4 @@ async def chatbot_talk(client: Client, message: Message):
 
         except Exception as e:
             LOGS.error(f"Error: message.text: {str(e)}")
-            return await message.reply_text("Server busy try again later")
+            return await message.reply_text("Server busy try again later")