Update app.py

app.py

@@ -75,7 +75,7 @@ with st.chat_message("assistant"):
         message_placeholder.markdown(full_response + "▌")
     message_placeholder.markdown(full_response)
 # Add assistant response to chat history
-st.session_state.messages.append({"role": "assistant", "content": final_response})
+st.session_state.messages.append({"role": "assistant", "content": full_response})
 
 if prompt := st.chat_input("What kind of class are you looking for?"):
     # Display user message in chat message container
@@ -109,5 +109,5 @@ if prompt := st.chat_input("What kind of class are you looking for?"):
                 message_placeholder.markdown(full_response + "▌")
             message_placeholder.markdown(full_response)
         # Add assistant response to chat history
-        st.session_state.messages.append({"role": "assistant", "content": final_response})
+        st.session_state.messages.append({"role": "assistant", "content": full_response})