Update app.py

app.py

@@ -1,52 +1,65 @@
 import pandas as pd
 import gradio as gr
 
-# Core conversion logic
+# Pivot between Model↔Texter schedules (7-day weekly format)
 def convert_schedule(file_path, direction):
-    # Read the uploaded Excel file with first column as index
+    # 1. Load schedule: first column is index (Model or Texter), headers are days of the week
     df = pd.read_excel(file_path, index_col=0)
+    # Ensure headers are strings and drop any 'Unnamed' placeholders
+    df.columns = df.columns.astype(str)
+    df = df.loc[:, ~df.columns.str.match(r'^Unnamed')]
 
-    # Clean up any extraneous columns (e.g., blank/Unnamed or hyperlink rows)
-    drop_cols = [col for col in df.columns if isinstance(col, str) and (col.startswith('Unnamed') or col.startswith('http'))]
-    if drop_cols:
-        df = df.drop(columns=drop_cols)
+    # Determine the name of the index column after reset_index()
+    index_name = df.index.name or 'index'
 
+    # 2. Setup melting parameters based on direction
     if direction == 'A to B':
-        # Models in rows → Texters as rows
-        df_reset = df.reset_index().rename(columns={df.index.name or df.columns[0]: 'Model'})
-        df_melt = df_reset.melt(id_vars=['Model'], var_name='Header', value_name='Texter')
-        df_melt = df_melt.dropna(subset=['Texter'])
-        pivot = df_melt.pivot_table(
-            index='Texter', columns='Header', values='Model', aggfunc='first'
-        )
-        result = pivot.reindex(columns=df.columns).fillna('')
+        # A-format: models in index, texters in cells → output: texters as rows, models as cell values
+        id_col = 'Model'
+        val_col = 'Texter'
+        df_reset = df.reset_index().rename(columns={index_name: id_col})
     else:
-        # Texters in rows → Models as rows
-        df_reset = df.reset_index().rename(columns={df.index.name or df.columns[0]: 'Texter'})
-        df_melt = df_reset.melt(id_vars=['Texter'], var_name='Header', value_name='Model')
-        df_melt = df_melt.dropna(subset=['Model'])
-        pivot = df_melt.pivot_table(
-            index='Model', columns='Header', values='Texter', aggfunc='first'
-        )
-        result = pivot.reindex(columns=df.columns).fillna('')
+        # B-format: texters in index, models in cells → output: models as rows, texters as cell values
+        id_col = 'Texter'
+        val_col = 'Model'
+        df_reset = df.reset_index().rename(columns={index_name: id_col})
 
-    return result
+    # 3. Melt into long form: [id_col, Day, val_col]
+    df_melt = df_reset.melt(id_vars=[id_col], var_name='Day', value_name=val_col)
+    # Drop empty assignments
+    df_melt = df_melt.dropna(subset=[val_col])
 
-# Build Gradio interface
-def main():
-    iface = gr.Interface(
-        fn=convert_schedule,
-        inputs=[
-            gr.File(label='Upload Schedule (.xlsx)', file_count='single', type='filepath'),
-            gr.Radio(['A to B', 'B to A'], label='Conversion Direction')
-        ],
-        outputs=gr.Dataframe(label='Converted Schedule'),
-        title='Dynamic Schedule Converter',
-        description=('Upload an Excel file in Format A or B, '
-                     'auto-clean blank/merged header rows, and convert between models-as-rows ↔ texters-as-rows.')
+    # 4. Pivot back to wide: index=val_col, columns=Day, values=id_col
+    pivot = df_melt.pivot_table(
+        index=val_col,
+        columns='Day',
+        values=id_col,
+        aggfunc='first'
     )
-    # Listen on all interfaces for Hugging Face Spaces
-    iface.launch(server_name='0.0.0.0', server_port=7860)
+
+    # 5. Reorder columns to match original days order and fill empties
+    result = pivot.reindex(columns=df.columns.tolist(), fill_value='')
+
+    # Cleanup axis names for display
+    result.index.name = None
+    result.columns.name = None
+    return result
+
+# Build Gradio app
+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'),
+    title='7-Day Schedule Pivot',
+    description=(
+        'Upload a 7-column weekly schedule (Models vs Days), then flip between '  
+        'Models→Texters or Texters→Models. Maintains original days order.'
+    ),
+    allow_flagging='never'
+)
 
 if __name__ == '__main__':
-    main()
+    iface.launch(server_name='0.0.0.0', server_port=7860)