Update graph_utils.py

graph_utils.py

@@ -6,17 +6,17 @@ import os
 model_id = "DReAMy-lib/t5-base-DreamBank-Generation-Act-Char"
 
 def get_graph_dict(graph_text):
-    edge_labels = {}
+    edge_labels = []
     if graph_text == "":
         edge_labels = {("No_Graphs", None):None}
 
     else:
         try:
-            for trpl in graph_text[1:-1].split(", "):
-                h,r,t = trpl.split(" : ") 
-                edge_labels[(h,t)] = "_".join(r.split(" "))
+            for trpl in graph_text[1:-1].split("), ("):
+                h,r,t = trpl.split(" : ")
+                edge_labels.append((h,t, "_".join(r.split(" "))))
         except:
-            edge_labels = {("Error", None):None}
+            edge_labels.append(("Error", None, None))
     return edge_labels
 
 def text_to_graph(text):
@@ -41,7 +41,7 @@ def text_to_graph(text):
     net = Network(directed=True)
 
     # nodes & edges
-    for (h, t), r  in edge_labels.items():
+    for (h, t, r) in edge_labels:
         if (h == "Error") or (h == "No_Graphs"):
             net.add_node(h, shape="circle")
             continue