Update app.py

app.py

@@ -32,7 +32,23 @@ def process_input(user_input, history):
     return history
 
 def visualize_memory():
-    output_html = visualize_reasoned_graph(memory.memories, relationships)
+    # Get all stored memories
+    memories = memory.memories
+
+    # Build relationships between memories (using simple semantic matching)
+    relationships = []
+    for i, (text1, vec1) in enumerate(memories):
+        for j, (text2, vec2) in enumerate(memories):
+            if i != j:
+                # Cosine similarity threshold
+                similarity = np.dot(vec1, vec2) / (np.linalg.norm(vec1) * np.linalg.norm(vec2))
+                if similarity > 0.35:  # threshold for linking related memories
+                    relationships.append((text1, text2, float(similarity)))
+
+    # Generate visualization
+    output_html = visualize_reasoned_graph(memories, relationships)
+
+    # Print confirmation and show output path
     print(f"✅ Visualization created at: {output_html}")
     if os.path.exists(output_html):
         return f"✅ Memory graph generated successfully!\n\nYou can open it here: {output_html}"