Update src/streamlit_app.py

src/streamlit_app.py

@@ -85,12 +85,28 @@ def get_image_base64(fig):
     fig.savefig(buf, format="png", bbox_inches='tight')
     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}"
     
     if state_key not in st.session_state:
         st.session_state[state_key] = None
 
+
+    # Create a reserved slot
+    info_slot = st.empty()
+
+    # Show existing explanation immediately in the reserved slot
+    if st.session_state[state_key]:
+        info_slot.info(st.session_state[state_key])
+
     if st.button("Explain with AI", key=f"btn_{key_id}"):
         # Use the key we retrieved at the top of the script
         api_key = GROQ_API_KEY 
@@ -116,6 +132,9 @@ def ai_explainer_ui(fig, key_id):
                 
                 if response.status_code == 200:
                     st.session_state[state_key] = response.json()['explanation']
+
+                    st.session_state[state_key] = explanation_text
+                    info_slot.info(explanation_text)
                 else:
                     st.error(f"API Error: {response.status_code}")
             except Exception as e: