Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -54,6 +54,7 @@ from google import genai
 from google.genai import types as ty
 
 trans = SyncTranslator()
+message_memory_state = {}
 
 async def geni_files_delete(name: str):
     url = f"https://generativelanguage.googleapis.com/v1beta/{name}"
@@ -234,29 +235,42 @@ async def closeed(client, callback):
 @Client.on_callback_query(filters.regex("^memory_(\d+)$"))
 async def memory_updated(client, callback):
     user_id = int(callback.matches[0].group(1))
-    check_memory = {}
-    if callback.message.id not in check_memory:
-        check_memory[callback.message.id] = False
-    is_memory = check_memory[callback.message.id]
-    if is_memory:
-        return await callback.answer("Memory is Already Updated!", False)
-    backup_chat = await db._get_chatbot_chat_from_db(user_id)
+
+    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", True)
+            return await callback.answer("Server busy, try again later", show_alert=True)
+
         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)
-        get_response = data_tr.get("translate_text", None)
+            return await callback.answer("User data not found", show_alert=True)
+            
+        backup_chat = await db._get_chatbot_chat_from_db(user_id)
+        if not backup_chat:
+            backup_chat = []
+
+        get_response = data_tr.get("translate_text")
         if not get_response:
-            return await callback.answer("Server busy try again later", True)
-        backup_chat.append({"role": "model", "parts": [{"text": get_response}]})
-        await db._update_chatbot_chat_in_db(user_id, backup_chat)
-        check_memory[callback.message.id] = True
-        await callback.answer("Memory Updated!", False)
+            return await callback.answer("No translation available", show_alert=True)
+        
+
+        backup_chat.append({
+            "role": "model",
+            "parts": [{"text": get_response}]
+        })
+
+        await asyncio.gather(
+            db._update_chatbot_chat_in_db(user_id, backup_chat),
+            callback.answer("Memory updated successfully!")
+        )
+        
+        message_memory_state[callback.message.id] = True
+        
     except Exception as e:
-        LOGS.error(f"Exception translate: {str(e)}")
-        await callback.answer("Server busy try again later", True)
+        LOGS.error(f"Memory update failed: {str(e)}")
+        await callback.answer("Server error", show_alert=True)
 
 @Client.on_callback_query(filters.regex("^tr_(\d+)$"))
 async def terjemahkan(client, callback):