Update Home.py

Home.py

@@ -1,4 +1,6 @@
 import streamlit as st
+import matplotlib.pyplot as plt
+import numpy as np
 
 # Inject custom CSS (without background image)
 st.markdown(
@@ -109,3 +111,45 @@ if st.button("📊 Player Information"):
 
 # Footer
 st.markdown("<div class='footer'>Created with ❤️ by Sai Kalyan Satwik</div>", unsafe_allow_html=True)
+
+
+# ---------- PIE CHART FUNCTION ----------
+
+def plot_pie_charts(player1_data, player2_data, player1_name, player2_name):
+    # Extract batting and bowling values with fallback to 0 if not found
+    sizes1 = [
+        player1_data.get('Batting', 0),
+        player1_data.get('Bowling', 0)
+    ]
+    sizes2 = [
+        player2_data.get('Batting', 0),
+        player2_data.get('Bowling', 0)
+    ]
+
+    # Replace NaNs or None with 0
+    sizes1 = [0 if x is None or np.isnan(x) else x for x in sizes1]
+    sizes2 = [0 if x is None or np.isnan(x) else x for x in sizes2]
+
+    # Warn and skip if all values are 0 (nothing to plot)
+    if sum(sizes1) == 0 and sum(sizes2) == 0:
+        st.warning("Unable to display pie charts. No valid data found for both players.")
+        return
+    elif sum(sizes1) == 0:
+        st.warning(f"No valid contribution data for {player1_name}.")
+        return
+    elif sum(sizes2) == 0:
+        st.warning(f"No valid contribution data for {player2_name}.")
+        return
+
+    # Plot pie charts
+    fig, axes = plt.subplots(1, 2, figsize=(10, 5))
+    labels = ['Batting', 'Bowling']
+
+    axes[0].pie(sizes1, labels=labels, autopct='%1.1f%%', startangle=90, colors=['#4CAF50', '#2196F3'])
+    axes[0].set_title(f'{player1_name} Contribution')
+
+    axes[1].pie(sizes2, labels=labels, autopct='%1.1f%%', startangle=90, colors=['#4CAF50', '#2196F3'])
+    axes[1].set_title(f'{player2_name} Contribution')
+
+    # Display in Streamlit
+    st.pyplot(fig)