Update app.py

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-import tabulate
 import matplotlib.pyplot as plt
 import networkx as nx
 
@@ -13,7 +12,6 @@ def parse(text):
 
     dependency_tree = render_dependency_tree(output["forms"], output["heads"], output["deprel"])
     table = render_table(output["forms"], output["lemmas"], output["upos"], output["xpos"], output["feats"], output["ne"])
-    table = gr.DataFrame(**table, interactive=False, datatype="markdown")
 
     return dependency_tree, table
 
@@ -57,78 +55,6 @@ description = """
 </div>
 """
 
-text = """1	Forretten	forrett	NOUN	_	Definite=Def|Gender=Masc|Number=Sing	2	nsubj	_	name=O
-2	lyder	lyde	VERB	_	Mood=Ind|Tense=Pres|VerbForm=Fin	0	root	_	name=O
-3	navnet	navn	NOUN	_	Definite=Def|Gender=Neut|Number=Sing	5	nmod	_	name=O
-4	"	$"	PUNCT	_	_	5	punct	_	SpaceAfter=No|name=O
-5	Coquilles	Coquilles	PROPN	_	_	2	obj	_	name=B-PROD
-6	St.	St.	PROPN	_	_	5	flat:name	_	name=I-PROD
-7	Jacques	Jacques	PROPN	_	_	5	flat:name	_	name=I-PROD
-8	Prince	Prince	PROPN	_	_	5	flat:name	_	name=I-PROD
-9	de	de	X	_	_	5	flat:name	_	name=I-PROD
-10	Norvege	Norvege	PROPN	_	_	5	flat:name	_	SpaceAfter=No|name=I-PROD
-11	"	$"	PUNCT	_	_	5	punct	_	SpaceAfter=No|name=O
-12	,	$,	PUNCT	_	_	5	punct	_	name=O
-13	som	som	PRON	_	PronType=Rel	16	nsubj	_	name=O
-14	er	være	AUX	_	Mood=Ind|Tense=Pres|VerbForm=Fin	16	cop	_	name=O
-15	grillet	grille	ADJ	_	Definite=Ind|Gender=Neut|Number=Sing|VerbForm=Part	16	amod	_	name=O
-16	kamskjell	kamskjell	NOUN	_	Definite=Ind|Gender=Neut|Number=Sing	5	acl:relcl	_	name=O
-17	på	på	ADP	_	_	19	case	_	name=O
-18	norsk	norsk	ADJ	_	Definite=Ind|Degree=Pos|Number=Sing	19	amod	_	name=O
-19	spekeskinke	spekeskinke	NOUN	_	Definite=Ind|Gender=Fem|Number=Sing	16	nmod	_	name=O
-20	-	$-	PUNCT	_	_	16	punct	_	name=O
-21	med	med	ADP	_	_	22	case	_	name=O
-22	trøffelhonningvinaigrette	trøffelhonningvinaigrette	NOUN	_	Definite=Ind|Gender=Masc|Number=Sing	16	nmod	_	SpaceAfter=No|name=O
-23	,	$,	PUNCT	_	_	22	punct	_	name=O
-24	ruccolasalat	ruccolasalat	NOUN	_	Definite=Ind|Gender=Masc|Number=Sing	22	conj	_	name=O
-25	og	og	CCONJ	_	_	27	cc	_	name=O
-26	ristede	riste	ADJ	_	Number=Plur|VerbForm=Part	27	amod	_	name=O
-27	gresskarkjerner	gresskarkjerne	NOUN	_	Definite=Ind|Gender=Fem|Number=Plur	22	conj	_	SpaceAfter=No|name=O
-28	.	$.	PUNCT	_	_	2	punct	_	name=O"""
-
-forms = [
-    line.split("\t")[1]
-    for line in text.split("\n")
-    if line and not line.startswith("#")
-]
-lemmas = [
-    line.split("\t")[2]
-    for line in text.split("\n")
-    if line and not line.startswith("#")
-]
-upos = [
-    line.split("\t")[3]
-    for line in text.split("\n")
-    if line and not line.startswith("#")
-]
-xpos = [
-    line.split("\t")[4]
-    for line in text.split("\n")
-    if line and not line.startswith("#")
-]
-feats = [
-    line.split("\t")[5]
-    for line in text.split("\n")
-    if line and not line.startswith("#")
-]
-ne = [
-    line.split("\t")[9].split('name=')[-1]
-    for line in text.split("\n")
-    if line and not line.startswith("#")
-]
-edges = [
-    int(line.split("\t")[6])
-    for line in text.split("\n")
-    if line and not line.startswith("#")
-]
-
-edge_labels = [
-    line.split("\t")[7]
-    for line in text.split("\n")
-    if line and not line.startswith("#")
-]
-
-print(ne, flush=True)
 
 def render_table(forms, lemmas, upos, xpos, feats, named_entities):
     feats = [[f"*{f.split('=')[0]}:* {f.split('=')[1]}" for f in (feat.split("|")) if '=' in f] for feat in feats]
@@ -159,8 +85,7 @@ def render_table(forms, lemmas, upos, xpos, feats, named_entities):
         ["*NE:*"] + named_entities_converted,
     ]
 
-    #return tabulate.tabulate(array, headers="firstrow", tablefmt="unsafehtml")
-    return {"value": array[1:], "headers": array[0]}
+    return {"data": array[1:], "headers": array[0]}
 
 
 custom_css = \
@@ -193,8 +118,8 @@ with gr.Blocks(theme='sudeepshouche/minimalist', css=custom_css) as demo:
 
     with gr.Column(scale=1, variant="panel"):
         gr.Label("", show_label=False, container=False)
-        table = gr.DataFrame(**render_table(forms, lemmas, upos, xpos, feats, ne), interactive=False, datatype="markdown")
-        dependency_plot = gr.Plot(render_dependency_tree(forms, edges, edge_labels), container=False)
+        table = gr.DataFrame([["", "", "", ""] for _ in range(3)], headers=["", "", "", ""], interactive=False, datatype="markdown")
+        dependency_plot = gr.Plot(plt.subplots(figsize=(40, 16))[0], container=False)
 
     source.submit(
         fn=parse, inputs=[source], outputs=[dependency_plot, table], queue=True