Update app.py

app.py

@@ -1,8 +1,7 @@
 import os
 from flask import Flask, render_template, request, jsonify
-#from memvid_sdk import Memvid
+# Keeping your working imports
 from memvid_sdk import create, open as open_memvid
-
 from huggingface_hub import hf_hub_download, upload_file, HfApi
 
 app = Flask(__name__)
@@ -10,12 +9,12 @@ app = Flask(__name__)
 # CONFIGURATION
 FILENAME = "knowledge.mv2"
 HF_TOKEN = os.environ.get("HF_TOKEN")
-DATASET_NAME = "memvid-storage" # Just the name, we will append username dynamically
+DATASET_NAME = "memvid-storage" 
 
 # Global variables
 db = None
 DB_PATH = os.path.abspath(FILENAME)
-DATASET_REPO_ID = None # Will be set during initialization
+DATASET_REPO_ID = None
 
 def get_repo_id():
     """Helper to dynamically resolve 'username/dataset_name'"""
@@ -50,7 +49,7 @@ def init_db():
         api = HfApi(token=HF_TOKEN)
         
         try:
-            # Create the repo if it doesn't exist (Fixes your 404 error)
+            # Create the repo if it doesn't exist
             api.create_repo(repo_id=repo_id, repo_type="dataset", exist_ok=True)
             
             # Check for file existence
@@ -72,19 +71,16 @@ def init_db():
         except Exception as e:
             print(f"⚠️ Cloud sync warning: {e}")
 
-    # 2. Initialize Memvid (Fixes 'no attribute create' error)
+    # 2. Initialize Memvid (YOUR WORKING LOGIC)
     try:
         if os.path.exists(DB_PATH):
-            db = open_memvid(DB_PATH, read_only=False) # Use the imported 'open' function
+            db = open_memvid(DB_PATH, read_only=False) 
         else:
-            db = create(DB_PATH)      # Use the imported 'create' function
+            db = create(DB_PATH)      
             
     except ImportError:
-        # If the above imports fail, fall back to the class method style
-        # (Only do this if the library version is different than expected)
         from memvid_sdk import Memvid
         if os.path.exists(DB_PATH):
-             # Some bindings require an empty init first
             db = Memvid()
             db.open(DB_PATH) 
         else:
@@ -134,14 +130,11 @@ def add_memory():
         return jsonify({"error": "No content provided"}), 400
 
     try:
-        payload = {
-            "text": content,
-            "title": "User Memory",
-            "tag": "web-entry"
-        }
-
-        db.put(payload)
+        # FIX: Pass arguments as labels (Keyword Arguments), NOT as a dict object.
+        # "put requires labels" means db.put(text=...) not db.put({"text":...})
+        db.put(text=content, title="User Memory", tag="web-entry")
         
+        # Force flush by deleting reference (Rust Drop trait)
         del db
         db = None
 
@@ -150,6 +143,8 @@ def add_memory():
         
         return jsonify({"success": True, "message": "Memory added and synced to cloud."})
     except Exception as e:
+        # Re-init db if something crashed so next request works
+        if db is None: init_db()
         return jsonify({"error": str(e)}), 500
 
 @app.route('/search', methods=['POST'])
@@ -162,16 +157,13 @@ def search_memory():
         return jsonify({"error": "No query provided"}), 400
 
     try:
-        search_req = {
-            "query": query,
-            "top_k": 5,
-            "snippet_chars": 200
-        }
-
-        results = db.find(search_req)
+        # FIX: Pass query as positional string, options as kwargs.
+        # Previous error: "dict cannot be converted to PyString" -> means first arg must be string.
+        results = db.find(query, top_k=5)
         
         formatted_results = []
         for hit in results:
+            # Handle result object vs dict safety
             if isinstance(hit, dict):
                 text = hit.get('text') or hit.get('content') or "No text"
             else: