diff --git a/src/ui/src/ui/app.py b/src/ui/src/ui/app.py index e075feb..1678625 100644 --- a/src/ui/src/ui/app.py +++ b/src/ui/src/ui/app.py @@ -36,6 +36,8 @@ async def main(): raw_response = response.json()['response']['messages'][-1]['content'] + with st.chat_message("assistant"): + st.markdown(raw_response) st.session_state.messages.append({ "role": "assistant", "content": raw_response