fix

app.py

@@ -7,19 +7,20 @@ nlp = spacy.load("en_core_web_md")
 
 # -- Main functions --
 
-def markTk(tk):
-    noun_tags = ['NN', 'NNS', 'WP']
-    if tk.tag_ in noun_tags:
-        return " (Plural)" 
-    return ""
-    
+
 def postag(tk):
-    return f" ({tk.tag_})"
+    tag = ""
+    plural_tags = ["NNS", "NNPS"]
+    if tk.tag_ in plural_tags:
+        tag = " (Plural)"
+    else:
+        tag = " ({})".format(tk.tag_)
+    return tag
     
 def genFlatDepthTree(expr):
     doc = nlp(expr)
     root = next(doc.sents).root
-    node = Node(root.text+": 0 (Root)"+markTk(root), parent=None)
+    node = Node(root.text+": 0 (Root)"+postag(root), parent=None)
 
     def tree(tk, last_node, depth):
         if tk.n_lefts + tk.n_rights > 0:
@@ -36,22 +37,22 @@ def genFlatDepthTree(expr):
     
     return "<center><div style='max-width: 800px; overflow-x:auto;'>"+img_tree+"</div></center>", flat_tree
     
-def syntacticTree(expr):
-    doc = nlp(expr)
-    depths = {}
+# def syntacticTree(expr):
+#     doc = nlp(expr)
+#     depths = {}
 
-    def walk_tree(tk, depth):
-        depths[tk.text] = depth
-        if tk.n_lefts + tk.n_rights > 0: # if Doesn't root
-            [walk_tree(child, depth + 1) for child in tk.children]
+#     def walk_tree(tk, depth):
+#         depths[tk.text] = depth
+#         if tk.n_lefts + tk.n_rights > 0: # if Doesn't root
+#             [walk_tree(child, depth + 1) for child in tk.children]
 
-    [walk_tree(sent.root, 0) for sent in doc.sents]
+#     [walk_tree(sent.root, 0) for sent in doc.sents]
 
-    flat_tree = [tk.text+" - "+str(depths[tk.text]) for tk in doc]
-    image_tree = displacy.render(doc, style='dep', options={'distance': 100})
-    image_tree = "<center><div style='max-width: 800px; overflow-x:auto;'>"+image_tree+"</div></center>"
+#     flat_tree = [tk.text+" - "+str(depths[tk.text]) for tk in doc]
+#     image_tree = displacy.render(doc, style='dep', options={'distance': 100})
+#     image_tree = "<center><div style='max-width: 800px; overflow-x:auto;'>"+image_tree+"</div></center>"
     
-    return image_tree,flat_tree
+#     return image_tree,flat_tree
 
 # -- Interface --
 demo = gr.Blocks(css=".container { max-width: 900px; margin: auto;}", title="Syntactic tree generator")