Update app.py

app.py

@@ -1,151 +1,114 @@
 import pandas as pd
-import openpyxl
+import openpyxl  # ensure XLSX engine is available
 import gradio as gr
+import uuid
+import os
+
+# 7-Day Schedule Converter with explicit iteration and header fill and download support
 
 def convert_schedule(file_path, direction):
-    try:
-        # 1. Load raw header rows to determine day labels
-        raw = pd.read_excel(file_path, header=None)
-        header1 = raw.iloc[0, 1:].astype(object)
-        header2 = raw.iloc[1, 1:].astype(object)
+    # 1. Load raw header rows to determine day labels
+    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:
-                if pd.isna(val) or str(val).startswith('Unnamed'):
-                    days.append(last)
-                else:
-                    last = str(val)
-                    days.append(last)
-            data_start = 1
+    # 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:
+            if pd.isna(val) or str(val).startswith('Unnamed'):
+                days.append(last)
+            else:
+                last = str(val)
+                days.append(last)
+        data_start = 1
 
-        # 2. Load actual data using resolved day columns
-        df = pd.read_excel(
-            file_path,
-            header=data_start,
-            index_col=0,
-            usecols=[0] + list(range(1, len(days) + 1))
-        )
-        df.columns = [str(day) for day in days]
+    # 2. Load actual data using resolved day columns
+    df = pd.read_excel(
+        file_path,
+        header=data_start,
+        index_col=0,
+        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)
+    # 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]
-                    if pd.isna(cell):
+    # 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]
+                for texter in str(cell).split(','):
+                    texter = texter.strip()
+                    if not texter or texter.lower() in ['nan', 'none']:
                         continue
-                    for texter in str(cell).split(','):
-                        texter = texter.strip()
-                        if not texter or texter.lower() in ['nan', 'none', '']:
-                            continue
-                        assignments.setdefault(texter, {d: [] for d in day_cols})
-                        assignments[texter][day].append(model)
-            
-            if not assignments:
-                result = pd.DataFrame(columns=day_cols)
-                first_col_name = 'Texter'
-            else:
-                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, [])
-                        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]
-                    if pd.isna(cell):
+                    assignments.setdefault(texter, {d: [] for d in day_cols})
+                    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, [])
+                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]
+                for model in str(cell).split(','):
+                    model = model.strip()
+                    if not model or model.lower() in ['nan', 'none']:
                         continue
-                    for model in str(cell).split(','):
-                        model = model.strip()
-                        if not model or model.lower() in ['nan', 'none', '']:
-                            continue
-                        assignments.setdefault(model, {d: [] for d in day_cols})
-                        assignments[model][day].append(texter)
-            
-            if not assignments:
-                result = pd.DataFrame(columns=day_cols)
-                first_col_name = 'Model'
-            else:
-                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'
+                    assignments.setdefault(model, {d: [] for d in day_cols})
+                    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
-        result.index.name = None
-        result.columns.name = None
+    # 5. Cleanup axis names
+    result.index.name = None
+    result.columns.name = None
 
         # For display, include index as a column
-        display_df = result.reset_index().rename(columns={'index': first_col_name})
-
-        # 6. Create CSV content as string
-        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)
+    display_df = result.reset_index().rename(columns={'index': first_col_name})
 
-        # Convert to CSV string
-        csv_content = result_clean.to_csv(index=True)
-        
-        return display_df, csv_content
+    # 6. Save to Excel for download (relative path so Gradio can serve it)
+    output_filename = f"converted_{uuid.uuid4().hex}.xlsx"
+    display_df.to_excel(output_filename, engine='openpyxl', index=False)
 
-    except Exception as e:
-        error_df = pd.DataFrame({'Error': [f"Error processing file: {str(e)}"]})
-        return error_df, f"Error: {str(e)}"
-
-# Create the interface
-with gr.Blocks(title="7-Day Schedule Converter") as iface:
-    gr.Markdown("# 7-Day Schedule Converter")
-    gr.Markdown("Upload a 7-column weekly schedule and convert between Models↔Texters")
-    
-    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.Row():
-        output_df = gr.Dataframe(label='Converted Schedule Preview')
-    
-    with gr.Row():
-        csv_output = gr.Textbox(
-            label="CSV Data - Copy this text and paste into Excel/Google Sheets", 
-            lines=10, 
-            max_lines=20,
-            placeholder="Converted CSV data will appear here..."
-        )
-    
-    gr.Markdown("### Instructions:")
-    gr.Markdown("1. After conversion, copy ALL the text from the CSV Data box above")
-    gr.Markdown("2. Open Excel or Google Sheets")
-    gr.Markdown("3. Paste the data (it will automatically format into columns)")
-    gr.Markdown("4. Save as .xlsx file")
-    
-    convert_btn.click(
-        fn=convert_schedule,
-        inputs=[file_input, direction_input],
-        outputs=[output_df, csv_output]
-    )
+    # Return both DataFrame and download path
+    return display_df, output_filename
 
-if __name__ == "__main__":
-    iface.launch(server_name='0.0.0.0', server_port=7860)
+# 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'
+)
+iface.launch(server_name='0.0.0.0', server_port=7860)