Create main.py

main.py

@@ -0,0 +1,48 @@
+import pandas as pd
+import gradio as gr
+
+def convert_schedule(file, direction):
+    # Read the uploaded Excel file; use the first column as index
+    df = pd.read_excel(file.name if hasattr(file, 'name') else file, index_col=0)
+
+    if direction == "A to B":
+        # A->B: models in rows, texters in cells -> texters as rows, models as cell values
+        df_reset = df.reset_index()
+        model_col = df_reset.columns[0]
+        df_melt = df_reset.melt(id_vars=[model_col], var_name='Header', value_name='Texter')
+        df_melt = df_melt.dropna(subset=['Texter'])
+        df_pivot = df_melt.pivot(index='Texter', columns='Header', values=model_col)
+        df_pivot = df_pivot.reindex(columns=df.columns).fillna("")
+        result = df_pivot
+    else:
+        # B->A: texters in rows, models in cells -> models as rows, texters as cell values
+        df_reset = df.reset_index()
+        texter_col = df_reset.columns[0]
+        df_melt = df_reset.melt(id_vars=[texter_col], var_name='Header', value_name='Model')
+        df_melt = df_melt.dropna(subset=['Model'])
+        df_pivot = df_melt.pivot(index='Model', columns='Header', values=texter_col)
+        df_pivot = df_pivot.reindex(columns=df.columns).fillna("")
+        result = df_pivot
+
+    # Save to Excel for download
+    out_file = "converted_schedule.xlsx"
+    result.to_excel(out_file)
+    return result, out_file
+
+# Build Gradio interface
+iface = gr.Interface(
+    fn=convert_schedule,
+    inputs=[
+        gr.File(label="Upload Schedule Excel (Format A or B)"),
+        gr.Radio(["A to B", "B to A"], label="Conversion Direction")
+    ],
+    outputs=[
+        gr.Dataframe(label="Converted Schedule"),
+        gr.File(label="Download Converted Excel")
+    ],
+    title="Schedule Format Converter",
+    description="Dynamically convert between two schedule formats: models-as-rows ↔ texters-as-rows."
+)
+
+if __name__ == "__main__":
+    iface.launch()