Update app.py

app.py

@@ -1,4 +1,3 @@
-###ORDER###
 import gradio as gr
 import pandas as pd
 from datetime import datetime
@@ -47,6 +46,25 @@ def order_csv(file, supplier, order_number):
 
     # Save filtered DataFrame as Excel file
     download_df.to_excel(output_file, index=False)
+
+    # Adjust column widths using openpyxl
+    from openpyxl import load_workbook
+    wb = load_workbook(output_file)
+    ws = wb.active
+
+    for column_cells in ws.columns:
+        max_length = 0
+        # Get the column letter (e.g. 'A', 'B', etc.)
+        column_letter = column_cells[0].column_letter
+        for cell in column_cells:
+            if cell.value is not None:
+                cell_length = len(str(cell.value))
+                if cell_length > max_length:
+                    max_length = cell_length
+        adjusted_width = max_length + 2  # Add a little extra space
+        ws.column_dimensions[column_letter].width = adjusted_width
+
+    wb.save(output_file)
     return df, output_file
 
 # Define the main block for the interface
@@ -77,4 +95,4 @@ with gr.Blocks() as demo:
     )
 
 # Run the interface
-demo.launch()
+demo.launch()