Update app.py

app.py

@@ -19,6 +19,10 @@ def parse(text):
 def render_dependency_tree(words, parents, labels):
     fig, ax = plt.subplots(figsize=(40, 16))
 
+    main_font_size = 40 if len(words) < 10 else 30 if len(words) < 20 else 24 if len(words) < 40 else 16
+    minor_font_size = 30 if len(words) < 10 else 22 if len(words) < 20 else 16 if len(words) < 40 else 12
+    pad = main_font_size // 2
+
     # Create a directed graph
     G = nx.DiGraph()
 
@@ -36,12 +40,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='#ffffff', node_size=0, node_shape='s', font_size=30, bbox = dict(facecolor="white", pad=14)
+            arrows=True, node_color='#ffffff', node_size=0, node_shape='s', font_size=main_font_size, bbox = dict(facecolor="white", pad=pad)
     )
 
     # 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, alpha=0.9, font_size=22)
+    nx.draw_networkx_edge_labels(G, pos, ax=ax, edge_labels=edge_labels, rotate=False, alpha=1.0, font_size=minor_font_size)
 
     return fig