Update pages/2player_comparison.py

pages/2player_comparison.py

@@ -21,51 +21,65 @@ def load_data():
 
 # Main App
 def main():
+    st.set_page_config(layout="centered")
     st.title("Cricket Player Comparison Tool 🏏")
-    st.write("Upload two player images and compare their performance visually.")
+    st.write("Upload images and manually select the players you want to compare.")
 
     df = load_data()
     model, encoder = load_model_and_encoder()
 
-    # Upload Images
+    # Upload player images
+    st.subheader("Step 1: Upload Player 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')
 
-    if img1 and img2:
-        # Placeholder: Extract player name from uploaded file name
-        player1_name = img1.name.split('.')[0]
-        player2_name = img2.name.split('.')[0]
+    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 in df['Player'].values and player2_name in df['Player'].values:
-            player1_data = df[df['Player'] == player1_name].squeeze()
-            player2_data = df[df['Player'] == player2_name].squeeze()
+    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}**")
+        st.success(f"Comparing **{player1_name}** vs **{player2_name}**")
 
-            # Combine both players into one DataFrame for visualization
-            comparison_df = pd.DataFrame([player1_data, player2_data])
-            comparison_df.set_index('Player', inplace=True)
+        # Combine data for comparison
+        comparison_df = pd.DataFrame([player1_data, player2_data])
+        comparison_df.set_index('Player', inplace=True)
 
-            # Display Side-by-Side Metrics
-            st.subheader("🔍 Key Metrics")
-            st.dataframe(comparison_df.T)
+        # 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)
 
-            # Radar chart / Barplot for comparison
-            st.subheader("📊 Visual Comparison")
-            num_cols = comparison_df.select_dtypes(include='number').columns.tolist()
+        # Show key metrics
+        st.subheader("📋 Key Metrics")
+        st.dataframe(comparison_df.T)
 
-            # Plot
-            plt.figure(figsize=(10, 5))
-            comparison_df[num_cols].T.plot(kind='bar', figsize=(10, 6))
-            plt.title(f"Stat Comparison: {player1_name} vs {player2_name}")
-            plt.ylabel("Metric Value")
-            plt.xticks(rotation=45)
-            st.pyplot(plt.gcf())
-        else:
-            st.error("Player names extracted from images do not match any in the dataset. Ensure filenames match player names in the CSV.")
+        # 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 __name__ == "__main__":
     main()