branch 1

src/streamlit_app.py

@@ -164,11 +164,10 @@ with st.container():
                         if st.button("I don't understand. Try again.", key="retry_button"):
                             st.session_state.retry_count += 1
                             alt_llm = get_llm(temperature=0.9)
-                            st.session_state.response = alt_llm.invoke(
-                                st.session_state.full_prompt
-                            )
-                        st.info(st.session_state.response)
-                        
+                            st.session_state.response = alt_llm.invoke([HumanMessage(content=st.session_state.full_prompt)])
+    if st.session_state.response:
+        st.markdown("response")
+        st.info(st.session_state.response)
 # Optional: Add footer or spacing
 st.markdown("<br><br>", unsafe_allow_html=True)