Update app.py

app.py

@@ -1,8 +1,8 @@
 import pandas as pd
 import openpyxl
 import gradio as gr
-import tempfile
-import os
+import io
+import base64
 from datetime import datetime
 
 def convert_schedule(file_path, direction):
@@ -103,73 +103,93 @@ 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. Create downloadable file with proper path handling
+        # 6. Create downloadable file using in-memory approach
         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)
 
-        # Create download file in a way that works with Gradio
-        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-        output_filename = f"converted_schedule_{timestamp}.xlsx"
+        # Create Excel file in memory
+        download_df = result_clean.reset_index().rename(columns={'index': first_col_name})
         
-        # Use tempfile.NamedTemporaryFile but keep it open
-        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.xlsx', prefix='schedule_')
-        temp_path = temp_file.name
-        temp_file.close()
+        # Use BytesIO to create file in memory
+        excel_buffer = io.BytesIO()
         
-        # Save to the temporary file
-        download_df = result_clean.reset_index().rename(columns={'index': first_col_name})
-        download_df.to_excel(temp_path, sheet_name='Converted Schedule', index=False, engine='openpyxl')
-
-        return display_df, temp_path
+        with pd.ExcelWriter(excel_buffer, engine='openpyxl') as writer:
+            download_df.to_excel(writer, sheet_name='Converted Schedule', index=False)
+        
+        excel_buffer.seek(0)
+        
+        # Create filename with timestamp
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        filename = f"converted_schedule_{timestamp}.xlsx"
+        
+        # Return the buffer content for download
+        return display_df, (excel_buffer.getvalue(), filename)
 
     except Exception as e:
         error_msg = f"Error processing file: {str(e)}"
-        print(f"DEBUG: {error_msg}")  # For debugging
+        print(f"DEBUG: {error_msg}")
         error_df = pd.DataFrame({'Error': [error_msg]})
         return error_df, None
 
-# Create the interface with better error handling
-def create_interface():
-    iface = gr.Interface(
-        fn=convert_schedule,
-        inputs=[
-            gr.File(
-                label='Upload Weekly Schedule (.xlsx)', 
-                file_count='single', 
-                file_types=['.xlsx', '.xls']
-            ),
-            gr.Radio(
-                ['A to B', 'B to A'], 
-                label='Convert Direction', 
-                value='A to B',
-                info='A to B: Models→Texters, B to A: Texters→Models'
-            )
-        ],
-        outputs=[
-            gr.Dataframe(label='Converted Schedule (Preview)', wrap=True),
-            gr.File(label='Download Converted Schedule')
-        ],
-        title='🔄 7-Day Schedule Converter',
-        description=(
-            '**How to use:**\n'
-            '1. Upload your Excel file with a 7-day schedule\n'
-            '2. Choose conversion direction\n'
-            '3. Preview the result and download the converted file\n\n'
-            '*Supports merged headers and handles Models ↔ Texters conversion*'
+# Wrapper function to handle the file download properly
+def process_and_download(file_path, direction):
+    display_result, download_data = convert_schedule(file_path, direction)
+    
+    if download_data is None:
+        return display_result, None
+    
+    # Create a temporary file that Gradio can serve
+    import tempfile
+    import os
+    
+    excel_content, filename = download_data
+    
+    # Save to a temporary file in the system temp directory
+    temp_dir = tempfile.gettempdir()
+    temp_path = os.path.join(temp_dir, filename)
+    
+    with open(temp_path, 'wb') as f:
+        f.write(excel_content)
+    
+    return display_result, temp_path
+
+# Create the interface
+iface = gr.Interface(
+    fn=process_and_download,
+    inputs=[
+        gr.File(
+            label='Upload Weekly Schedule (.xlsx)', 
+            file_count='single', 
+            file_types=['.xlsx', '.xls']
         ),
-        flagging_mode='never',
-        allow_flagging='never'
-    )
-    return iface
+        gr.Radio(
+            ['A to B', 'B to A'], 
+            label='Convert Direction', 
+            value='A to B',
+            info='A to B: Models→Texters, B to A: Texters→Models'
+        )
+    ],
+    outputs=[
+        gr.Dataframe(label='Converted Schedule (Preview)', wrap=True),
+        gr.File(label='Download Converted Schedule (.xlsx)')
+    ],
+    title='🔄 7-Day Schedule Converter',
+    description=(
+        '**How to use:**\n'
+        '1. Upload your Excel file with a 7-day schedule\n'
+        '2. Choose conversion direction\n'
+        '3. Preview the result and download the converted file\n\n'
+        '*Supports merged headers and handles Models ↔ Texters conversion*'
+    ),
+    flagging_mode='never'
+)
 
 if __name__ == "__main__":
-    iface = create_interface()
     iface.launch(
         server_name='0.0.0.0', 
         server_port=7860,
-        share=False,
-        debug=True
+        share=False
     )