Update app.py

app.py

@@ -3,6 +3,8 @@ import pandas as pd
 from fuzzywuzzy import process, fuzz
 import tempfile
 import os
+from openpyxl import load_workbook
+from openpyxl.styles import Alignment
 
 def auto_correct_names(series, threshold=90):
     """Auto-correct typos in chatter names using fuzzy matching."""
@@ -21,6 +23,23 @@ def auto_correct_names(series, threshold=90):
     
     return series.replace(name_mapping)
 
+def adjust_excel_formatting(file_path):
+    """Adjust column widths and enable text wrapping in Excel file."""
+    wb = load_workbook(file_path)
+    ws = wb.active
+
+    # Adjust column widths and enable text wrapping
+    for col in ws.columns:
+        max_length = 0
+        col_letter = col[0].column_letter
+        for cell in col:
+            if cell.value:
+                max_length = max(max_length, len(str(cell.value)))
+            cell.alignment = Alignment(wrap_text=True)  # Enable text wrapping
+        ws.column_dimensions[col_letter].width = max_length + 2  # Adjust column width
+
+    wb.save(file_path)
+
 def process_file(input_file):
     """Process uploaded Excel file and return output"""
     try:
@@ -61,6 +80,10 @@ def process_file(input_file):
         # Create temp file for output
         temp_file_path = os.path.join(tempfile.gettempdir(), "processed_schedule.xlsx")
         final_df.to_excel(temp_file_path, index=False, sheet_name='Schedule')
+
+        # Adjust formatting to ensure everything fits within cells
+        adjust_excel_formatting(temp_file_path)
+
         return final_df, temp_file_path
 
     except Exception as e:
@@ -104,4 +127,3 @@ with gr.Blocks(title="Schedule Processor") as demo:
 if __name__ == "__main__":
     demo.launch()
 
-