Update app.py

app.py

@@ -11,7 +11,7 @@ parser = Parser()
 def parse(text):
     output = parser.parse(text)
 
-    dependency_tree = render_dependency_tree(output["forms"], output["heads"], output["deprels"])
+    dependency_tree = render_dependency_tree(output["forms"], output["heads"], output["deprel"])
     table = render_table(output["forms"], output["lemmas"], output["upos"], output["xpos"], output["ne"])
 
     return dependency_tree, table
@@ -135,16 +135,16 @@ def render_table(forms, lemmas, upos, xpos, feats, named_entities):
     feats = [feat + [""] * (max_len - len(feat)) for feat in feats]
     feats = list(zip(*feats))
 
-    named_entities = []
+    named_entities_converted = []
     for i, ne in enumerate(named_entities):
         if ne == "O":
-            named_entities.append("")
-        if ne.startswith("B"):
-            named_entities.append(f"<< {ne.split('-')[1]}")
+            named_entities_converted.append("")
+        elif ne.startswith("B"):
+            named_entities_converted.append(f"<< {ne.split('-')[1]}")
         elif ne.startswith("I") and i + 1 < len(named_entities) and named_entities[i + 1].startswith("I"):
-            named_entities.append(ne.split('-')[1])
+            named_entities_converted.append(ne.split('-')[1])
         else:
-            named_entities.append(f"{ne.split('-')[1]} >>")
+            named_entities_converted.append(f"{ne.split('-')[1]} >>")
 
     array = [
         [""] + forms,
@@ -153,7 +153,7 @@ def render_table(forms, lemmas, upos, xpos, feats, named_entities):
         ["*XPOS:*"] + xpos,
         ["*UFEATS:*"] + list(feats[0]),
         *([""] + list(row) for row in feats[1:]),
-        ["*NE:*"] + named_entities,
+        ["*NE:*"] + named_entities_converted,
     ]
 
     #return tabulate.tabulate(array, headers="firstrow", tablefmt="unsafehtml")