Update app.py

app.py

@@ -5,7 +5,7 @@ import networkx as nx
 
 
 def render_dependency_tree(words, parents, labels):
-    fig, ax = plt.subplots(figsize=(32, 16))
+    fig, ax = plt.subplots(figsize=(32, 24))
 
     # Create a directed graph
     G = nx.DiGraph()
@@ -24,11 +24,12 @@ def render_dependency_tree(words, parents, labels):
 
     # Draw the graph
     nx.draw(G, pos, ax=ax, with_labels=True, labels=nx.get_node_attributes(G, 'label'), 
-            arrows=True, node_color='white', node_size=3000)
+            arrows=True, node_color='#eeeeee', node_size=1000, node_shape='o', font_size=10
+    )
 
     # Draw edge labels
     edge_labels = nx.get_edge_attributes(G, 'label')
-    nx.draw_networkx_edge_labels(G, pos, ax=ax, edge_labels=edge_labels, rotate=False)
+    nx.draw_networkx_edge_labels(G, pos, ax=ax, edge_labels=edge_labels, rotate=False, alpha=0.9, font_size=8)
 
     return fig