Update utils/memory_utils.py

utils/memory_utils.py

@@ -10,7 +10,6 @@ import gradio as gr
 from llama_index.core import StorageContext, load_index_from_storage
 
 # Local Imports setup
-# فرض بر این است که مسیرها درست تنظیم شده‌اند
 bank_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), '../memory_bank')
 sys.path.append(bank_path)
 
@@ -18,16 +17,14 @@ try:
     from build_memory_index import build_memory_index
     from summarize_memory import summarize_memory
 except ImportError:
-    # print("Warning: Could not import memory build/summary modules.")
     def build_memory_index(*args, **kwargs): pass
     def summarize_memory(*args, **kwargs): return {}
 
 # ==========================================
 # 1. تنظیمات مسیر (Single Source of Truth)
 # ==========================================
-# مسیردهی نسبی برای سازگاری با Hugging Face Spaces (Linux)
 CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
-BASE_DIR = os.path.dirname(CURRENT_DIR) # یک پله بالاتر (روت پروژه)
+BASE_DIR = os.path.dirname(CURRENT_DIR) 
 MEMORIES_DIR = os.path.join(BASE_DIR, "memories")
 MEMORY_FILE_PATH = os.path.join(MEMORIES_DIR, "update_memory_0512_eng.json")
 
@@ -38,51 +35,59 @@ os.makedirs(MEMORIES_DIR, exist_ok=True)
 # 2. توابع اصلی I/O (خواندن و نوشتن)
 # ==========================================
 
-def load_memory():
-    """
-    حافظه را از فایل JSON بارگذاری می‌کند. 
-    اگر فایل وجود نداشته باشد یا خراب باشد، یک دیکشنری خالی برمی‌گرداند.
-    """
-    if os.path.exists(MEMORY_FILE_PATH):
-        try:
-            with open(MEMORY_FILE_PATH, 'r', encoding='utf-8') as f:
-                return json.load(f)
-        except (json.JSONDecodeError, FileNotFoundError):
-            print("⚠️ Memory file is corrupted or not found. Starting with empty memory.")
-            return {}
-    return {}
-
-
 def save_memory(data):
     """
     کل آبجکت حافظه را به صورت ایمن (Atomic Write) در فایل JSON ذخیره می‌کند.
     """
     try:
-        # نوشتن در فایل موقت ابتدا برای جلوگیری از خرابی داده در صورت قطع برق/کرش
+        # اطمینان مجدد از وجود پوشه
+        os.makedirs(MEMORIES_DIR, exist_ok=True)
+        
+        # نوشتن در فایل موقت
         temp_file = MEMORY_FILE_PATH + ".tmp"
         with open(temp_file, "w", encoding="utf-8") as f:
             json.dump(data, f, ensure_ascii=False, indent=4)
         
-        # جایگزینی فایل اصلی با فایل موقت
+        # جایگزینی فایل اصلی
         os.replace(temp_file, MEMORY_FILE_PATH)
         # print(f"✅ Memory successfully saved to {MEMORY_FILE_PATH}")
     except Exception as e:
         print(f"❌ CRITICAL ERROR saving memory: {e}")
 
 
+def load_memory():
+    """
+    حافظه را بارگذاری می‌کند. اگر فایل نباشد، آن را می‌سازد.
+    """
+    if not os.path.exists(MEMORY_FILE_PATH):
+        print(f"⚠️ Memory file not found. Creating new file at: {MEMORY_FILE_PATH}")
+        # ایجاد فایل خالی فیزیکی
+        save_memory({})
+        return {}
+
+    try:
+        with open(MEMORY_FILE_PATH, 'r', encoding='utf-8') as f:
+            # اگر فایل خالی بود، هندل شود
+            content = f.read()
+            if not content.strip():
+                return {}
+            return json.loads(content)
+    except (json.JSONDecodeError, FileNotFoundError):
+        print("⚠️ Memory file is corrupted. Starting with empty memory.")
+        return {}
+
+
 # ==========================================
 # 3. توابع منطق حافظه (Memory Logic)
 # ==========================================
 
 def enter_name_llamaindex(name, memory_data, data_args, update_memory_index=True):
     """
-    کاربر را در دیکشنری حافظه ایجاد یا پیدا می‌کند و ایندکس‌های LlamaIndex را لود می‌کند.
-    
-    IMPORTANT: این تابع دیگر فایل را نمی‌خواند/نمی‌نویسد. فقط روی آبجکت memory_data در حافظه RAM کار می‌کند.
-    ذخیره‌سازی باید توسط فراخواننده (caller) انجام شود.
+    کاربر را ایجاد کرده و فایل را فوراً ذخیره می‌کند تا از دست نرود.
     """
     
-    # 1. مدیریت ساختار دیکشنری در RAM
+    # 1. مدیریت ساختار دیکشنری
+    is_new_user = False
     if name not in memory_data:
         print(f"🆕 Creating new user profile for: {name}")
         memory_data[name] = {
@@ -92,15 +97,19 @@ def enter_name_llamaindex(name, memory_data, data_args, update_memory_index=True
             "semantic_memory": {},
             "profile": {}
         }
+        is_new_user = True
     
     user_memory = memory_data[name]
 
+    # نکته مهم: اگر کاربر جدید است، همین الان ذخیره کن تا اگر برنامه کرش کرد، نام بماند
+    if is_new_user:
+        save_memory(memory_data)
+
     # 2. مدیریت ایندکس‌های وکتوری (LlamaIndex)
     sessions_memory = None
     episodic_memory = None
     semantic_memory = None
 
-    # مسیردهی برای ایندکس‌ها (نسبی و لینوکسی)
     base_index_dir = os.path.join(MEMORIES_DIR, "memory_index", "llamaindex", name)
     
     sessions_path = os.path.join(base_index_dir, "sessions_index.json")
@@ -113,16 +122,11 @@ def enter_name_llamaindex(name, memory_data, data_args, update_memory_index=True
         os.path.exists(semantic_path)
     )
 
-    # ساخت ایندکس اگر وجود ندارد یا درخواست آپدیت شده است
     if update_memory_index or not indices_exist:
-        # print(f'Initializing memory indices for {name}...')
         try:
             os.makedirs(base_index_dir, exist_ok=True)
-            # توجه: build_memory_index ممکن است نیاز به خواندن فایل داشته باشد، 
-            # اما فعلاً فرض می‌کنیم درست کار می‌کند یا safe fail می‌شود.
             build_memory_index(memory_data, data_args, name=name)
         except Exception as e:
-            # خطای بیلد ایندکس نباید برنامه را متوقف کند
             print(f"Warning: Error building memory index (skipping): {e}")
 
     # بارگذاری ایندکس‌ها
@@ -148,14 +152,10 @@ def enter_name_llamaindex(name, memory_data, data_args, update_memory_index=True
 
 
 def summarize_memory_event_personality(data_args, memory_data, user_name):
-    """
-    Wrapper برای تابع summarize_memory.
-    """
-    # ابتدا مطمئن می‌شویم داده‌های فعلی ذخیره شده‌اند تا summarizer فایل به‌روز را بخواند
+    # ذخیره قبل از سامرایز برای اطمینان
     save_memory(memory_data)
     
     try:
-        # تابع summarize_memory احتمالاً خودش فایل را می‌خواند
         updated_memory = summarize_memory(MEMORY_FILE_PATH, user_name, language='en')
         return updated_memory.get(user_name, {})
     except Exception as e: