Update app.py

app.py

@@ -2,8 +2,7 @@ import pandas as pd
 import openpyxl
 import gradio as gr
 import io
-import base64
-from datetime import datetime
+import os
 
 def convert_schedule(file_path, direction):
     try:
@@ -100,39 +99,76 @@ def convert_schedule(file_path, direction):
         # For display, include index as a column
         display_df = result.reset_index().rename(columns={'index': first_col_name})
 
-        # 6. Return the DataFrame for download - let Gradio handle the file creation
+        # 6. Create CSV content as string for download
         result_clean = result.copy().fillna('OFF')
         
         # Ensure all values are strings
         for col in result_clean.columns:
             result_clean[col] = result_clean[col].astype(str)
 
-        # Return the cleaned DataFrame directly - Gradio will convert it
-        return display_df, result_clean
+        # Convert to CSV string
+        csv_content = result_clean.to_csv(index=True)
+        
+        return display_df, csv_content
 
     except Exception as e:
         error_df = pd.DataFrame({'Error': [f"Error processing file: {str(e)}"]})
-        return error_df, None
+        return error_df, "Error occurred during conversion"
+
+def download_csv(csv_content):
+    """Helper function to create downloadable CSV"""
+    if csv_content and csv_content != "Error occurred during conversion":
+        # Create a temporary file
+        filename = "converted_schedule.csv"
+        with open(filename, 'w', newline='', encoding='utf-8') as f:
+            f.write(csv_content)
+        return filename
+    return None
 
-# Gradio interface with different output types
-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', value='A to B')
-    ],
-    outputs=[
-        gr.Dataframe(label='Converted Schedule (Preview)'),
-        gr.Dataframe(label='Download Data (Right-click to copy/export)', type='pandas')
-    ],
-    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. '
-        'Use the second table to copy the data or export it manually.'
-    ),
-    flagging_mode='never'
-)
+# Create the interface with blocks for better control
+with gr.Blocks(title="7-Day Schedule Converter") as iface:
+    gr.Markdown("# 7-Day Schedule Converter")
+    gr.Markdown("Upload a 7-column weekly schedule (Models vs Days) with merged or single headers, then flip between Models→Texters or Texters→Models.")
+    
+    with gr.Row():
+        with gr.Column():
+            file_input = gr.File(label='Upload Weekly Schedule (.xlsx)', file_count='single', type='filepath')
+            direction_input = gr.Radio(['A to B', 'B to A'], label='Convert Direction', value='A to B')
+            convert_btn = gr.Button("Convert Schedule", variant="primary")
+        
+        with gr.Column():
+            output_df = gr.Dataframe(label='Converted Schedule')
+            download_btn = gr.Button("Download as CSV", variant="secondary")
+            download_file = gr.File(label="Download", visible=False)
+    
+    # Store CSV content in state
+    csv_state = gr.State()
+    
+    # Convert button click
+    def convert_and_store(file_path, direction):
+        display_df, csv_content = convert_schedule(file_path, direction)
+        return display_df, csv_content
+    
+    convert_btn.click(
+        fn=convert_and_store,
+        inputs=[file_input, direction_input],
+        outputs=[output_df, csv_state]
+    )
+    
+    # Download button click
+    def create_download(csv_content):
+        if csv_content and csv_content != "Error occurred during conversion":
+            filename = "converted_schedule.csv"
+            with open(filename, 'w', newline='', encoding='utf-8') as f:
+                f.write(csv_content)
+            return gr.File(value=filename, visible=True)
+        return gr.File(visible=False)
+    
+    download_btn.click(
+        fn=create_download,
+        inputs=[csv_state],
+        outputs=[download_file]
+    )
 
 if __name__ == "__main__":
     iface.launch(server_name='0.0.0.0', server_port=7860)