Update app.py

app.py

@@ -33,6 +33,7 @@ def init_db():
     
     repo_id = get_repo_id()
     
+    # 1. Cloud Sync
     if HF_TOKEN and repo_id:
         api = HfApi(token=HF_TOKEN)
         try:
@@ -51,17 +52,40 @@ def init_db():
         except Exception as e:
             print(f"Cloud sync error: {e}")
 
+    # 2. Initialize Memvid (The Fix)
     try:
-        # FIX 1: Explicitly disable read_only mode to allow mutation (put)
-        # We try passing it as a keyword argument first
-        try:
-            db = Memvid(DB_PATH, read_only=False)
-        except TypeError:
-            # Fallback for older bindings: positional arg
-            db = Memvid(DB_PATH, False)
-            
+        if os.path.exists(DB_PATH):
+            print(f"📂 Opening existing database at {DB_PATH}")
+            # FIX: Use the static .open() method. 
+            # The constructor Memvid(path) defaults to ReadOnly.
+            # Memvid.open(path) typically opens in Read/Write mode.
+            if hasattr(Memvid, 'open'):
+                db = Memvid.open(DB_PATH)
+            else:
+                # Fallback if .open isn't static
+                db = Memvid(DB_PATH)
+        else:
+            print(f"✨ Creating new database at {DB_PATH}")
+            # FIX: Use the static .create() method for new files.
+            if hasattr(Memvid, 'create'):
+                db = Memvid.create(DB_PATH)
+            else:
+                # Fallback
+                db = Memvid(DB_PATH)
+                
+        # Double check we have a valid object
+        if db:
+            print("✅ Database initialized successfully.")
+        else:
+            print("❌ Database object is None after init.")
+
     except Exception as e:
-        print(f"Memvid init error: {e}")
+        print(f"❌ Memvid init CRITICAL ERROR: {e}")
+        # Print attributes to help debug if it fails again
+        try:
+            print(f"Available attributes: {dir(Memvid)}")
+        except:
+            pass
         db = None
 
 def sync_to_hub():
@@ -81,6 +105,7 @@ def sync_to_hub():
     except Exception as e:
         print(f"Sync failed: {e}")
 
+# Run init on start
 init_db()
 
 @app.route('/')
@@ -93,14 +118,14 @@ def add_memory():
     if not db:
         init_db()
         if not db:
-            return jsonify({"error": "Database could not be initialized."}), 500
+            return jsonify({"error": "Database could not be initialized. Check Server Logs."}), 500
 
     content = request.form.get('content')
-    
     if not content:
         return jsonify({"error": "No content provided"}), 400
 
     try:
+        # Payload must be a dictionary
         payload = {
             "text": content,
             "title": "User Memory",
@@ -109,14 +134,19 @@ def add_memory():
 
         db.put(payload)
         
+        # Force flush/close to save data before sync
         del db
         db = None
 
+        # Sync to Hugging Face
         sync_to_hub()
+        
+        # Re-open the database for the next request
         init_db()
         
         return jsonify({"success": True, "message": "Memory added and synced to cloud."})
     except Exception as e:
+        print(f"Add Error: {e}")
         return jsonify({"error": str(e)}), 500
 
 @app.route('/search', methods=['POST'])
@@ -129,16 +159,16 @@ def search_memory():
         return jsonify({"error": "No query provided"}), 400
 
     try:
-        # FIX 2: Pass 'query' as a string, not a dict. 
-        # The previous error "cannot convert dict to PyString" means 'find' expects a string arg.
-        # We pass top_k as a keyword argument.
+        # FIX: query must be string, top_k is keyword arg
         results = db.find(query, top_k=5)
         
         formatted_results = []
         for hit in results:
+            # Handle result object vs dict
             if isinstance(hit, dict):
                 text = hit.get('text') or hit.get('content') or "No text"
             else:
+                # Try common attribute names for the hit object
                 text = getattr(hit, 'text', getattr(hit, 'content', str(hit)))
 
             formatted_results.append({
@@ -147,6 +177,7 @@ def search_memory():
             
         return jsonify({"success": True, "results": formatted_results})
     except Exception as e:
+        print(f"Search Error: {e}")
         return jsonify({"error": str(e)}), 500
 
 if __name__ == '__main__':