Update app.py

app.py

@@ -1,42 +1,65 @@
-import gradio as gr
+import os
 import pandas as pd
 from itertools import combinations
 from copy import deepcopy
-import io
+import streamlit as st
 
-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
+st.set_page_config(page_title="250 Captain–Vice-Captain Generator", layout="wide")
+st.title("🏏 Generate All 250 Captain–Vice-Captain Combinations")
+
+# Default file option
+default_path = "default.xlsx"
+use_default = False
+
+if os.path.exists(default_path):
+    use_default = st.checkbox("✅ Use saved default file (default.xlsx)?")
 
+uploaded_file = None
+if not use_default:
+    uploaded_file = st.file_uploader("📤 Upload Excel File", type=["xlsx"])
+
+if use_default or uploaded_file is not None:
+    try:
         # Read Excel
-        df = pd.read_excel("default.xlsx") if use_default else pd.read_excel(file.name)
+        df = pd.read_excel(default_path) if use_default else pd.read_excel(uploaded_file)
 
-        # Extract team names
+        st.subheader("📄 Uploaded Excel File Preview")
+        st.dataframe(df)
+
+        # Extract team names from columns A & B
         team_a_names = df.iloc[:, 0].dropna().tolist()
         team_b_names = df.iloc[:, 1].dropna().tolist()
 
         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 by swapping 1 player
+        # Confirm teams
+        st.write("🔵 **Team A:**", team_a_names)
+        st.write("🟢 **Team B:**", team_b_names)
+
+        # Generate all 25 teams by swapping 1 player
         all_teams = []
-        for i in range(5):
-            for j in range(5):
+        team_number = 1
+
+        for i in range(5):        # index in Team A to swap
+            for j in range(5):    # index from Team B
                 new_team = deepcopy(team_a)
-                new_team[i] = deepcopy(team_b[j])
+                new_team[i] = deepcopy(team_b[j])  # swap one player
                 all_teams.append({
-                    "Team Name": f"Team {len(all_teams)+1}",
+                    "Team Name": f"Team {team_number}",
                     "Players": new_team
                 })
+                team_number += 1
 
-        # Generate 250 captain–vice-captain combinations
+        # Generate all 250 combinations (10 per team)
         full_team_list = []
         team_output_counter = 1
+
         for team in all_teams:
             player_list = team["Players"]
             team_name = team["Team Name"]
+
+            # All C–VC pairs from 5 players
             cvc_combos = list(combinations(player_list, 2))
 
             for cap, vc in cvc_combos:
@@ -53,38 +76,26 @@ def generate_teams(file, use_default):
                             "Player": member['name'],
                             "Role": role
                         })
+
                     full_team_list.extend(team_rows)
                     team_output_counter += 1
 
+        # Create DataFrame
         final_df = pd.DataFrame(full_team_list)
 
-        # Save CSV to memory
-        csv_buffer = io.StringIO()
-        final_df.to_csv(csv_buffer, index=False)
-        csv_buffer.seek(0)
-
-        return final_df, csv_buffer
-
-    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.")
+        st.subheader("📋 All 250 Captain–Vice-Captain Combinations")
+        st.dataframe(final_df)
 
-    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"])
+        # Download button
+        csv = final_df.to_csv(index=False).encode("utf-8")
+        st.download_button(
+            label="📥 Download CSV",
+            data=csv,
+            file_name="250_captain_vc_teams.csv",
+            mime="text/csv"
+        )
 
-    generate_btn = gr.Button("🔄 Generate Teams")
+        st.success(f"✅ Total Unique Team Combinations Generated: {team_output_counter - 1}")
 
-    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()
+    except Exception as e:
+        st.error(f"❌ Error occurred: {e}")