Update app.py

app.py

@@ -126,22 +126,30 @@ def add_memory():
     if not db:
         init_db()
         if not db:
-            return jsonify({"error": "Database could not be initialized. Check logs."}), 500
+            return jsonify({"error": "Database could not be initialized."}), 500
 
     content = request.form.get('content')
-    tags = request.form.get('tags', '') # Can be passed as comma-separated string
-
+    
     if not content:
         return jsonify({"error": "No content provided"}), 400
 
     try:
-        db.put(content) 
- 
+        payload = {
+            "text": content,
+            "title": "User Memory",
+            "tag": "web-entry"
+        }
+
+        db.put(payload)
+        
+        del db
+        db = None
+
         sync_to_hub()
+        init_db()
         
         return jsonify({"success": True, "message": "Memory added and synced to cloud."})
     except Exception as e:
-        print(f"Error adding memory: {e}")
         return jsonify({"error": str(e)}), 500
 
 @app.route('/search', methods=['POST'])
@@ -149,23 +157,28 @@ def search_memory():
     if not db:
         return jsonify({"error": "Database not initialized"}), 500
 
-    # --- DEBUGGING SNIPPET ---
-    print("🔎 AVAILABLE METHODS ON DB OBJECT:")
-    print([method for method in dir(db) if not method.startswith('_')])
-    # -------------------------
-
     query = request.form.get('query')
     if not query:
         return jsonify({"error": "No query provided"}), 400
 
     try:
-        # Search
-        results = db.find(query, top_k=5)
+        search_req = {
+            "query": query,
+            "top_k": 5,
+            "snippet_chars": 200
+        }
+
+        results = db.find(search_req)
         
         formatted_results = []
         for hit in results:
+            if isinstance(hit, dict):
+                text = hit.get('text') or hit.get('content') or "No text"
+            else:
+                text = getattr(hit, 'text', getattr(hit, 'content', str(hit)))
+
             formatted_results.append({
-                "text": hit.text, 
+                "text": text
             })
             
         return jsonify({"success": True, "results": formatted_results})