update streamlit_app.py 3

streamlit_app.py

@@ -67,9 +67,9 @@ def main():
             st.write(f"**User:** {prompt}")
 
         # Generate response (cached if already asked before)
+        response = cached_response(prompt)
         if hasattr(st, "chat_message"):
             with st.chat_message("assistant"):
-                response = cached_response(prompt)
                 st.markdown(response)
         else:
             st.write(f"**Assistant:** {response}")