Update app.py

app.py

@@ -205,6 +205,8 @@ def flag_if_shared(flag_method):
 def process_and_output_files(input_files):
     data = []
 
+    total_amount = 0
+    
     for file in input_files:
         # Extract and categorize text for each file
         text = extract_text(file)
@@ -232,13 +234,24 @@ def process_and_output_files(input_files):
             "Amount(Rs.)": response_dict.get("total")
         }
 
+        total_amount+=response_dict.get("total")
+
         # Append the relevant data for this file to the data list
         data.append(extracted_data)
         
-    #data_dicts = [json.loads(item[0]) for item in data]
 
+    string_data = []
+    for item in data:
+        string_item = {key: str(value) for key, value in item.items()}
+        string_data.append(string_item)
+        
+    df = pd.DataFrame(string_data)
+    table_html = df.to_html(classes="table table-bordered", index=True)
 
-    return data
+    scrollable_table = f'<div style="overflow-x: auto;">{table_html}</div>'
+    return scrollable_table    
+
+    #return data
 
 
 
@@ -335,8 +348,9 @@ with gr.Blocks(title="Automatic Reimbursement Tool Demo") as page:
                         interactive=False,
                         height=240,
                     )
-                information = gr.JSON(label="Extracted information")
-
+                #information = gr.JSON(label="Extracted information")
+                information = gr.HTML()
+                
                 with gr.Row():
                     flag_incorrect_button = gr.Button(
                         "Flag as incorrect", variant="stop", interactive=True