Ok

chatbot/plugins/chat.py

@@ -695,37 +695,43 @@ async def reshch(client, callback):
 async def closeed(client, callback):
     await callback.message.delete()
 
-@Client.on_callback_query(filters.regex("^memory_(\d+)$"))
+@Client.on_callback_query(filters.regex("^memory_(\d+)_([a-f0-9]{8})$"))
 async def memory_updated(client, callback):
-    user_id = int(callback.matches[0].group(1))
-    user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+    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):
         await callback.answer("Sorry your account is disabled", True)
         return
 
+    translate_result = ""
     if message_memory_state.get(callback.message.id, False):
         return await callback.answer("Memory already updated!", show_alert=False)
     try:
         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": user_id})
+        data_tr = await db.backup_chatbot.find_one({"user_id": int(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(user_id)
+        backup_chat = await db._get_chatbot_chat_from_db(int(user_id))
         if not backup_chat:
             backup_chat = []
 
-        get_response = data_tr.get("translate_text")
-        if not get_response:
-            return await callback.answer("No translation available", show_alert=True)
+        if not data_tr or "translate_history" not in data_tr:
+            return await callback.answer("translate history not found")
+        
+        for translations in data_tr["translate_history"]:
+            if translations.get("uuid") == uuidstr:
+                translate_result += translations.get("text", "no translation")
+                break
+    
         backup_chat.append({
             "role": "model",
-            "parts": [{"text": get_response}]
+            "parts": [{"text": translate_result}]
         })
         await asyncio.gather(
-            db._update_chatbot_chat_in_db(user_id, backup_chat),
+            db._update_chatbot_chat_in_db(int(user_id), backup_chat),
             callback.answer("Memory updated successfully!")
         )
         message_memory_state[callback.message.id] = True
@@ -752,7 +758,6 @@ async def terjemahkan(client, callback):
             return await callback.answer("translate history not found")
         
         for translations in data_tr["translate_history"]:
-            LOGS.info(f"Update translation: {translations['uuid']}")
             if translations.get("uuid") == uuidstr:
                 translate_result += translations.get("text", "no translation")
                 break
@@ -838,6 +843,9 @@ async def multiple_langagues(client, callback):
         InlineKeyboardButton("🇲🇾", callback_data=f"tr_{int(user_id)}_ms_{uuidstr}"),
         InlineKeyboardButton("🇷🇺", callback_data=f"tr_{int(user_id)}_ru_{uuidstr}"),
     ])
+    keyboard.append([
+        InlineKeyboardButton("💾", callback_data=f"memory_{int(user_id)}_{uuidstr}")
+    ])
     try:
         await callback.edit_message_reply_markup(
             reply_markup=InlineKeyboardMarkup(keyboard)
@@ -870,8 +878,6 @@ async def response_call(client, callback):
             [
                 InlineKeyboardButton(f"🗃️", callback_data=f"menutools_{int(user_id)}"),
                 InlineKeyboardButton(f"🔎", url=f"https://t.me/{chat}"),
-                InlineKeyboardButton(f"♾️", callback_data=f"trmulti_{int(user_id)}"),
-                InlineKeyboardButton(f"💾", callback_data=f"memory_{int(user_id)}"),
             ],
             [
                 InlineKeyboardButton(f"👍 0", callback_data="like"),
@@ -896,7 +902,7 @@ def create_keyboard(
                     InlineKeyboardButton(f"🗃️", callback_data=f"menutools_{user_id}"),
                     InlineKeyboardButton(f"🔎", url=f"https://t.me/{chat}"),
                     InlineKeyboardButton(f"♾️", callback_data=f"trmulti_{user_id}_{uuid_str}"),
-                    InlineKeyboardButton(f"💾", callback_data=f"memory_{user_id}"),
+                    InlineKeyboardButton(f"💾", callback_data=f"memory_{user_id}_{uuid_str}"),
                 ],
                 [
                     InlineKeyboardButton(f"👍 {likes}", callback_data="vote:like"),
@@ -911,7 +917,7 @@ def create_keyboard(
                     InlineKeyboardButton(f"🗃️", callback_data=f"menutools_{user_id}"),
                     InlineKeyboardButton(f"♾️", callback_data=f"trmulti_{user_id}_{uuid_str}"),
                     InlineKeyboardButton(f"🔁", callback_data=f"refreshch"),
-                    InlineKeyboardButton(f"💾", callback_data=f"memory_{user_id}")
+                    InlineKeyboardButton(f"💾", callback_data=f"memory_{user_id}_{uuid_str}")
                 ],
                 [
                     InlineKeyboardButton(f"👍 {likes}", callback_data="vote:like"),