Update app.py

app.py

@@ -72,6 +72,7 @@ def display_file(input_files):
 
 
 
+
 def show_intermediate_outputs(show_intermediate):
     if show_intermediate:
         return gr.Accordion.update(visible=True)
@@ -202,6 +203,32 @@ def flag_if_shared(flag_method):
             return flag_method(request, *args, **kwargs)
     return proxy
 
+def save_df_to_excel_with_autowidth(df, filename):
+    # Save DataFrame to Excel without any formatting
+    df.to_excel(filename, index=False, engine='openpyxl')
+    
+    # Open the Excel file with openpyxl to adjust column widths
+    book = load_workbook(filename)
+    sheet = book.active
+    
+    # Loop through columns and adjust the width based on max length in each column
+    for column in sheet.columns:
+        max_length = 0
+        column = [cell for cell in column]  
+        for cell in column:
+            try:
+                if len(str(cell.value)) > max_length:
+                    max_length = len(cell.value)
+            except:
+                pass
+        adjusted_width = (max_length + 2)  # adding a little extra space
+        sheet.column_dimensions[get_column_letter(column[0].column)].width = adjusted_width
+
+    # Save the changes back to the Excel file
+    book.save(filename)
+    
+
+
 def process_and_output_files(input_files):
     data = []
 
@@ -268,7 +295,7 @@ def process_and_output_files(input_files):
     df = pd.DataFrame(string_data)
     
     filename = "output.xlsx"
-    df.to_excel(filename, index=False)
+    save_df_to_excel_with_autowidth(df, filename)
     
     table_html = df.to_html(classes="table table-bordered", index=True)
     scrollable_table = f'<div style="overflow-x: auto;">{table_html}</div>'
@@ -279,6 +306,8 @@ def process_and_output_files(input_files):
 
 
 
+
+
 with gr.Blocks(title="Automatic Reimbursement Tool Demo") as page:
     gr.Markdown("<center><h1>Automatic Reimbursement Tool Demo</h1></center>")
     gr.Markdown("<h2>Description</h2>")