Merge pull request #62 from BinNong/main

Graph Visualization

.gitignore

@@ -3,3 +3,4 @@ __pycache__
 dickens/
 book.txt
 lightrag-dev/
+.idea/

examples/graph_visual.py

@@ -0,0 +1,14 @@
+import networkx as nx
+from pyvis.network import Network
+
+# Load the GraphML file
+G = nx.read_graphml('./dickens/graph_chunk_entity_relation.graphml')
+
+# Create a Pyvis network
+net = Network(notebook=True)
+
+# Convert NetworkX graph to Pyvis network
+net.from_nx(G)
+
+# Save and display the network
+net.show('knowledge_graph.html')

requirements.txt

@@ -11,3 +11,4 @@ tiktoken
 torch
 transformers
 xxhash
+pyvis