Update app.py

app.py

@@ -180,5 +180,5 @@ if prompt := st.chat_input("Tell me about YSA"):
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})
     st.session_state.messages.append(
-        {"role": "assistant", "content": ref.to_json()}
+        {"role": "assistant", "content": final_ref.to_json()}
     )