Update app.py

app.py

@@ -1,59 +1,59 @@
 import streamlit as st
-from nltk import pos_tag, word_tokenize, download
-from nltk.tree import Tree
-from nltk.chunk.regexp import RegexpParser
-import plotly.graph_objects as go
-
-# Ensure NLTK resources are downloaded
-download('punkt')
-download('averaged_perceptron_tagger')
-
-# Function to create a syntax tree
-def create_syntax_tree(sentence):
-    tokens = word_tokenize(sentence)
-    tagged = pos_tag(tokens)
-    grammar = "NP: {<DT>?<JJ>*<NN>}"
-    cp = RegexpParser(grammar)
-    chunked = cp.parse(tagged)
-    return chunked
-
-# Function to manually plot a syntax tree using Plotly
-def plot_tree(tree):
-    trace = go.Scatter(x=[], y=[], text=[], mode="text+markers", textposition="bottom center")
-    
-    def add_node(node, pos=0, level=1):
-        x = pos
-        y = -level
-        trace['x'] += (x,)
-        trace['y'] += (y,)
-        trace['text'] += (node.label(),)
-        
-        child_pos = x - 0.5 * len(node) + 0.5
-        for child in node:
-            if isinstance(child, Tree):
-                trace['x'] += (x, child_pos)
-                trace['y'] += (y, -(level + 1))
-                add_node(child, pos=child_pos, level=level + 1)
-            else:
-                trace['x'] += (child_pos,)
-                trace['y'] += (-(level + 1),)
-                trace['text'] += (child,)
-            child_pos += 1
-    
-    add_node(tree)
-    
-    layout = go.Layout(title="Syntax Tree", xaxis={'showgrid': False, 'zeroline': False, 'showticklabels': False},
-                       yaxis={'showgrid': False, 'zeroline': False, 'showticklabels': False}, showlegend=False)
-    fig = go.Figure(data=[trace], layout=layout)
-    return fig
-
-st.title('Syntax Tree Generator')
-sentence = st.text_input('Enter a sentence:', '')
+import spacy
+from spacy import displacy
+from anytree import Node, RenderTree
+from anytree.exporter import DotExporter
+import subprocess
+from typing import Tuple
+import os
+import base64
+
+# Initialize spaCy
+nlp = spacy.load("en_core_web_md")
+
+def postag(tk) -> str:
+    tag = ""
+    plural_tags = ["NNS", "NNPS"]
+    if tk.tag_ in plural_tags:
+        tag = f" ({tk.tag_}) (Plural)"
+    else:
+        tag = f" ({tk.tag_})"
+    return tag
+
+def gen_syntactic_tree(expr: str) -> Tuple[str, str]:
+    doc = nlp(expr)
+    root = next(doc.sents).root
+    node = Node("" + root.text + ": (Root)" + postag(root), parent=None)
+
+    def tree(tk, last_node: Node, depth: int) -> None:
+        if tk.n_lefts + tk.n_rights > 0:
+            for child in tk.children:
+                tree(child, Node("" + child.text + ": " + str(depth + 1) + postag(child), parent=last_node), depth + 1)
+
+    tree(root, node, 0)
+    syntactic_str_tree = ""
+
+    for pre, fill, node in RenderTree(node):
+        syntactic_str_tree += f"{pre}{node.name}\n"
+
+    syntactic_tree = displacy.render(doc, style='dep', options={'distance': 100})
+    syntactic_html_tree = f"""
+        <center>
+            <div style='max-width: 800px; overflow-x:auto;'>
+                {syntactic_tree}
+            </div>
+        </center>
+    """
+    return syntactic_html_tree, syntactic_str_tree
+
+st.title('Syntactic Tree Generator')
+sentence = st.text_area("Enter a sentence:")
 
 if sentence:
-    tree = create_syntax_tree(sentence)
-    try:
-        fig = plot_tree(tree)
-        st.plotly_chart(fig, use_container_width=True)
-    except Exception as e:
-        st.error(f"An error occurred while generating the tree diagram: {e}")
+    error, syntactic_html_tree, syntactic_str_tree = "", "", ""
+    if sentence.strip() == "":
+        st.warning("The sentence cannot be empty!")
+    else:
+        syntactic_html_tree, syntactic_str_tree = gen_syntactic_tree(sentence)
+        st.markdown(syntactic_html_tree, unsafe_allow_html=True)
+        st.text(syntactic_str_tree)