Update app.py

app.py

@@ -108,6 +108,7 @@ if prompt := st.chat_input("How can I help?"):
 
     with st.chat_message("assistant", avatar=BOT_AVATAR):
         message_placeholder = st.empty()
+        full_response = chain.invoke(prompt)['result']
         message_placeholder.markdown(chain.invoke(prompt)['result'])
-    st.session_state.messages.append({"role": "assistant", "content": full_response})
+        st.session_state.messages.append({"role": "assistant", "content": full_response})
 save_chat_history(st.session_state.messages)