Update graph_view.py

graph_view.py

@@ -1,21 +1,21 @@
-import networkx as nx
-import matplotlib.pyplot as plt
-
-def visualize_graph(memories):
-    G = nx.Graph()
-    for m in memories:
-        for other in memories:
-            if m != other:
-                G.add_edge(m['text'][:20], other['text'][:20])
-    plt.figure(figsize=(8,6))
-    nx.draw(G, with_labels=True, node_color='lightblue', font_size=8)
-    plt.show()
-    def visualize_reasoned_graph(memories, relationships, output_html="/mnt/data/graph_reasoned.html"):
+def visualize_reasoned_graph(memories, relationships, output_html="/mnt/data/graph_reasoned.html"):
     from pyvis.network import Network
+
+    # Initialize graph
     net = Network(height="600px", width="100%", bgcolor="#000000", font_color="white")
+
+    # Add nodes (each memory)
     for m in memories:
         net.add_node(m["text"], label=m["text"], color="#00aaff")
+
+    # Add edges (relationships between memories)
     for r in relationships:
-        net.add_edge(r["source"], r["target"], value=r["weight"], title=str(r["weight"]))
+        net.add_edge(
+            r["source"], r["target"],
+            value=r["weight"],
+            title=f"Similarity: {r['weight']}"
+        )
+
+    # Export visualization
     net.show(output_html)
     return output_html