Upload chat.py

chatbot/plugins/chat.py

@@ -236,19 +236,19 @@ async def modelgeminimenu(client, callback):
     uuidstr = callback.matches[0].group(1)
     keyboard = []
     keyboard.append([
-        InlineKeyboardButton("Gemini Flash EXP", callback_data=f"us:gemini-2.0-flash-exp:{uuidstr}"),
-        InlineKeyboardButton("Gemini EXP", callback_data=f"us:gemini-exp-1206:{uuidstr}")
+        InlineKeyboardButton("G Flash EXP", callback_data=f"us:gemini-2.0-flash-exp:{uuidstr}"),
+        InlineKeyboardButton("G EXP 1206", callback_data=f"us:gemini-exp-1206:{uuidstr}")
     ])
     keyboard.append([
-        InlineKeyboardButton("Gemini Flash 8B", callback_data=f"us:gemini-1.5-flash-8b:{uuidstr}"),
-        InlineKeyboardButton("Gemini 2.0 Flash ", callback_data=f"us:gemini-2.0-flash-001:{uuidstr}")
+        InlineKeyboardButton("G 1.5 Flash 8B", callback_data=f"us:gemini-1.5-flash-8b:{uuidstr}"),
+        InlineKeyboardButton("G 2.0 Flash 001", callback_data=f"us:gemini-2.0-flash-001:{uuidstr}")
     ])
     keyboard.append([
-        InlineKeyboardButton("Gemini PRO", callback_data=f"us:gemini-1.5-pro:{uuidstr}"),
-        InlineKeyboardButton("Gemini Flash", callback_data=f"us:gemini-1.5-flash:{uuidstr}"),
+        InlineKeyboardButton("G 1.5 PRO", callback_data=f"us:gemini-1.5-pro:{uuidstr}"),
+        InlineKeyboardButton("G 1.5 Flash", callback_data=f"us:gemini-1.5-flash:{uuidstr}"),
     ])
     keyboard.append([
-        InlineKeyboardButton("Close", callback_data="closedd")
+        InlineKeyboardButton("Back To Menu", callback_data=f"menu_back:{uuidstr}")
     ])
     await callback.edit_message_reply_markup(reply_markup=InlineKeyboardMarkup(keyboard))
 
@@ -282,9 +282,9 @@ async def deletemydata(client, callback):
     else:
         await callback.answer(delm, True)
 
-@Client.on_callback_query(filters.regex("^genprompt_(\d+)$"))
+@Client.on_callback_query(filters.regex("^genprompt$"))
 async def geminigen_prompt(client, callback):
-    user_id = int(callback.matches[0].group(1))
+    user_id = callback.from_user.id
     user_bl = await db.user_blacklists.find_one({"user_id": user_id})
     if user_bl and user_bl.get("is_frozen", False) and user_id not in [6477856957]:
         await callback.answer("Sorry your account is frozen", True)
@@ -364,9 +364,9 @@ async def geminigen_prompt(client, callback):
             except:
                 pass
 
-@Client.on_callback_query(filters.regex("^fluxprompt_(\d+)$"))
+@Client.on_callback_query(filters.regex("^fluxprompt$"))
 async def flux_prompt(client, callback):
-    user_id = int(callback.matches[0].group(1))
+    user_id = callback.from_user.id
     user_bl = await db.user_blacklists.find_one({"user_id": user_id})
     if user_bl and user_bl.get("is_frozen", False) and user_id not in [6477856957]:
         await callback.answer("Sorry your account is frozen", True)
@@ -468,16 +468,15 @@ async def flux_prompt(client, callback):
             except:
                 pass
 
-@Client.on_callback_query(filters.regex("^removerbg_(\d+)$"))
+@Client.on_callback_query(filters.regex("^removerbg$"))
 async def remover_bg(client, callback):
-    user_id = int(callback.matches[0].group(1))
+    user_id = callback.from_user.id
     user_bl = await db.user_blacklists.find_one({"user_id": user_id})
     if user_bl and user_bl.get("is_frozen", False) and user_id not in [6477856957]:
         await callback.answer("Sorry your account is frozen", True)
         return
-
     try:
-        data = await db.backup_chatbot.find_one({"user_id": int(user_id)})
+        data = await db.backup_chatbot.find_one({"user_id": user_id})
         if not data:
             return await callback.answer("Can't found user", True)
         get_response = data.get("background_file_id", None)
@@ -505,11 +504,12 @@ async def remover_bg(client, callback):
             reply_markup=None
         )
 
-@Client.on_callback_query(filters.regex("^memory_(\d+)_([a-f0-9]{8})$"))
+@Client.on_callback_query(filters.regex("^memory_([a-f0-9]{8})$"))
 async def memory_updated(client, callback):
-    user_id, uuidstr = callback.matches[0].groups()
-    user_bl = await db.user_blacklists.find_one({"user_id": int(user_id)})
-    if user_bl and user_bl.get("is_frozen", False) and int(user_id) not in [6477856957]:
+    uuidstr = callback.matches[0].group(1)
+    user_id = callback.from_user.id
+    user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+    if user_bl and user_bl.get("is_frozen", False) and user_id not in [6477856957]:
         await callback.answer("Sorry your account is frozen", True)
         return
 
@@ -520,11 +520,11 @@ async def memory_updated(client, callback):
         if user_id == 0:
             return await callback.answer("Server busy, try again later", show_alert=True)
 
-        data_tr = await db.backup_chatbot.find_one({"user_id": int(user_id)})
+        data_tr = await db.backup_chatbot.find_one({"user_id": user_id})
         if not data_tr:
             return await callback.answer("User data not found", show_alert=True)
             
-        backup_chat = await db._get_chatbot_chat_from_db(int(user_id))
+        backup_chat = await db._get_chatbot_chat_from_db(user_id)
         if not backup_chat:
             backup_chat = []
 
@@ -541,7 +541,7 @@ async def memory_updated(client, callback):
             "parts": [{"text": translate_result}]
         })
         await asyncio.gather(
-            db._update_chatbot_chat_in_db(int(user_id), backup_chat),
+            db._update_chatbot_chat_in_db(user_id, backup_chat),
             callback.answer("Memory updated successfully!")
         )
         message_memory_state[callback.message.id] = True
@@ -549,11 +549,12 @@ async def memory_updated(client, callback):
         LOGS.error(f"Memory update failed: {str(e)}")
         await callback.answer("Server error", show_alert=True)
 
-@Client.on_callback_query(filters.regex("^tr_(\d+)_(\w+)_([a-f0-9]{8})$"))
+@Client.on_callback_query(filters.regex("^tr_(\w+)_([a-f0-9]{8})$"))
 async def terjemahkan(client, callback):
-    user_id, targetlang, uuidstr = callback.matches[0].groups()
-    user_bl = await db.user_blacklists.find_one({"user_id": int(user_id)})
-    if user_bl and user_bl.get("is_frozen", False) and int(user_id) not in [6477856957]:
+    targetlang, uuidstr = callback.matches[0].groups()
+    user_id = callback.from_user.id
+    user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+    if user_bl and user_bl.get("is_frozen", False) and user_id not in [6477856957]:
         await callback.answer("Sorry your account is disabled", True)
         return
 
@@ -561,7 +562,7 @@ async def terjemahkan(client, callback):
     try:
         if user_id == 0:
             return await callback.answer("Server busy try again later", True)
-        data_tr = await db.backup_chatbot.find_one({"user_id": int(user_id)})
+        data_tr = await db.backup_chatbot.find_one({"user_id": user_id})
         if not data_tr:
             return await callback.answer("Can't found user", True)
         if not data_tr or "translate_history" not in data_tr:
@@ -590,10 +591,10 @@ async def terjemahkan(client, callback):
         LOGS.error(f"Exception translate: {str(e)}")
         await callback.answer("Server busy try again later", True)
 
-@Client.on_callback_query(filters.regex("^showchat_(\d+)$"))
+@Client.on_callback_query(filters.regex("^showchat$"))
 async def createchat_cb(client, callback):
-    user_id = int(callback.matches[0].group(1))
-    data = await db.backup_chatbot.find_one({"user_id": int(user_id)})
+    user_id = callback.from_user.id
+    data = await db.backup_chatbot.find_one({"user_id": user_id})
     if not data:
         await callback.answer("Can't found user", True)
         return
@@ -603,41 +604,42 @@ async def createchat_cb(client, callback):
         return
     await callback.answer(f"Created Chat: {create_chat_uuid}", True)
             
-@Client.on_callback_query(filters.regex("^trmulti_(\d+)_([a-f0-9]{8})$"))
+@Client.on_callback_query(filters.regex("^trmulti_([a-f0-9]{8})$"))
 async def multiple_langagues(client, callback):
-    user_id, uuidstr = callback.matches[0].groups()
-    user_bl = await db.user_blacklists.find_one({"user_id": int(user_id)})
-    if user_bl and user_bl.get("is_frozen", False) and int(user_id) not in [6477856957]:
+    uuidstr = callback.matches[0].group(1)
+    user_id = callback.from_user.id
+    user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+    if user_bl and user_bl.get("is_frozen", False) and user_id not in [6477856957]:
         await callback.answer("Sorry your account is frozen", True)
         return
 
     keyboard = []
     keyboard.append([
-        InlineKeyboardButton("🇦🇪", callback_data=f"tr_{int(user_id)}_ar_{uuidstr}"),
-        InlineKeyboardButton("🇦🇿", callback_data=f"tr_{int(user_id)}_az_{uuidstr}")
+        InlineKeyboardButton("🇦🇪", callback_data=f"tr_ar_{uuidstr}"),
+        InlineKeyboardButton("🇦🇿", callback_data=f"tr_az_{uuidstr}")
     ])
     keyboard.append([
-        InlineKeyboardButton("🇧🇾", callback_data=f"tr_{int(user_id)}_be_{uuidstr}"),
-        InlineKeyboardButton("🇪🇸", callback_data=f"tr_{int(user_id)}_es_{uuidstr}")
+        InlineKeyboardButton("🇧🇾", callback_data=f"tr_be_{uuidstr}"),
+        InlineKeyboardButton("🇪🇸", callback_data=f"tr_es_{uuidstr}")
     ])
     keyboard.append([
-        InlineKeyboardButton("🇹🇷", callback_data=f"tr_{int(user_id)}_tr_{uuidstr}"),
-        InlineKeyboardButton("🇻🇳", callback_data=f"tr_{int(user_id)}_vi_{uuidstr}")
+        InlineKeyboardButton("🇹🇷", callback_data=f"tr_tr_{uuidstr}"),
+        InlineKeyboardButton("🇻🇳", callback_data=f"tr_vi_{uuidstr}")
     ])
     keyboard.append([
-        InlineKeyboardButton("🇮🇩", callback_data=f"tr_{int(user_id)}_id_{uuidstr}"),
-        InlineKeyboardButton("🇺🇸", callback_data=f"tr_{int(user_id)}_en_{uuidstr}")
+        InlineKeyboardButton("🇮🇩", callback_data=f"tr_id_{uuidstr}"),
+        InlineKeyboardButton("🇺🇸", callback_data=f"tr_en_{uuidstr}")
     ])
     keyboard.append([
-        InlineKeyboardButton("🇮🇳", callback_data=f"tr_{int(user_id)}_hi_{uuidstr}"),
-        InlineKeyboardButton("🇯🇵", callback_data=f"tr_{int(user_id)}_ja_{uuidstr}")
+        InlineKeyboardButton("🇮🇳", callback_data=f"tr_hi_{uuidstr}"),
+        InlineKeyboardButton("🇯🇵", callback_data=f"tr_ja_{uuidstr}")
     ])
     keyboard.append([
-        InlineKeyboardButton("🇲🇾", callback_data=f"tr_{int(user_id)}_ms_{uuidstr}"),
-        InlineKeyboardButton("🇷🇺", callback_data=f"tr_{int(user_id)}_ru_{uuidstr}"),
+        InlineKeyboardButton("🇲🇾", callback_data=f"tr_ms_{uuidstr}"),
+        InlineKeyboardButton("🇷🇺", callback_data=f"tr_ru_{uuidstr}"),
     ])
     keyboard.append([
-        InlineKeyboardButton("💾", callback_data=f"memory_{int(user_id)}_{uuidstr}")
+        InlineKeyboardButton("💾", callback_data=f"memory_{uuidstr}")
     ])
     try:
         await callback.edit_message_reply_markup(
@@ -658,6 +660,16 @@ def anonymous_user(user_id: int = None):
     )
     return keyboard
 
+@Client.on_callback_query(filters.regex("^menu_back:([a-f0-9]{8})$"))
+async def menu_back(client, callback):
+    uuidstr = callback.matches[0].group(1)
+    user_id = callback.from_user.id
+    keyboard = create_keyboard(
+        user_id=user_id,
+        uuid_str=uuidstr
+    )
+    await callback.edit_message_reply_markup(reply_markup=keyboard)
+
 def create_keyboard(
     likes: int = 0,
     unlikes: int = 0,
@@ -669,8 +681,8 @@ def create_keyboard(
         keyboard = InlineKeyboardMarkup(
             [
                 [
-                    InlineKeyboardButton(f"♾️", callback_data=f"trmulti_{user_id}_{uuid_str}"),
-                    InlineKeyboardButton(f"💾", callback_data=f"memory_{user_id}_{uuid_str}"),
+                    InlineKeyboardButton(f"♾️", callback_data=f"trmulti_{uuid_str}"),
+                    InlineKeyboardButton(f"💾", callback_data=f"memory_{uuid_str}"),
                 ],
                 [
                     InlineKeyboardButton(f"👍 {likes}", callback_data="vote:like"),
@@ -682,10 +694,10 @@ def create_keyboard(
         keyboard = InlineKeyboardMarkup(
             [
                 [
-                    InlineKeyboardButton(f"♾️", callback_data=f"trmulti_{user_id}_{uuid_str}"),
-                    InlineKeyboardButton(f"🔒", callback_data=f"showchat_{user_id}"),
+                    InlineKeyboardButton(f"♾️", callback_data=f"trmulti_{uuid_str}"),
+                    InlineKeyboardButton(f"🔒", callback_data=f"showchat"),
                     InlineKeyboardButton(f"⛏️", callback_data=f"modelmenu:{uuid_str}"),
-                    InlineKeyboardButton(f"💾", callback_data=f"memory_{user_id}_{uuid_str}")
+                    InlineKeyboardButton(f"💾", callback_data=f"memory_{uuid_str}")
                 ],
                 [
                     InlineKeyboardButton(f"👍 {likes}", callback_data="vote:like"),
@@ -982,7 +994,7 @@ async def chatbot_talk(client: Client, message: Message):
                     [
                         InlineKeyboardButton(
                             text="✂️ Background Remover",
-                            callback_data=f"removerbg_{message.from_user.id}"
+                            callback_data="removerbg"
                         )
                     ],
                     [
@@ -1246,13 +1258,13 @@ async def chatbot_talk(client: Client, message: Message):
                         [
                             InlineKeyboardButton(
                                 text="🎨 Gemini AI Generate Image",
-                                callback_data=f"genprompt_{message.from_user.id}"
+                                callback_data="genprompt"
                             )
                         ],
                         [
                             InlineKeyboardButton(
                                 text="🖌️ FLUX AI Generate Image",
-                                callback_data=f"fluxprompt_{message.from_user.id}"
+                                callback_data="fluxprompt"
                             )
                         ],
                         [