Update app.py

app.py

@@ -268,17 +268,18 @@ for message in st.session_state.messages:
 if query := st.chat_input("Ask a question"):
     st.session_state.messages.append({"role": "user", "content": query})
     with st.chat_message("user"):
-        st.write(query)
+        st.markdown(query)
 
     moa_agent: MOAgent = st.session_state.moa_agent
     with st.chat_message("assistant"):
-        message_placeholder = st.empty()
-        ast_mess = stream_response(moa_agent.chat(query, output_format='json'))
+        ast_mess = stream_response(moa_agent.chat(query, output_format="json"))
         response = st.write_stream(ast_mess)
-    
+
+    # Save the final response to session state
     st.session_state.messages.append({"role": "assistant", "content": response})
 
 
+
 # Add acknowledgment at the bottom
 st.markdown("---")
 st.markdown("""