Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-from flask import Flask, render_template, request, jsonify
+from flask import Flask, render_template, request, jsonify, stream_with_context, Response
 from memvid_sdk import create, open as open_memvid
 from huggingface_hub import hf_hub_download, upload_file, HfApi
 
@@ -116,37 +116,56 @@ def index():
 
 @app.route('/add', methods=['POST'])
 def add_memory():
+    # 1. Setup Validation
     global db
-    if not db:
-        init_db()
-        if not db:
-            return jsonify({"error": "Database could not be initialized."}), 500
-
     content = request.form.get('content')
     
     if not content:
         return jsonify({"error": "No content provided"}), 400
 
-    try:
-        payload = {
-            "text": content,
-            "labels": ["web-entry"], 
-            "title": "User Memory"
-        }
+    # 2. Define the Stream Generator
+    def generate():
+        try:
+            # Step A: Re-init if needed inside the stream
+            global db
+            if not db:
+                init_db()
+                if not db:
+                    yield '{"status": "error", "message": "Database init failed"}\n'
+                    return
+
+            # Step B: Database Put
+            yield '{"status": "processing", "message": "Ingesting content..."}\n'
+            
+            payload = {
+                "text": content,
+                "labels": ["web-entry"], 
+                "title": "User Memory"
+            }
+            db.put(payload)
+            
+            # Step C: Flush to Disk
+            yield '{"status": "processing", "message": "Flushing to disk..."}\n'
+            del db
+            db = None
+
+            # Step D: Sync
+            yield '{"status": "processing", "message": "Syncing to cloud (this may take a moment)..."}\n'
+            sync_to_hub()
+            
+            # Step E: Reload
+            yield '{"status": "processing", "message": "Reloading index..."}\n'
+            init_db()
+            
+            # Final Success Message
+            yield '{"status": "success", "message": "Memory added and synced."}\n'
 
-        db.put(payload)
-        
-        # Force flush by deleting reference (Rust Drop trait)
-        del db
-        db = None
+        except Exception as e:
+            # Capture any errors during the process
+            yield f'{{"status": "error", "message": "{str(e)}"}}\n'
 
-        sync_to_hub()
-        init_db()
-        
-        return jsonify({"success": True, "message": "Memory added and synced to cloud."})
-    except Exception as e:
-        if db is None: init_db()
-        return jsonify({"error": str(e)}), 500
+    # 3. Return the Stream
+    return Response(stream_with_context(generate()), mimetype='application/x-ndjson')
 
 @app.route('/search', methods=['POST'])
 def search_memory():