prettytable test

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import prettytable
 
 
 description = """
@@ -78,52 +79,31 @@ edge_labels = [
     if line and not line.startswith("#")
 ]
 
+def render_table(forms, lemmas, upos, xpos, feats, metadata, edges, edge_labels):
+    feats = [feat.split("|") for feat in feats]
+    max_len = max([len(feat) for feat in feats])
+    feats = [feat + [""] * (max_len - len(feat)) for feat in feats]
+    feats = list(zip(*feats))
+
+    array = [
+        ["<b>forms</b>          "] + forms,
+        ["<b>lemmas</b>          "] + lemmas,
+        ["<b>upos</b>          "] + upos,
+        ["<b>xpos</b>          "] + xpos,
+        ["<b>feats</b>          "] + feats[0],
+        *([""] + row for row in feats[1:])
+    ]
+
+    table = prettytable.PrettyTable()
+    table.field_names = array[0]
+    for row in array[1:]:
+        table.add_row(row)
+    
+    return table.get_html_string()
 
 
 with gr.Blocks(theme='sudeepshouche/minimalist') as demo:
     gr.HTML(description)
-
-    gr.Dataframe([forms], interactive=False, type="array")
-
-    with gr.Accordion("Lemmas", open=True):
-        gr.Dataframe([lemmas], interactive=False, type="array")
-    
-    with gr.Accordion("UPOS", open=True):
-        gr.Dataframe([upos], interactive=False, type="array")
-    
-    with gr.Accordion("XPOS", open=True):
-        gr.Dataframe([xpos], interactive=False, type="array")
-
-        gr.HTML("""<table>
-  <thead>
-    <tr>
-      <th>City name</th>
-      <th>Area</th>
-      <th>Population</th>
-      <th>Annual Rainfall</th>
-    </tr>
-  </thead>
-  <tbody>
-    <tr>
-      <td>Adelaide</td>
-      <td>1295</td>
-      <td>1158259</td>
-      <td>600.5</td>
-    </tr>
-    <tr>
-      <td>Brisbane</td>
-      <td>5905</td>
-      <td>1857594</td>
-      <td>1146.4</td>
-    </tr>
-  </tbody>
-</table>""")
-    
-    with gr.Accordion("UFeats", open=True):
-        feats = [feat.split("|") for feat in feats]
-        max_len = max([len(feat) for feat in feats])
-        feats = [feat + [""] * (max_len - len(feat)) for feat in feats]
-        feats = list(zip(*feats))
-        gr.Dataframe(feats, interactive=False, type="array")
+    gr.HTML(render_table(forms, lemmas, upos, xpos, feats, metadata, edges, edge_labels))
 
 demo.launch()

requirements.txt

@@ -0,0 +1 @@
+prettytable