Update utils/memory_utils.py

utils/memory_utils.py

@@ -13,6 +13,7 @@ except ImportError:
     pass
 
 # Local Imports setup
+# مسیر برای ایمپورت کردن فایل‌های کمکی
 bank_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), '../memory_bank')
 sys.path.append(bank_path)
 
@@ -24,7 +25,7 @@ except ImportError:
     def summarize_memory(*args, **kwargs): return {}
 
 # ==========================================
-# 1. تنظیمات مسیر و هاب (Cloud Config)
+# 1. تنظیمات مسیر و هاب (Cloud Config) - اصلاح شده
 # ==========================================
 
 # !!! مهم: نام کاربری و دیتاست خود را اینجا وارد کنید !!!
@@ -34,8 +35,9 @@ REPO_TYPE = "dataset"
 # دریافت توکن
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
-CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
-BASE_DIR = os.path.dirname(CURRENT_DIR) 
+# --- اصلاح مسیرها به Absolute Path ---
+# استفاده از os.getcwd() تضمین می‌کند مسیر همیشه /app/memories است
+BASE_DIR = os.path.abspath(os.getcwd()) 
 MEMORIES_DIR = os.path.join(BASE_DIR, "memories")
 
 # فایل اصلی JSON
@@ -46,8 +48,14 @@ MEMORY_FILE_PATH = os.path.join(MEMORIES_DIR, MEMORY_FILE_NAME)
 MEMORY_INDEX_DIR_NAME = "memory_index"
 MEMORY_INDEX_PATH = os.path.join(MEMORIES_DIR, MEMORY_INDEX_DIR_NAME)
 
-# اطمینان از وجود پوشه در زمان ایمپورت
+# اطمینان از وجود پوشه در زمان اجرا
 os.makedirs(MEMORIES_DIR, exist_ok=True)
+os.makedirs(MEMORY_INDEX_PATH, exist_ok=True)
+
+print(f"📂 System Paths Configuration:")
+print(f"   - Base Dir: {BASE_DIR}")
+print(f"   - Memories Dir: {MEMORIES_DIR}")
+print(f"   - Index Path: {MEMORY_INDEX_PATH}")
 
 
 # ==========================================
@@ -84,10 +92,9 @@ def push_to_hub():
         if os.path.exists(MEMORY_INDEX_PATH):
             # الف) لیست کردن فایل‌ها برای اطمینان از وجود داشتن
             files_count = sum([len(files) for r, d, files in os.walk(MEMORY_INDEX_PATH)])
-            print(f"📂 Found {files_count} files in local index folder. Preparing to upload...")
+            print(f"📂 Found {files_count} files in local index folder ({MEMORY_INDEX_PATH}). Preparing to upload...")
 
             # ب) ساخت یک فایل موقت برای فریب دادن سیستم جهت تشخیص تغییر
-            # این کار باعث می‌شود هش پوشه تغییر کند و آپلود انجام شود
             timestamp_file = os.path.join(MEMORY_INDEX_PATH, "last_sync_log.txt")
             with open(timestamp_file, "w") as f:
                 f.write(f"Sync triggered at: {time.time()}")
@@ -99,7 +106,7 @@ def push_to_hub():
                 repo_id=REPO_ID,
                 repo_type=REPO_TYPE,
                 commit_message=f"Auto-save Indices (Forced): {time.strftime('%Y-%m-%d %H:%M:%S')}",
-                ignore_patterns=[".DS_Store", "*.git*"] # جلوگیری از خطاهای احتمالی سیستمی
+                ignore_patterns=[".DS_Store", "*.git*"]
             )
             
         print(f"✅ Synced all data to Hub Successfully.")
@@ -133,8 +140,7 @@ def pull_from_hub():
         else:
             print("🆕 Memory JSON not found on Hub. Creating fresh.")
 
-        # B) دانلود پوشه ایندکس‌ها (اگر وجود داشته باشد)
-        # بررسی می‌کنیم آیا پوشه memory_index در فایل‌های ریپو هست یا نه
+        # B) دانلود پوشه ایندکس‌ها
         has_index = any(f.startswith(MEMORY_INDEX_DIR_NAME) for f in files)
         
         if has_index:
@@ -143,8 +149,8 @@ def pull_from_hub():
                 repo_id=REPO_ID,
                 repo_type=REPO_TYPE,
                 local_dir=MEMORIES_DIR,
-                allow_patterns=f"{MEMORY_INDEX_DIR_NAME}/**", # فقط پوشه ایندکس را بگیر
-                force_download=True # برای اطمینان از بروزرسانی
+                allow_patterns=f"{MEMORY_INDEX_DIR_NAME}/**",
+                force_download=True
             )
             print("✅ Indices downloaded.")
         else:
@@ -187,7 +193,6 @@ def load_memory():
     # 2. Local Read / Create
     if not os.path.exists(MEMORY_FILE_PATH):
         print(f"⚠️ Memory file missing locally. Creating new.")
-        # فایل خالی می‌سازیم اما هنوز آپلود نمی‌کنیم تا وقتی دیتایی بیاید
         with open(MEMORY_FILE_PATH, "w", encoding="utf-8") as f:
             json.dump({}, f)
         return {}
@@ -210,7 +215,6 @@ def load_memory():
 def enter_name_llamaindex(name, memory_data, data_args, update_memory_index=True):
     is_new_user = False
     
-    # اطمینان از وجود ساختار کاربر در دیکشنری
     if name not in memory_data:
         print(f"🆕 Creating new user profile for: {name}")
         memory_data[name] = {
@@ -221,7 +225,6 @@ def enter_name_llamaindex(name, memory_data, data_args, update_memory_index=True
             "profile": {}
         }
         is_new_user = True
-        # اگر کاربر جدید است، همان اول فایل جیسون را ذخیره کن
         save_memory(memory_data)
     
     user_memory = memory_data[name]
@@ -231,35 +234,37 @@ def enter_name_llamaindex(name, memory_data, data_args, update_memory_index=True
     episodic_memory = None
     semantic_memory = None
 
+    # استفاده از مسیر مطلق برای ساخت آدرس ایندکس
     base_index_dir = os.path.join(MEMORY_INDEX_PATH, "llamaindex", name)
     
     sessions_path = os.path.join(base_index_dir, "sessions_index.json")
     episodic_path = os.path.join(base_index_dir, "episodic_memory_index.json")
     semantic_path = os.path.join(base_index_dir, "semantic_memory_index.json")
 
-    # بررسی اینکه آیا فایل‌های ایندکس به صورت لوکال وجود دارند (یا از ابر دانلود شده‌اند)
     indices_exist = (
         os.path.exists(sessions_path) and 
         os.path.exists(episodic_path) and 
         os.path.exists(semantic_path)
     )
 
-    # اگر ایندکس وجود ندارد یا درخواست آپدیت داریم، بسازیم
     if update_memory_index or not indices_exist:
         try:
             print(f"⚙️ Building/Updating indices locally for {name}...")
+            # اطمینان از وجود پوشه قبل از ساخت
             os.makedirs(base_index_dir, exist_ok=True)
+            
+            # نکته مهم: اگر build_memory_index از مسیرهای نسبی استفاده کند،
+            # باید فایل build_memory_index.py هم بررسی شود.
+            # اما اینجا فرض می‌کنیم مسیرها را از آرگومان‌ها یا گلوبال‌های صحیح می‌گیرد.
             build_memory_index(memory_data, data_args, name=name)
             
-            # **نکته مهم:** الان که فایل‌های ایندکس ساخته شدند، باید آنها را به ابر بفرستیم
-            # تا اگر سرور ریستارت شد، این فایل‌ها نپرند.
             print("☁️ Syncing new indices to cloud...")
             push_to_hub()
             
         except Exception as e:
             print(f"Warning: Error building memory index (skipping): {e}")
 
-    # بارگذاری ایندکس‌ها در حافظه رم (RAM) برای استفاده مدل
+    # بارگذاری ایندکس‌ها
     if os.path.exists(sessions_path):
         try:
             storage_context = StorageContext.from_defaults(persist_dir=sessions_path)
@@ -282,16 +287,11 @@ def enter_name_llamaindex(name, memory_data, data_args, update_memory_index=True
 
 
 def summarize_memory_event_personality(data_args, memory_data, user_name):
-    # قبل از عملیات سنگین، وضعیت فعلی را ذخیره کن
     save_memory(memory_data)
     
     try:
-        # این تابع احتمالا فایل‌های ایندکس را تغییر می‌دهد
         updated_memory = summarize_memory(MEMORY_FILE_PATH, user_name, language='en')
-        
-        # چون خلاصه سازی ممکن است ایندکس‌ها را تغییر دهد، یک سینک نهایی انجام می‌دهیم
         push_to_hub()
-        
         return updated_memory.get(user_name, {})
     except Exception as e:
         print(f"Error in summarize_memory: {e}")