Update app.py

app.py

@@ -1,26 +1,20 @@
 import pandas as pd
-import openpyxl
 import gradio as gr
-import io
-import base64
-from datetime import datetime
 
 def convert_schedule(file_path, direction):
     if file_path is None:
-        return pd.DataFrame({'Error': ['Please upload a file']}), None
+        return pd.DataFrame({'Message': ['Please upload a file']}), ""
         
     try:
-        # 1. Load raw header rows to determine day labels
+        # Same conversion logic as before
         raw = pd.read_excel(file_path, header=None)
         header1 = raw.iloc[0, 1:].astype(object)
         header2 = raw.iloc[1, 1:].astype(object)
 
-        # 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
         else:
-            # Forward-fill merged first-row headers
             days = []
             last = None
             for val in header1:
@@ -31,7 +25,6 @@ def convert_schedule(file_path, direction):
                     days.append(last)
             data_start = 1
 
-        # 2. Load actual data using resolved day columns
         df = pd.read_excel(
             file_path,
             header=data_start,
@@ -39,14 +32,10 @@ def convert_schedule(file_path, direction):
             usecols=[0] + list(range(1, len(days) + 1))
         )
         df.columns = [str(day) for day in days]
-
-        # 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]
@@ -71,7 +60,6 @@ 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]
@@ -96,98 +84,44 @@ 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
         result.index.name = None
         result.columns.name = None
-
-        # For display, include index as a column
+        
+        # For display
         display_df = result.reset_index().rename(columns={'index': first_col_name})
-
-        # 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 CSV file in memory (more reliable than Excel)
+        # Create CSV text for copying
+        result_clean = result.copy().fillna('OFF')
         download_df = result_clean.reset_index().rename(columns={'index': first_col_name})
+        csv_text = download_df.to_csv(index=False)
         
-        # Create CSV content
-        csv_buffer = io.StringIO()
-        download_df.to_csv(csv_buffer, index=False)
-        csv_content = csv_buffer.getvalue()
-        
-        # Create filename with timestamp
-        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-        filename = f"converted_schedule_{timestamp}.csv"
-        
-        # Return the CSV content for download
-        return display_df, (csv_content.encode('utf-8'), filename)
+        return display_df, csv_text
 
     except Exception as e:
-        error_msg = f"Error processing file: {str(e)}"
-        print(f"DEBUG: {error_msg}")
-        error_df = pd.DataFrame({'Error': [error_msg]})
-        return error_df, None
+        error_df = pd.DataFrame({'Error': [f"Error: {str(e)}"]})
+        return error_df, f"Error: {str(e)}"
 
-# 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)
-    
-    # Write CSV content
-    with open(temp_path, 'w', encoding='utf-8') as f:
-        f.write(excel_content.decode('utf-8'))
-    
-    return display_result, temp_path
-
-# Create the interface
+# Interface with copy-paste text output
 iface = gr.Interface(
-    fn=process_and_download,
+    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'
-        )
+        gr.File(label='Upload Weekly Schedule (.xlsx)', file_types=['.xlsx', '.xls']),
+        gr.Radio(['A to B', 'B to A'], label='Convert Direction', value='A to B')
     ],
     outputs=[
-        gr.Dataframe(label='Converted Schedule (Preview)', wrap=True),
-        gr.File(label='Download Converted Schedule (.csv)')
+        gr.Dataframe(label='Converted Schedule (Preview)'),
+        gr.Textbox(
+            label='CSV Data (Copy This Text)', 
+            lines=20, 
+            max_lines=50,
+            show_copy_button=True,
+            info="Copy this text and paste it into Excel or save as .csv file"
+        )
     ],
-    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*'
-    ),
+    title='🔄 Schedule Converter - Copy/Paste Version',
+    description='Upload → Convert → Copy the CSV text → Paste into Excel or save as .csv',
     flagging_mode='never'
 )
 
 if __name__ == "__main__":
-    iface.launch(
-        server_name='0.0.0.0', 
-        server_port=7860,
-        share=False
-    )
+    iface.launch(server_name='0.0.0.0', server_port=7860)