Update app.py

app.py

@@ -109,6 +109,6 @@ if user_prompt and uploaded_file:
 if st.session_state.messages[-1]["role"] != "assistant":
     with st.chat_message("assistant"):
         response = handle_query(user_prompt)
-        full_response = st.write_stream(response)
+        full_response = st.write(response)
     message = {"role": "assistant", "content": full_response}
     st.session_state.messages.append(message)