Update app.py

app.py

@@ -55,6 +55,7 @@ def convert_schedule(file_path, direction):
                     assignments[texter][day].append(model)
         index = sorted(assignments.keys())
         result = pd.DataFrame(index=index, columns=day_cols)
+        first_col_name = 'Texter'
         for texter, days_map in assignments.items():
             for day in day_cols:
                 models = days_map.get(day, [])
@@ -72,45 +73,43 @@ def convert_schedule(file_path, direction):
                     assignments[model][day].append(texter)
         index = sorted(assignments.keys())
         result = pd.DataFrame(index=index, columns=day_cols)
+        first_col_name = 'Model'
         for model, days_map in assignments.items():
             for day in day_cols:
                 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 in the current working directory
+    # For display, include index as a column
+    display_df = result.reset_index().rename(columns={'index': first_col_name})
+
+    # 6. Save to Excel for download (relative path so Gradio can serve it)
     output_filename = f"converted_{uuid.uuid4().hex}.xlsx"
-    output_path = os.path.join(os.getcwd(), output_filename)
-    result.to_excel(output_path, engine='openpyxl', index=True)
+    result.to_excel(output_filename, 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,
-        inputs=[
-            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'),
-            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.'
-        ),
-        allow_flagging='never'
-    )
-    iface.launch(server_name='0.0.0.0', server_port=7860)
+    return display_df, output_filename
 
-# Call main to start the app
-if __name__ == '__main__':
-    main()
+# Build Gradio interface and launch immediately for Hugging Face Spaces
+iface = gr.Interface(
+    fn=convert_schedule,
+    inputs=[
+        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'),
+        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.'
+    ),
+    flagging_mode='never'
+)
+# Launch on 0.0.0.0:7860 for Spaces
+iface.launch(server_name='0.0.0.0', server_port=7860)