Update app.py

app.py

@@ -7,21 +7,27 @@ 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 genFlatDepthTree(expr):
     doc = nlp(expr)
     root = next(doc.sents).root
-    node = Node(root.text+" : depth="+str(0), parent=None)
+    node = Node(root.text+": 0 (Root)"+markTk(root), parent=None)
 
     def tree(tk, last_node, depth):
         if tk.n_lefts + tk.n_rights > 0:
             for ch in tk.children:
-                tree(ch, Node(ch.text+" : depth="+str(depth+1), parent=last_node), depth+1)
+                tree(ch, Node(ch.text+" :"+str(depth+1)+markTk(ch), parent=last_node), depth+1)
     
     tree(root, node, 0)
     flat_tree = ""
 
     for pre, fill, node in RenderTree(node):
-        flat_tree += "{}{}\n".format(pre, node.name)
+        flat_tree += """{}{}\n""".format(pre, node.name)
 
     img_tree = displacy.render(doc, style='dep', options={'distance': 100})