Update app.py

app.py

@@ -1,16 +1,10 @@
-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:
-        # Load file
-        if use_default:
-            df = pd.read_excel("default.xlsx")
-        else:
-            df = pd.read_excel(file.name)
+        if file is None and not use_default:
+            return "❌ Please upload a file or select 'Use default.xlsx'", None
+
+        # Read Excel
+        df = pd.read_excel("default.xlsx") if use_default else pd.read_excel(file.name)
 
         # Extract team names
         team_a_names = df.iloc[:, 0].dropna().tolist()
@@ -19,7 +13,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 all 25 teams by swapping 1 player
+        # Generate 25 teams
         all_teams = []
         for i in range(5):
             for j in range(5):
@@ -30,10 +24,9 @@ def generate_teams(file, use_default):
                     "Players": new_team
                 })
 
-        # Generate all 250 combinations
+        # Generate 250 combinations
         full_team_list = []
         team_output_counter = 1
-
         for team in all_teams:
             player_list = team["Players"]
             team_name = team["Team Name"]
@@ -67,29 +60,3 @@ 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 in column A** and **Team B in column B**. You can optionally use a default Excel file named `default.xlsx`.")
-
-    with gr.Row():
-        use_default_checkbox = gr.Checkbox(label="✅ Use default.xlsx file?", value=False)
-        file_input = gr.File(label="📤 Upload Excel File (.xlsx)", file_types=[".xlsx"])
-    
-    with gr.Row():
-        generate_btn = gr.Button("🔄 Generate Teams")
-
-    output_table = gr.Dataframe(label="📋 250 Combinations", interactive=False)
-    download_output = 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_output]
-    )
-
-demo.launch()