Update app.py

app.py

@@ -1,13 +1,13 @@
 import pandas as pd
 import gradio as gr
+import uuid
+import os
 
 # 7-Day Schedule Converter with explicit iteration and header fill
 
 def convert_schedule(file_path, direction):
     # 1. Load Excel sheet into DataFrame
-    # Read without header to inspect first two rows for header fill
     raw = pd.read_excel(file_path, header=None)
-    # Extract potential header rows (skip first column)
     header1 = raw.iloc[0,1:].astype(object)
     header2 = raw.iloc[1,1:].astype(object)
 
@@ -16,7 +16,6 @@ def convert_schedule(file_path, direction):
         days = header2.tolist()
         data_start = 2
     else:
-        # Forward-fill merged first-row headers
         days = []
         last = None
         for val in header1:
@@ -36,13 +35,12 @@ def convert_schedule(file_path, direction):
     )
     df.columns = [str(day) for day in days]
 
-    # 2. Ensure we have exactly the original 7 day columns
+    # 2. Ensure we have exactly the original day columns
     day_cols = list(df.columns)
 
-    # 3. Build mapping: for each Model (or Texter) assign days
+    # 3. Build mapping of assignments
     assignments = {}
     if direction == 'A to B':
-        # Rows are Models → output rows are Texters
         for model in df.index.astype(str):
             for day in day_cols:
                 cell = df.at[model, day]
@@ -52,7 +50,6 @@ def convert_schedule(file_path, direction):
                         continue
                     assignments.setdefault(texter, {d: [] for d in day_cols})
                     assignments[texter][day].append(model)
-        # Build output DataFrame
         index = sorted(assignments.keys())
         result = pd.DataFrame(index=index, columns=day_cols)
         for texter, days_map in assignments.items():
@@ -60,7 +57,6 @@ def convert_schedule(file_path, direction):
                 models = days_map.get(day, [])
                 result.at[texter, day] = ', '.join(models) if models else 'OFF'
     else:
-        # Rows are Texters → output rows are Models
         for texter in df.index.astype(str):
             for day in day_cols:
                 cell = df.at[texter, day]
@@ -80,7 +76,12 @@ def convert_schedule(file_path, direction):
     # 4. Cleanup axis names
     result.index.name = None
     result.columns.name = None
-    return result
+
+    # 5. Save to Excel for download
+    output_filename = f"converted_{uuid.uuid4().hex}.xlsx"
+    output_path = os.path.join(os.getcwd(), output_filename)
+    result.to_excel(output_path)
+    return result, output_path
 
 # Gradio UI
 def main():
@@ -90,10 +91,13 @@ def main():
             gr.File(label='Upload Weekly Schedule (.xlsx)', file_count='single', type='filepath'),
             gr.Radio(['A to B', 'B to A'], label='Convert Direction')
         ],
-        outputs=gr.Dataframe(label='Converted Schedule'),
+        outputs=[
+            gr.Dataframe(label='Converted Schedule'),
+            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. Uses explicit iteration.'),
+        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)