Spaces:
Paused
Paused
Hemang Thakur
commited on
Commit
·
981afc2
1
Parent(s):
e948f71
made changes to graph rag file and some ui elements
Browse files- src/rag/graph_rag.py +6 -0
src/rag/graph_rag.py
CHANGED
@@ -1302,11 +1302,17 @@ class GraphRAG:
|
|
1302 |
net.options["layout"] = {"improvedLayout": True}
|
1303 |
net.options["interaction"] = {"dragNodes": True}
|
1304 |
|
|
|
|
|
|
|
1305 |
net.save_graph("temp_graph.html")
|
1306 |
|
1307 |
with open("temp_graph.html", "r", encoding="utf-8") as f:
|
1308 |
html_str = f.read()
|
|
|
1309 |
os.remove("temp_graph.html")
|
|
|
|
|
1310 |
return html_str
|
1311 |
|
1312 |
def verify_graph_integrity(self):
|
|
|
1302 |
net.options["layout"] = {"improvedLayout": True}
|
1303 |
net.options["interaction"] = {"dragNodes": True}
|
1304 |
|
1305 |
+
original_dir = os.getcwd()
|
1306 |
+
os.chdir(os.getenv("WRITABLE_DIR", "/tmp"))
|
1307 |
+
|
1308 |
net.save_graph("temp_graph.html")
|
1309 |
|
1310 |
with open("temp_graph.html", "r", encoding="utf-8") as f:
|
1311 |
html_str = f.read()
|
1312 |
+
|
1313 |
os.remove("temp_graph.html")
|
1314 |
+
os.chdir(original_dir)
|
1315 |
+
|
1316 |
return html_str
|
1317 |
|
1318 |
def verify_graph_integrity(self):
|