Update app.py

app.py

@@ -267,10 +267,13 @@ def process_and_output_files(input_files):
         
     df = pd.DataFrame(string_data)
     
-    table_html = df.to_html(classes="table table-bordered", index=False)
+    filename = "output.xlsx"
+    df.to_excel(filename, index=False)
     
+    table_html = df.to_html(classes="table table-bordered", index=True)
     scrollable_table = f'<div style="overflow-x: auto;">{table_html}</div>'
-    return scrollable_table    
+    
+    return scrollable_table, filename   
 
     #return data
 
@@ -370,7 +373,9 @@ with gr.Blocks(title="Automatic Reimbursement Tool Demo") as page:
                         height=240,
                     )
                 #information = gr.JSON(label="Extracted information")
-                information = gr.HTML()
+                table_display = gr.HTML(label="Table Display")
+                excel_download = gr.File(label="Download Excel", type="file")
+
                 
                 with gr.Row():
                     flag_incorrect_button = gr.Button(
@@ -400,11 +405,11 @@ with gr.Blocks(title="Automatic Reimbursement Tool Demo") as page:
             submit_button.click(
                 clear_outputs,
                 [input_file],
-                [extracted_text, category, chatbot, information],
+                [extracted_text, category, chatbot, table_display],
             ).then(
                 process_and_output_files,
                 [input_file],
-                [information],
+                [table_display, excel_download],  # Adding excel_download here
             ).then(
                 flag_if_shared(flag_method),
                 [