Update app.py

app.py

@@ -1,6 +1,8 @@
 import os
 from flask import Flask, render_template, request, jsonify
-from memvid_sdk import Memvid
+#from memvid_sdk import Memvid
+from memvid_sdk import create, open as open_memvid
+
 from huggingface_hub import hf_hub_download, upload_file, HfApi
 
 app = Flask(__name__)
@@ -72,14 +74,22 @@ def init_db():
 
     # 2. Initialize Memvid (Fixes 'no attribute create' error)
     try:
-        # In Python SDKs, the constructor usually handles Open OR Create.
-        # If the file exists, it opens it. If not, it creates it.
-        db = Memvid(DB_PATH)
-        print(f"✨ Memvid initialized at {DB_PATH}")
+        if os.path.exists(DB_PATH):
+            db = open_memvid(DB_PATH) # Use the imported 'open' function
+        else:
+            db = create(DB_PATH)      # Use the imported 'create' function
             
-    except Exception as e:
-        print(f"❌ CRITICAL ERROR initializing Memvid: {e}")
-        db = None
+    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:
+            db = Memvid()
+            db.create(DB_PATH)
 
 def sync_to_hub():
     """Uploads the local .mv2 file back to Hugging Face"""