Update app.py

app.py

@@ -1,5 +1,12 @@
+import gradio as gr
+import pandas as pd
+from itertools import combinations
+from copy import deepcopy
+import io
+
 def generate_teams(file, use_default):
     try:
+        # Handle missing input
         if file is None and not use_default:
             return "❌ Please upload a file or select 'Use default.xlsx'", None
 
@@ -13,7 +20,7 @@ def generate_teams(file, use_default):
         team_a = [{"name": str(name).strip()} for name in team_a_names]
         team_b = [{"name": str(name).strip()} for name in team_b_names]
 
-        # Generate 25 teams
+        # Generate 25 teams by swapping 1 player
         all_teams = []
         for i in range(5):
             for j in range(5):
@@ -24,7 +31,7 @@ def generate_teams(file, use_default):
                     "Players": new_team
                 })
 
-        # Generate 250 combinations
+        # Generate 250 captain–vice-captain combinations
         full_team_list = []
         team_output_counter = 1
         for team in all_teams:
@@ -51,7 +58,7 @@ def generate_teams(file, use_default):
 
         final_df = pd.DataFrame(full_team_list)
 
-        # Save to CSV in-memory
+        # Save CSV to memory
         csv_buffer = io.StringIO()
         final_df.to_csv(csv_buffer, index=False)
         csv_buffer.seek(0)
@@ -60,3 +67,24 @@ def generate_teams(file, use_default):
 
     except Exception as e:
         return f"❌ Error: {str(e)}", None
+
+# Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("# 🏏 Generate All 250 Captain–Vice-Captain Combinations")
+    gr.Markdown("Upload an Excel file with Team A (Column A) and Team B (Column B) or use a default file.")
+
+    with gr.Row():
+        use_default_checkbox = gr.Checkbox(label="✅ Use default.xlsx?", value=False)
+        file_input = gr.File(label="📤 Upload Excel File (.xlsx)", file_types=[".xlsx"])
+
+    generate_btn = gr.Button("🔄 Generate Teams")
+
+    output_table = gr.Dataframe(label="📋 250 C–VC Combinations")
+    download_button = gr.File(label="📥 Download CSV")
+
+    def wrapper(file, use_default):
+        return generate_teams(file, use_default)
+
+    generate_btn.click(fn=wrapper, inputs=[file_input, use_default_checkbox], outputs=[output_table, download_button])
+
+demo.launch()