Update src/streamlit_app.py

src/streamlit_app.py

@@ -27,6 +27,23 @@ def get_api_key():
     # 2. If that fails, look in the Hugging Face Environment variables
     return os.environ.get("GROQ_API_KEY")
 
+
+@st.fragment
+def render_chart_section(title, func, key_id):
+    with st.container(border=True):
+        st.subheader(title)
+        col1, col2 = st.columns([2, 1])
+        with col1:
+            fig, ax = plt.subplots()
+            func() # Ye plotting function execute karega
+            if any(x in title for x in ["Venue", "Season", "City"]):
+                plt.xticks(rotation=45)
+            st.pyplot(fig, use_container_width=True) # Container width se jumping rukegi
+        with col2:
+            ai_explainer_ui(fig, key_id)
+        plt.close(fig)
+        
+
 # Now, use this variable throughout your app
 GROQ_API_KEY = get_api_key()
 
@@ -86,12 +103,7 @@ def get_image_base64(fig):
     return base64.b64encode(buf.getvalue()).decode('utf-8')
 
 
-@st.fragment
-def render_chart_section(title, fig, key_id):
-    with st.container(border=True):
-        st.subheader(title)
-        st.pyplot(fig)
-        ai_explainer_ui(fig, key_id)
+
 
 def ai_explainer_ui(fig, key_id):
     state_key = f"explanation_{key_id}"
@@ -140,8 +152,7 @@ def ai_explainer_ui(fig, key_id):
             except Exception as e:
                 st.error(f"Connection Error: {e}")
 
-    if st.session_state[state_key]:
-        st.info(st.session_state[state_key])
+
         
 st.title(f"🏏 {analysis_mode}")
 if analysis_mode == "Overview":
@@ -188,14 +199,17 @@ if analysis_mode == "Overview":
         )
 
         fig1, ax1 = plt.subplots(figsize=(8, 5))
-        ax1.plot(season_counts.index.astype(str), season_counts.values,
-                 marker='o', color='#1f77b4', linewidth=2)
-        ax1.set_xlabel("Season", fontsize=11)
-        ax1.set_ylabel("Matches Played", fontsize=11)
-        ax1.tick_params(axis='x', rotation=45)
-        ax1.grid(axis='y', alpha=0.3)
-        fig1.tight_layout()          # ← instance method, not plt.tight_layout()
-        st.pyplot(fig1)
+        # ax1.plot(season_counts.index.astype(str), season_counts.values,
+        #          marker='o', color='#1f77b4', linewidth=2)
+        # ax1.set_xlabel("Season", fontsize=11)
+        # ax1.set_ylabel("Matches Played", fontsize=11)
+        # ax1.tick_params(axis='x', rotation=45)
+        # ax1.grid(axis='y', alpha=0.3)
+        # fig1.tight_layout()          # ← instance method, not plt.tight_layout()
+        # st.pyplot(fig1)
+
+        render_chart_section("Matches per Season", fig1, "overview_season")
+        
         plt.close(fig1)              # ← free memory immediately
         ai_explainer_ui(fig1, "overview_season")
     st.divider()
@@ -211,18 +225,20 @@ if analysis_mode == "Overview":
         )
 
         fig2, ax2 = plt.subplots(figsize=(8, 5))   # ← plt.subplots, NOT Figure()
-        colors = plt.cm.viridis_r(
-            [i / max(len(team_runs) - 1, 1) for i in range(len(team_runs))]
-        )
-        ax2.barh(team_runs.index[::-1], team_runs.values[::-1], color=colors[::1])
-        ax2.set_xlabel("Total Runs", fontsize=11)
-        ax2.set_ylabel("Team", fontsize=11)
-        ax2.xaxis.set_major_formatter(
-            plt.FuncFormatter(lambda x, _: f"{int(x/1000)}k" if x >= 1000 else str(int(x)))
-        )
-        ax2.grid(axis='x', alpha=0.3)
-        fig2.tight_layout()
-        st.pyplot(fig2)
+        # colors = plt.cm.viridis_r(
+        #     [i / max(len(team_runs) - 1, 1) for i in range(len(team_runs))]
+        # )
+        # ax2.barh(team_runs.index[::-1], team_runs.values[::-1], color=colors[::1])
+        # ax2.set_xlabel("Total Runs", fontsize=11)
+        # ax2.set_ylabel("Team", fontsize=11)
+        # ax2.xaxis.set_major_formatter(
+        #     plt.FuncFormatter(lambda x, _: f"{int(x/1000)}k" if x >= 1000 else str(int(x)))
+        # )
+        # ax2.grid(axis='x', alpha=0.3)
+        # fig2.tight_layout()
+        # st.pyplot(fig2)
+
+        render_chart_section("Top 10 Run Scoring Teams", fig2, "overview_teams")
         plt.close(fig2)
         ai_explainer_ui(fig2, "overview_teams")
 
@@ -312,19 +328,20 @@ elif analysis_mode == "IPL Team Analysis (2008-2025)":
 
 
     for i, (title, func) in enumerate(plots):
-        st.subheader(f"{i+1}. {title}")
-        col1, col2 = st.columns([2, 1])
-        with col1:
-            fig, ax = plt.subplots()
-            func()
+        render_chart_section(f"{i+1}. {title}", func, f"team_btn_{i}")
+        # st.subheader(f"{i+1}. {title}")
+        # col1, col2 = st.columns([2, 1])
+        # with col1:
+        #     fig, ax = plt.subplots()
+        #     func()
             
-            if "Venue" in title or "Season" in title or "City" in title:
-                plt.xticks(rotation=45)
+        #     if "Venue" in title or "Season" in title or "City" in title:
+        #         plt.xticks(rotation=45)
                 
-            st.pyplot(fig)
-            plt.close(fig)
-        with col2:
-            ai_explainer_ui(fig, f"team_btn_{i}")
+        #     st.pyplot(fig)
+        #     plt.close(fig)
+        # with col2:
+        #     ai_explainer_ui(fig, f"team_btn_{i}")
     st.divider()
         
 elif analysis_mode == "Ball-by-Ball Analysis":
@@ -367,13 +384,15 @@ elif analysis_mode == "Ball-by-Ball Analysis":
     ]
 
     for i, (title, func) in enumerate(bb_plots):
-        st.subheader(f"{i+1}. {title}")
-        col1, col2 = st.columns([2, 1])
-        with col1:
-            fig, ax = plt.subplots()
-            func()
-            st.pyplot(fig)
-            plt.close(fig)
-        with col2:
-            ai_explainer_ui(fig, f"bb_btn_{i}")
+        # st.subheader(f"{i+1}. {title}")
+        # col1, col2 = st.columns([2, 1])
+        # with col1:
+        #     fig, ax = plt.subplots()
+        #     func()
+        #     st.pyplot(fig)
+        #     plt.close(fig)
+        # with col2:
+        #     ai_explainer_ui(fig, f"bb_btn_{i}")
+
+        render_chart_section(f"{i+1}. {title}", func, f"bb_btn_{i}")
         st.divider()