Update app.py

app.py

@@ -1,11 +1,8 @@
 import pandas as pd
-import openpyxl  # ensure XLSX engine is available
+import openpyxl
 import gradio as gr
-import uuid
-import os
-import tempfile
-
-# 7-Day Schedule Converter with explicit iteration and header fill and download support
+import io
+import base64
 
 def convert_schedule(file_path, direction):
     try:
@@ -49,7 +46,6 @@ def convert_schedule(file_path, direction):
             for model in df.index.astype(str):
                 for day in day_cols:
                     cell = df.at[model, day]
-                    # Handle NaN values properly
                     if pd.isna(cell):
                         continue
                     for texter in str(cell).split(','):
@@ -59,7 +55,7 @@ def convert_schedule(file_path, direction):
                         assignments.setdefault(texter, {d: [] for d in day_cols})
                         assignments[texter][day].append(model)
             
-            if not assignments:  # Handle empty assignments
+            if not assignments:
                 result = pd.DataFrame(columns=day_cols)
                 first_col_name = 'Texter'
             else:
@@ -75,7 +71,6 @@ def convert_schedule(file_path, direction):
             for texter in df.index.astype(str):
                 for day in day_cols:
                     cell = df.at[texter, day]
-                    # Handle NaN values properly
                     if pd.isna(cell):
                         continue
                     for model in str(cell).split(','):
@@ -85,7 +80,7 @@ def convert_schedule(file_path, direction):
                         assignments.setdefault(model, {d: [] for d in day_cols})
                         assignments[model][day].append(texter)
             
-            if not assignments:  # Handle empty assignments
+            if not assignments:
                 result = pd.DataFrame(columns=day_cols)
                 first_col_name = 'Model'
             else:
@@ -104,49 +99,35 @@ 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. Save to Excel for download - use current directory for Gradio compatibility
-        output_filename = f"converted_{uuid.uuid4().hex}.xlsx"
-        
-        # Clean the DataFrame before saving
-        result_clean = result.copy()
-        
-        # Replace any problematic values
-        result_clean = result_clean.fillna('OFF')  # Fill NaN with 'OFF'
+        # 6. Create Excel file in memory
+        output_buffer = io.BytesIO()
+        result_clean = result.copy().fillna('OFF')
         
-        # Ensure all values are strings to avoid Excel compatibility issues
+        # Ensure all values are strings
         for col in result_clean.columns:
             result_clean[col] = result_clean[col].astype(str)
         
-        # Save with better error handling
-        try:
-            # Use simple to_excel with minimal options for better compatibility
-            result_clean.to_excel(output_filename, index=True, engine='openpyxl')
-            
-            # Return the filename directly for Gradio to handle
-            if os.path.exists(output_filename):
-                print(f"File created successfully: {output_filename}")
-            else:
-                raise FileNotFoundError("File creation failed")
-                
-        except Exception as save_error:
-            print(f"Excel save error: {save_error}")
-            # Return None if save fails
-            return display_df, None
+        # Write to buffer
+        result_clean.to_excel(output_buffer, index=True, engine='openpyxl')
+        output_buffer.seek(0)
+        
+        # Save buffer content to a temporary file that Gradio can serve
+        temp_filename = "converted_schedule.xlsx"
+        with open(temp_filename, 'wb') as f:
+            f.write(output_buffer.getvalue())
 
-        # Return both DataFrame and download path
-        return display_df, output_filename
+        return display_df, temp_filename
 
     except Exception as e:
-        # Return error information
         error_df = pd.DataFrame({'Error': [f"Error processing file: {str(e)}"]})
         return error_df, None
 
-# Build Gradio interface and launch immediately for Hugging Face Spaces
+# Gradio interface
 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')  # Added default value
+        gr.Radio(['A to B', 'B to A'], label='Convert Direction', value='A to B')
     ],
     outputs=[
         gr.Dataframe(label='Converted Schedule'),
@@ -160,6 +141,5 @@ iface = gr.Interface(
     flagging_mode='never'
 )
 
-# Launch on 0.0.0.0:7860 for Spaces
 if __name__ == "__main__":
     iface.launch(server_name='0.0.0.0', server_port=7860)