Update Emma/app.py

Emma/app.py

@@ -35,7 +35,6 @@ try:
     from utils.app_modules.presets import *
     from utils.app_modules.overwrites import *
     from utils.prompt_utils import *
-    # نکته: save_local_memory قدیمی حذف شد و با save_memory جایگزین شد
 except ImportError:
     data_args = {}
     boot_actual_name_dict = {'en': 'Emma'}
@@ -102,9 +101,6 @@ def read_apis(path):
 
 api_keys = read_apis(api_path)
 
-# نکته: بخش لود کردن memory به صورت سراسری از اینجا حذف شد.
-# مدیریت فایل کاملاً به memory_utils سپرده شده است.
-
 language = 'en'
 boot_actual_name = boot_actual_name_dict.get(language, "Emma")
 meta_prompt = generate_meta_prompt_dict_chatgpt().get(language, "")
@@ -124,6 +120,7 @@ def chatgpt_chat(prompt, system, history, gpt_config, api_index=0):
             messages = [{"role": "system", "content": system}]
             if history:
                 for h in history:
+                    # اطمینان از فرمت صحیح پیام‌ها قبل از ارسال به API
                     if isinstance(h, dict) and 'role' in h and 'content' in h:
                         messages.append(h)
             messages.append({"role": "user", "content": prompt})
@@ -168,7 +165,7 @@ def predict_new(
         system_prompt, _ = build_prompt_with_search_memory_llamaindex(
             history=history,
             query=text,
-            user_memory=user_memory, # اینجا user_memory تازه دریافت می‌شود
+            user_memory=user_memory,
             user_name=user_name,
             user_memory_index=None,
             service_context=None, 
@@ -190,14 +187,12 @@ def predict_new(
     hist_for_llm = history[-10:] if len(history) > 10 else history
     response = chatgpt_chat(text, system_prompt, hist_for_llm, chat_cfg, api_index)
     
+    # اینجا فرمت جدید دیکشنری را می‌سازیم که با type="messages" سازگار است
     new_history = history + [
         {"role": "user", "content": text},
         {"role": "assistant", "content": response}
     ]
     
-    # نکته مهم: عملیات ذخیره‌سازی از اینجا حذف شد.
-    # ذخیره‌سازی باید در لایه بالاتر (respond) انجام شود که کل آبجکت مموری را در اختیار دارد.
-
     return new_history, new_history
 
 # ==========================================
@@ -205,7 +200,6 @@ def predict_new(
 # ==========================================
 
 def create_gradio_interface():
-    # استایل‌های CSS قدیمی شما
     css = """
     .mobile-button button {
         width: 100% !important;
@@ -234,8 +228,6 @@ def create_gradio_interface():
 
     with gr.Blocks(title="EMMA AI", css=css) as demo:
         
-        # State: فقط داده‌های موقت را نگه می‌دارد. 
-        # کل مموری دیگر در State نیست تا از تداخل جلوگیری شود.
         state = gr.State({
             "history": [], 
             "user_name": None, 
@@ -246,7 +238,6 @@ def create_gradio_interface():
 
         header = gr.Markdown("## 🧠 EMMA: Your Empathetic Mental Health Assistant\nWelcome! Please enter your name to begin.")
         
-        # --- بخش لاگین ---
         with gr.Column(visible=True) as login_col:
             with gr.Accordion("🔐 Start New Session", open=True):
                 with gr.Row():
@@ -258,11 +249,10 @@ def create_gradio_interface():
                     residence_input = gr.Textbox(label="Place of Residence", placeholder="e.g., Berlin")
                 start_btn = gr.Button("🎯 Start Session", variant="primary")
 
-        # --- بخش اعلان سیستم ---
         system_msg = gr.Textbox(label="🔔 System Messages", interactive=False, max_lines=2, visible=False)
 
-        # --- بخش چت ---
         with gr.Column(visible=False) as chat_col:
+            # type="messages" یعنی ورودی باید لیست دیکشنری باشد
             chatbot = gr.Chatbot(height=500, type="messages", label="💬 EMMA Conversation")
             
             with gr.Row():
@@ -275,26 +265,25 @@ def create_gradio_interface():
                 switch_user_btn = gr.Button("👥 Switch User", variant="primary", elem_classes=["mobile-button"])
 
         # ==========================================
-        # توابع کنترلی UI (اصلاح شده برای ذخیره‌سازی صحیح)
+        # توابع کنترلی UI (اصلاح شده)
         # ==========================================
 
         def start_session(name, age, gender, job, city, current_state):
             if not name or not name.strip():
                 return (
-                    gr.update(), gr.update(), gr.update(), # login, chat, system
+                    gr.update(), gr.update(), gr.update(),
                     gr.update(value="## ⚠️ Please enter a valid name."),
                     current_state,
-                    [] # empty history for chatbot
+                    [] 
                 )
             
-            # 1. بارگذاری حافظه تازه از دیسک
+            # 1. بارگذاری حافظه
             memory_data = load_memory()
             
-            # 2. ایجاد یا به‌روزرسانی یوزر در حافظه
-            # enter_name_llamaindex روی دیکشنری memory_data تغییر ایجاد می‌کند
+            # 2. ایجاد یا به‌روزرسانی یوزر
             enter_name_llamaindex(name, memory_data, data_args)
             
-            # 3. ذخیره اطلاعات پروفایل
+            # 3. آپدیت پروفایل
             if name in memory_data:
                 memory_data[name]["profile"] = {
                     "age": age, "gender": gender, 
@@ -302,32 +291,53 @@ def create_gradio_interface():
                 }
                 
                 # اطمینان از وجود لیست سشن‌ها
-                if "sessions" not in memory_data[name] or not memory_data[name]["sessions"]:
+                if "sessions" not in memory_data[name] or not isinstance(memory_data[name]["sessions"], list):
+                     memory_data[name]["sessions"] = [{
+                        "session_id": 0, "date": time.strftime("%Y-%m-%d"), "conversation": []
+                    }]
+                elif len(memory_data[name]["sessions"]) == 0:
                      memory_data[name]["sessions"] = [{
                         "session_id": 0, "date": time.strftime("%Y-%m-%d"), "conversation": []
                     }]
 
-            # 4. ذخیره کل حافظه روی دیسک (بسیار مهم)
+            # 4. ذخیره حافظه
             save_memory(memory_data)
             print(f"✅ User {name} loaded/created and saved.")
 
-            # 5. به‌روزرسانی State
+            # 5. استخراج و استانداردسازی تاریخچه (FIXED HERE)
+            # ------------------------------------------------------------------
+            user_mem = memory_data.get(name, {})
+            raw_history = []
+            
+            sessions = user_mem.get("sessions", [])
+            if sessions and len(sessions) > 0:
+                # گرفتن آخرین سشن
+                raw_history = sessions[-1].get("conversation", [])
+            
+            # تبدیل فرمت‌های قدیمی یا ناسازگار به فرمت جدید type="messages"
+            formatted_history = []
+            for item in raw_history:
+                # حالت ۱: فرمت قدیمی لیست [user, bot]
+                if isinstance(item, list) and len(item) == 2:
+                    if item[0]: # user
+                        formatted_history.append({"role": "user", "content": str(item[0])})
+                    if item[1]: # bot
+                        formatted_history.append({"role": "assistant", "content": str(item[1])})
+                # حالت ۲: فرمت صحیح دیکشنری
+                elif isinstance(item, dict) and "role" in item and "content" in item:
+                    formatted_history.append(item)
+            # ------------------------------------------------------------------
+
             new_st = copy.deepcopy(current_state)
             new_st["user_name"] = name
             new_st["init"] = True
+            new_st["history"] = formatted_history # ذخیره فرمت صحیح در state
             
-            # استخراج حافظه معنایی برای کانتکست
             try:
-                user_mem = memory_data.get(name, {})
                 if isinstance(user_mem, dict):
                     new_st["semantic"] = str(user_mem.get("semantic_memory", ""))
-                
-                # بازیابی تاریخچه آخرین سشن (اختیاری - اگر می‌خواهید چت قبلی نمایش داده شود)
-                last_history = user_mem.get("sessions", [{}])[-1].get("conversation", [])
-                new_st["history"] = last_history
             except: 
-                last_history = []
-                new_st["history"] = []
+                new_st["semantic"] = ""
 
             welcome_txt = f"## 🧠 EMMA: Session for {name}"
             sys_txt = f"Welcome {name}! Memory loaded successfully."
@@ -338,7 +348,7 @@ def create_gradio_interface():
                 gr.update(visible=True, value=sys_txt), # Show System Msg
                 gr.update(value=welcome_txt), # Update Header
                 new_st,
-                last_history # نمایش تاریخچه در چت‌بات
+                formatted_history # ارسال تاریخچه استاندارد شده به چت‌بات
             )
 
         def respond(msg, current_state):
@@ -347,39 +357,45 @@ def create_gradio_interface():
             
             user_name = current_state["user_name"]
             
-            # 1. بارگذاری حافظه تازه برای اطمینان از داشتن آخرین وضعیت
+            # لود مجدد برای اطمینان
             memory_data = load_memory()
             user_memory = memory_data.get(user_name, {})
 
             cat = classify_query_local(msg)
             
-            # 2. تولید پاسخ (فقط تولید، بدون ذخیره)
+            # استفاده از هیستوری موجود در state که قبلاً استاندارد شده
+            current_history = current_state.get("history", [])
+
             _, new_hist = predict_new(
                 text=msg,
-                history=current_state["history"],
+                history=current_history,
                 top_p=0.9, temperature=0.7, max_length_tokens=512, max_context_length_tokens=200,
                 user_name=user_name,
-                user_memory=user_memory, # ارسال آبجکت یوزر مموری
+                user_memory=user_memory,
                 api_index=current_state["api_index"],
-                semantic_memory_text=current_state["semantic"],
+                semantic_memory_text=current_state.get("semantic", ""),
                 query_category=cat
             )
             
-            # 3. اعمال تغییرات در آبجکت کل حافظه
-            if user_name in memory_data and memory_data[user_name].get("sessions"):
-                memory_data[user_name]["sessions"][-1]["conversation"] = new_hist
+            # ذخیره در آبجکت کل
+            if user_name in memory_data:
+                sessions = memory_data[user_name].get("sessions", [])
+                if sessions:
+                    sessions[-1]["conversation"] = new_hist
+                else:
+                    # اگر به هر دلیلی سشن نبود بساز
+                    memory_data[user_name]["sessions"] = [{
+                        "session_id": 0, "date": time.strftime("%Y-%m-%d"), "conversation": new_hist
+                    }]
             
-            # 4. ذخیره نهایی روی دیسک
             save_memory(memory_data)
             
-            # 5. آپدیت State گرادیو
             new_st = copy.deepcopy(current_state)
             new_st["history"] = new_hist
             
             return new_hist, new_st, ""
 
         def clear_chat(current_state):
-            # پاک کردن چت فقط در محیط بصری و State فعلی (بدون حذف از دیتابیس برای امنیت)
             new_st = copy.deepcopy(current_state)
             new_st["history"] = []
             return [], new_st, "Conversation view cleared (Memory retained)."
@@ -389,40 +405,31 @@ def create_gradio_interface():
             if not u_name:
                 return [], current_state, "Error: No user."
 
-            # 1. بارگذاری حافظه
             memory_data = load_memory()
 
             if u_name in memory_data:
                 try:
                     user_sessions = memory_data[u_name].get("sessions", [])
-                    
-                    # ایجاد سشن جدید
                     new_s = {
                         "session_id": len(user_sessions),
                         "date": time.strftime("%Y-%m-%d"),
-                        "conversation": []
+                        "conversation": [] # شروع با لیست خالی
                     }
-                    
-                    if "sessions" not in memory_data[u_name]:
-                        memory_data[u_name]["sessions"] = []
+                    if not isinstance(user_sessions, list):
+                         memory_data[u_name]["sessions"] = []
                     
                     memory_data[u_name]["sessions"].append(new_s)
-                    
-                    # 2. ذخیره تغییرات
                     save_memory(memory_data)
                     print(f"🔄 New session created for {u_name}")
-
                 except Exception as e:
                     print(f"New Session Error: {e}")
 
-            # ریست کردن State
             new_st = copy.deepcopy(current_state)
             new_st["history"] = []
             
             return [], new_st, "🆕 New session started & Saved."
 
         def switch_user_logic(current_state):
-            # ریست کامل State
             new_st = {
                 "history": [], 
                 "user_name": None, 
@@ -431,17 +438,17 @@ def create_gradio_interface():
                 "init": False
             }
             return (
-                gr.update(visible=True),   # Show Login
-                gr.update(visible=False),  # Hide Chat
-                gr.update(visible=False, value=""), # Hide System Msg
+                gr.update(visible=True),   
+                gr.update(visible=False),  
+                gr.update(visible=False, value=""), 
                 gr.update(value="## 🧠 EMMA: Switch User Mode"),
                 new_st,
-                gr.update(value=""), # Clear Name Input
-                [] # Clear Chatbot
+                gr.update(value=""), 
+                [] 
             )
 
         # ==========================================
-        # اتصال رویدادها (Event Listeners)
+        # اتصا�� رویدادها
         # ==========================================
         
         start_btn.click(
@@ -508,7 +515,7 @@ def main():
 
     demo = create_gradio_interface()
     
-    print("🚀 Launching Gradio Server (Fixed Memory Saving)...")
+    print("🚀 Launching Gradio Server (Fixed Message Format)...")
     demo.queue().launch(
         server_name="0.0.0.0",
         server_port=7860,