Upload app.py

app.py

@@ -10,6 +10,7 @@ from glycowork.motif.graph import glycan_to_nxGraph
 import networkx as nx
 import matplotlib.pyplot as plt
 from networkx.drawing.nx_agraph import write_dot
+import pydot
 # import pygraphviz as pgv
 
 
@@ -40,10 +41,12 @@ model3 = torch.load("model3.pt", map_location=torch.device('cpu'))
 
 def fn(glycan, model):
     # Draw graph
-    # graph = glycan_to_nxGraph(glycan)
-    # node_labels = nx.get_node_attributes(graph, 'string_labels')
-    # labels = {i:node_labels[i] for i in range(len(graph.nodes))}
-    # graph = nx.relabel_nodes(graph, labels)
+    graph = glycan_to_nxGraph(glycan)
+    node_labels = nx.get_node_attributes(graph, 'string_labels')
+    labels = {i:node_labels[i] for i in range(len(graph.nodes))}
+    graph = nx.relabel_nodes(graph, labels)
+    graph = nx.drawing.nx_pydot.to_pydot(graph)
+    graph.write_png("graph.png")
     # write_dot(graph, "graph.dot")
     # graph=pgv.AGraph("graph.dot")  
     # graph.layout(prog='dot')