Update pages/2player_comparison.py

pages/2player_comparison.py

@@ -33,6 +33,7 @@ def plot_horizontal_bar(df, player1, player2):
     num_cols = df.select_dtypes(include='number').columns
     df_num = df[num_cols].T
     df_num.columns = [player1, player2]
+    df_num = df_num.fillna(0)
     df_num = df_num.sort_values(by=player1, ascending=False).head(15)
 
     fig, ax = plt.subplots(figsize=(10, 7))
@@ -49,8 +50,17 @@ def plot_pie_charts(player1_data, player2_data, player1, player2):
     col1, col2 = st.columns(2)
 
     for col, player_data, player_name in zip([col1, col2], [player1_data, player2_data], [player1, player2]):
-        batting_total = player_data.get('Runs_ODI', 0) + player_data.get('Runs_T20', 0) + player_data.get('Runs_Test', 0)
-        bowling_total = player_data.get('Wickets_ODI', 0) + player_data.get('Wickets_T20', 0) + player_data.get('Wickets_Test', 0)
+        batting_total = sum([
+            player_data.get('Runs_ODI', 0) or 0,
+            player_data.get('Runs_T20', 0) or 0,
+            player_data.get('Runs_Test', 0) or 0
+        ])
+
+        bowling_total = sum([
+            player_data.get('Wickets_ODI', 0) or 0,
+            player_data.get('Wickets_T20', 0) or 0,
+            player_data.get('Wickets_Test', 0) or 0
+        ])
 
         labels = ['Batting', 'Bowling']
         sizes = [batting_total, bowling_total]
@@ -68,7 +78,7 @@ def plot_bowling_comparison(df, player1, player2):
     bowling_cols = [col for col in df.columns if 'Wickets' in col or 'Economy' in col or 'Bowling_Average' in col]
     df_bowling = df[bowling_cols].T
     df_bowling.columns = [player1, player2]
-    df_bowling = df_bowling.dropna().sort_values(by=player1, ascending=False)
+    df_bowling = df_bowling.fillna(0).sort_values(by=player1, ascending=False)
 
     fig, ax = plt.subplots(figsize=(10, 6))
     df_bowling.plot(kind='bar', ax=ax)
@@ -78,6 +88,18 @@ def plot_bowling_comparison(df, player1, player2):
     ax.legend(loc="upper right")
     st.pyplot(fig)
 
+# Bowling stats summary
+def show_bowling_summary(player_data, player_name):
+    st.subheader(f"🎳 Bowling Summary: {player_name}")
+    col1, col2, col3 = st.columns(3)
+    col1.metric("Wickets (ODI)", int(player_data.get("Wickets_ODI", 0) or 0))
+    col2.metric("Wickets (T20)", int(player_data.get("Wickets_T20", 0) or 0))
+    col3.metric("Wickets (Test)", int(player_data.get("Wickets_Test", 0) or 0))
+
+    col4, col5 = st.columns(2)
+    col4.metric("Economy (ODI)", round(player_data.get("Economy_ODI", 0) or 0, 2))
+    col5.metric("Bowling Avg (Test)", round(player_data.get("Bowling_Average_Test", 0) or 0, 2))
+
 # Main app
 def main():
     st.set_page_config(layout="wide")
@@ -102,8 +124,8 @@ def main():
         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()
+            player1_data = df[df['Player'].str.lower() == player1_name.lower()].fillna(0).squeeze()
+            player2_data = df[df['Player'].str.lower() == player2_name.lower()].fillna(0).squeeze()
 
             st.success(f"Comparing **{player1_name}** vs **{player2_name}**")
 
@@ -119,6 +141,10 @@ def main():
             st.subheader("📋 Full Stats Table")
             st.dataframe(comparison_df.T)
 
+            # Stats summaries
+            show_bowling_summary(player1_data, player1_name)
+            show_bowling_summary(player2_data, player2_name)
+
             # Visualizations
             plot_horizontal_bar(comparison_df, player1_name, player2_name)
             plot_pie_charts(player1_data, player2_data, player1_name, player2_name)