Update app.py

app.py

@@ -7,12 +7,12 @@ import os
 # 7-Day Schedule Converter with explicit iteration and header fill and download support
 
 def convert_schedule(file_path, direction):
-    # 1. Load Excel sheet without header to inspect real headers
+    # 1. Load raw header rows to determine day labels
     raw = pd.read_excel(file_path, header=None)
-    header1 = raw.iloc[0,1:].astype(object)
-    header2 = raw.iloc[1,1:].astype(object)
+    header1 = raw.iloc[0, 1:].astype(object)
+    header2 = raw.iloc[1, 1:].astype(object)
 
-    # Decide which header row to use: prefer fully populated second row
+    # Decide which header row to use: prefer second if fully populated
     if header2.notna().all() and not header2.str.startswith('Unnamed').any():
         days = header2.tolist()
         data_start = 2
@@ -28,7 +28,7 @@ def convert_schedule(file_path, direction):
                 days.append(last)
         data_start = 1
 
-    # 2. Load actual data with resolved day columns
+    # 2. Load actual data using resolved day columns
     df = pd.read_excel(
         file_path,
         header=data_start,
@@ -37,12 +37,13 @@ def convert_schedule(file_path, direction):
     )
     df.columns = [str(day) for day in days]
 
-    # 3. Retain exactly the original day columns
+    # 3. Retain original day column order
     day_cols = list(df.columns)
 
     # 4. Build assignment mapping via explicit iteration
     assignments = {}
     if direction == 'A to B':
+        # Models in rows → Texters as rows
         for model in df.index.astype(str):
             for day in day_cols:
                 cell = df.at[model, day]
@@ -52,7 +53,6 @@ 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():
@@ -60,6 +60,7 @@ def convert_schedule(file_path, direction):
                 models = days_map.get(day, [])
                 result.at[texter, day] = ', '.join(models) if models else 'OFF'
     else:
+        # Texters in rows → Models as rows
         for texter in df.index.astype(str):
             for day in day_cols:
                 cell = df.at[texter, day]
@@ -76,24 +77,20 @@ def convert_schedule(file_path, direction):
                 texters = days_map.get(day, [])
                 result.at[model, day] = ', '.join(texters) if texters else 'OFF'
 
-        # 5. Cleanup axis names
+    # 5. Cleanup axis names
     result.index.name = None
     result.columns.name = None
 
-        # 6. Save to Excel for download, ensuring directory exists
+    # 6. Save to Excel for download
     os.makedirs('/mnt/data', exist_ok=True)
     output_filename = f"converted_{uuid.uuid4().hex}.xlsx"
-    # Save under /mnt/data so HF Spaces can serve it
-    output_path = os.path.join('/mnt/data', output_filename)
-    output_filename = f"converted_{uuid.uuid4().hex}.xlsx"
-    # Save under /mnt/data so HF Spaces can serve it
     output_path = os.path.join('/mnt/data', output_filename)
     result.to_excel(output_path, engine='openpyxl', index=True)
 
-    # Return both DataFrame and download path
     return result, output_path
 
 # Gradio UI definition
+
 def main():
     iface = gr.Interface(
         fn=convert_schedule,
@@ -115,4 +112,4 @@ def main():
     iface.launch(server_name='0.0.0.0', server_port=7860)
 
 if __name__ == '__main__':
-    main()
+    main()