add dependency tree plot

app.py

@@ -1,5 +1,36 @@
 import gradio as gr
 import tabulate
+import matplotlib.pyplot as plt
+import networkx as nx
+
+
+def render_dependency_tree(words, parents, labels):
+    fig, ax = plt.subplots(figsize=(10, 6))
+
+    # Create a directed graph
+    G = nx.DiGraph()
+
+    # Adding nodes to the graph
+    for i, word in enumerate(words):
+        G.add_node(i, label=word)
+
+    # Adding edges with labels
+    for i, (parent, label) in enumerate(zip(parents, labels)):
+        if parent != -1:
+            G.add_edge(parent, i, label=label)
+
+    # Position nodes using Graphviz
+    pos = nx.nx_agraph.graphviz_layout(G, prog='dot')
+
+    # 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)
+
+    # 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)
+
+    return fig
 
 
 description = """
@@ -108,5 +139,6 @@ custom_css = \
 with gr.Blocks(theme='sudeepshouche/minimalist', css=custom_css) as demo:
     gr.HTML(description)
     gr.DataFrame(**render_table(forms, lemmas, upos, xpos, feats, metadata, edges, edge_labels), interactive=False, datatype="markdown")
+    gr.Plot(render_dependency_tree(forms, edges, edge_labels), interactive=False)
 
 demo.launch()

requirements.txt

@@ -1 +1,4 @@
-tabulate
+tabulate
+matplotlib
+networkx
+pygraphviz