Update app.py

app.py

@@ -28,45 +28,37 @@ def adjust_excel_formatting(file_path):
     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
+            cell.alignment = Alignment(wrap_text=True)
+        ws.column_dimensions[col_letter].width = max_length + 2
 
     wb.save(file_path)
 
 def process_file(input_file):
     """Process uploaded Excel file and return output"""
     try:
-        # Read input file
         input_df = pd.read_excel(input_file, header=1)
-        
-        # Store original date order
         date_columns = input_df.columns[1:].tolist()
         
-        # Melt to long format
         df_long = input_df.melt(
             id_vars=[input_df.columns[0]], 
             var_name='DATE', 
             value_name='CHATTER'
         )
         
-        # Force date order
         df_long['DATE'] = pd.Categorical(
             df_long['DATE'], 
             categories=date_columns,
             ordered=True
         )
         
-        # Clean names
         df_long['CHATTER'] = auto_correct_names(df_long['CHATTER'])
         
-        # Group and pivot
         grouped = df_long.groupby(['CHATTER', 'DATE'], observed=True)[input_df.columns[0]] \
             .apply(lambda x: ', '.join(sorted(x))).reset_index()
         
@@ -74,14 +66,11 @@ def process_file(input_file):
         chatter_order = grouped['CHATTER'].value_counts().index.tolist()
         final_df = pivoted.reindex(chatter_order)[date_columns].fillna("OFF")
         
-        # Reset index to show chatter names in preview
         final_df = final_df.reset_index()
         
-        # 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
@@ -90,6 +79,11 @@ def process_file(input_file):
         error_df = pd.DataFrame({"Error": [f"⚠️ {str(e)}"]})
         return error_df, None
 
+# Add the missing download_file function
+def download_file(out_path):
+    """Return the processed file path for download"""
+    return out_path
+
 # Gradio interface
 with gr.Blocks(title="Schedule Processor") as demo:
     gr.Markdown("# 📅 Schedule Processor")
@@ -121,7 +115,6 @@ with gr.Blocks(title="Schedule Processor") as demo:
             return df, out_path, gr.update(visible=True)
         return df, None, gr.update(visible=False)
 
-    # Event handlers
     process_btn.click(
         process_and_show,
         inputs=input_file,