Update app.py

app.py

@@ -1,21 +1,23 @@
 import pandas as pd
+import openpyxl  # ensure XLSX engine is available
 import gradio as gr
 import uuid
 import os
 
-# 7-Day Schedule Converter with explicit iteration and header fill
+# 7-Day Schedule Converter with explicit iteration and header fill and download support
 
 def convert_schedule(file_path, direction):
-    # 1. Load Excel sheet into DataFrame
+    # 1. Load Excel sheet without header to inspect real headers
     raw = pd.read_excel(file_path, header=None)
     header1 = raw.iloc[0,1:].astype(object)
     header2 = raw.iloc[1,1:].astype(object)
 
-    # Decide which header row to use: prefer second if fully populated
+    # Decide which header row to use: prefer fully populated second row
     if header2.notna().all() and not header2.str.startswith('Unnamed').any():
         days = header2.tolist()
         data_start = 2
     else:
+        # Forward-fill merged first-row headers
         days = []
         last = None
         for val in header1:
@@ -26,19 +28,19 @@ def convert_schedule(file_path, direction):
                 days.append(last)
         data_start = 1
 
-    # Load actual data with day columns
+    # 2. Load actual data with resolved day columns
     df = pd.read_excel(
         file_path,
         header=data_start,
         index_col=0,
-        usecols=[0] + list(range(1, len(days)+1))
+        usecols=[0] + list(range(1, len(days) + 1))
     )
     df.columns = [str(day) for day in days]
 
-    # 2. Ensure we have exactly the original day columns
+    # 3. Retain exactly the original day columns
     day_cols = list(df.columns)
 
-    # 3. Build mapping of assignments
+    # 4. Build assignment mapping via explicit iteration
     assignments = {}
     if direction == 'A to B':
         for model in df.index.astype(str):
@@ -50,6 +52,7 @@ def convert_schedule(file_path, direction):
                         continue
                     assignments.setdefault(texter, {d: [] for d in day_cols})
                     assignments[texter][day].append(model)
+        # Create result DataFrame
         index = sorted(assignments.keys())
         result = pd.DataFrame(index=index, columns=day_cols)
         for texter, days_map in assignments.items():
@@ -73,17 +76,19 @@ def convert_schedule(file_path, direction):
                 texters = days_map.get(day, [])
                 result.at[model, day] = ', '.join(texters) if texters else 'OFF'
 
-    # 4. Cleanup axis names
+    # 5. Cleanup axis names
     result.index.name = None
     result.columns.name = None
 
-    # 5. Save to Excel for download
+    # 6. Save to Excel for download, specifying engine
     output_filename = f"converted_{uuid.uuid4().hex}.xlsx"
     output_path = os.path.join(os.getcwd(), output_filename)
-    result.to_excel(output_path)
+    result.to_excel(output_path, engine='openpyxl', index=True)
+
+    # Return both DataFrame and download path
     return result, output_path
 
-# Gradio UI
+# Gradio UI definition
 def main():
     iface = gr.Interface(
         fn=convert_schedule,
@@ -96,8 +101,10 @@ def main():
             gr.File(label='Download Converted Excel')
         ],
         title='7-Day Schedule Converter',
-        description=('Upload a 7-column weekly schedule (Models vs Days) with merged or single headers, '
-                     'then flip between Models→Texters or Texters→Models. Download the result as .xlsx.'),
+        description=(
+            'Upload a 7-column weekly schedule (Models vs Days) with merged or single headers, '
+            'then flip between Models→Texters or Texters→Models. Download the result as .xlsx.'
+        ),
         allow_flagging='never'
     )
     iface.launch(server_name='0.0.0.0', server_port=7860)