Change JSON to DataFrame

app.py

@@ -5,6 +5,7 @@ from io import BytesIO
 from pathlib import Path
 
 import gradio as gr
+import pandas as pd
 from langchain.schema.output_parser import OutputParserException
 from PIL import Image
 
@@ -141,7 +142,9 @@ def parse(category, chatbot):
     chain = categories.category_modules[category].chain
     yield {"status": "Parsing response..."}
     try:
-        information = chain.output_parser.parse(answer)
+        information = chain.output_parser.parse(answer) or {
+            "Error": "No output"
+        }
         information = information.json() if information else {}
     except OutputParserException as e:
         information = {
@@ -149,7 +152,7 @@ def parse(category, chatbot):
             "details": str(e),
             "output": e.llm_output,
         }
-    yield information
+    yield pd.json_normalize(information)
 
 
 def activate_flags():
@@ -260,7 +263,14 @@ with gr.Blocks(title="Automatic Reimbursement Tool Demo") as page:
                         interactive=False,
                         height=240,
                     )
-                information = gr.JSON(label="Extracted information")
+                information = gr.Dataframe(
+                    pd.json_normalize({
+                        "Waiting for input...": ''
+                    }),
+                    label="Extracted information",
+                    interactive=False,
+                    row_count = (1, "fixed"),  # 1 till multiple inputs/outputs is implemented
+                )
                 with gr.Row():
                     flag_incorrect_button = gr.Button(
                         "Flag as incorrect", variant="stop", interactive=True