Update pages/2player_comparison.py

pages/2player_comparison.py

@@ -3,6 +3,7 @@ import pandas as pd
 import pickle
 import matplotlib.pyplot as plt
 import seaborn as sns
+import numpy as np
 
 # Load model and encoder
 @st.cache_resource
@@ -19,67 +20,102 @@ def load_data():
     df = pd.read_csv('Reduced_final_teams.csv')
     return df
 
-# Main App
+def get_matching_player(name_from_file, player_list):
+    name_lower = name_from_file.lower()
+    for player in player_list:
+        if player.lower() == name_lower:
+            return player
+    return None
+
+def plot_horizontal_bar(df, player1, player2):
+    st.subheader("📊 Stat Comparison - Horizontal Bar Chart")
+    num_cols = df.select_dtypes(include='number').columns
+    df_num = df[num_cols].T
+    df_num.columns = [player1, player2]
+    df_num = df_num.sort_values(by=player1, ascending=False).head(15)  # top 15 metrics
+
+    fig, ax = plt.subplots(figsize=(10, 7))
+    df_num.plot(kind='barh', ax=ax)
+    ax.set_title(f"{player1} vs {player2} - Key Stats")
+    ax.set_xlabel("Value")
+    ax.set_ylabel("Metric")
+    ax.legend(loc="lower right")
+    st.pyplot(fig)
+
+def plot_radar_chart(player1_data, player2_data, player1, player2):
+    st.subheader("🕸 Radar Chart Comparison")
+    categories = ['Matches_ODI', 'Runs_ODI', 'Batting_Average_ODI', 'Strike_Rate_ODI', 'Wickets_ODI', 'Economy_ODI']
+    categories = [cat for cat in categories if cat in player1_data.index]
+
+    values1 = [player1_data[cat] for cat in categories]
+    values2 = [player2_data[cat] for cat in categories]
+
+    angles = np.linspace(0, 2 * np.pi, len(categories), endpoint=False).tolist()
+    values1 += values1[:1]
+    values2 += values2[:1]
+    angles += angles[:1]
+
+    fig, ax = plt.subplots(figsize=(6, 6), subplot_kw=dict(polar=True))
+    ax.plot(angles, values1, label=player1, linewidth=2)
+    ax.fill(angles, values1, alpha=0.25)
+
+    ax.plot(angles, values2, label=player2, linewidth=2)
+    ax.fill(angles, values2, alpha=0.25)
+
+    ax.set_thetagrids(np.degrees(angles[:-1]), categories)
+    ax.set_title("Overall Performance Comparison")
+    ax.legend(loc='upper right', bbox_to_anchor=(1.3, 1.1))
+    st.pyplot(fig)
+
 def main():
-    st.set_page_config(layout="centered")
+    st.set_page_config(layout="wide")
     st.title("Cricket Player Comparison Tool 🏏")
-    st.write("Upload images and manually select the players you want to compare.")
 
     df = load_data()
     model, encoder = load_model_and_encoder()
+    player_list = df['Player'].tolist()
 
-    # Upload player images
-    st.subheader("Step 1: Upload Player Images")
+    # Upload images
     col1, col2 = st.columns(2)
     with col1:
         img1 = st.file_uploader("Upload Image for Player 1", type=['png', 'jpg', 'jpeg'], key='img1')
     with col2:
         img2 = st.file_uploader("Upload Image for Player 2", type=['png', 'jpg', 'jpeg'], key='img2')
 
-    st.subheader("Step 2: Select Players from the Dropdown")
-    player_names = df['Player'].unique().tolist()
-    col3, col4 = st.columns(2)
-    with col3:
-        player1_name = st.selectbox("Select Player 1", player_names, key='player1')
-    with col4:
-        player2_name = st.selectbox("Select Player 2", player_names, key='player2')
-
-    if player1_name and player2_name and player1_name != player2_name:
-        player1_data = df[df['Player'] == player1_name].squeeze()
-        player2_data = df[df['Player'] == player2_name].squeeze()
-
-        st.success(f"Comparing **{player1_name}** vs **{player2_name}**")
-
-        # Combine data for comparison
-        comparison_df = pd.DataFrame([player1_data, player2_data])
-        comparison_df.set_index('Player', inplace=True)
-
-        # Show images if uploaded
-        col5, col6 = st.columns(2)
-        with col5:
-            if img1:
-                st.image(img1, caption=player1_name, use_column_width=True)
-        with col6:
-            if img2:
-                st.image(img2, caption=player2_name, use_column_width=True)
-
-        # Show key metrics
-        st.subheader("📋 Key Metrics")
-        st.dataframe(comparison_df.T)
-
-        # Plot visual comparison
-        st.subheader("📊 Stat Comparison Graph")
-        num_cols = comparison_df.select_dtypes(include='number').columns.tolist()
-
-        plt.figure(figsize=(10, 6))
-        comparison_df[num_cols].T.plot(kind='bar')
-        plt.title(f"{player1_name} vs {player2_name} - Performance Comparison")
-        plt.ylabel("Value")
-        plt.xticks(rotation=45)
-        plt.tight_layout()
-        st.pyplot(plt.gcf())
-    elif player1_name == player2_name:
-        st.warning("Please select two different players.")
+    if img1 and img2:
+        name1_raw = img1.name.rsplit('.', 1)[0]
+        name2_raw = img2.name.rsplit('.', 1)[0]
+
+        player1_name = get_matching_player(name1_raw, player_list)
+        player2_name = get_matching_player(name2_raw, player_list)
+
+        if player1_name and player2_name and player1_name != player2_name:
+            player1_data = df[df['Player'].str.lower() == player1_name.lower()].squeeze()
+            player2_data = df[df['Player'].str.lower() == player2_name.lower()].squeeze()
+
+            st.success(f"Comparing **{player1_name}** vs **{player2_name}**")
+
+            col3, col4 = st.columns(2)
+            with col3:
+                st.image(img1, caption=player1_name, use_container_width=True)
+            with col4:
+                st.image(img2, caption=player2_name, use_container_width=True)
+
+            # Comparison DataFrame
+            comparison_df = pd.DataFrame([player1_data, player2_data])
+            comparison_df.set_index('Player', inplace=True)
+
+            st.subheader("📋 Detailed Stats")
+            st.dataframe(comparison_df.T)
+
+            # Plot visualizations
+            plot_horizontal_bar(comparison_df, player1_name, player2_name)
+            plot_radar_chart(player1_data, player2_data, player1_name, player2_name)
+
+        else:
+            st.error("Player names from image files don't match or are the same. Please check file names.")
+    else:
+        st.info("Please upload two player images to continue.")
 
 if __name__ == "__main__":
     main()