Update app.py

app.py

@@ -123,14 +123,14 @@ def launch_bot():
     # Display chat messages
     for message in st.session_state.messages:
         with st.chat_message(message["role"], avatar=message["avatar"]):
-            st.text(message["content"])
+            st.write(message["content"])
 
     # User-provided prompt
     if prompt := st.chat_input():
         st.session_state.messages.append({"role": "user", "content": prompt, "avatar": '🧑‍💻'})
         with st.chat_message("user", avatar='🧑‍💻'):
             print_text(f"Starting new question: {prompt}\n", color='green')
-            st.text(prompt)
+            st.write(prompt)
 
     # Generate a new response if last message is not from assistant
     if st.session_state.messages[-1]["role"] != "assistant":
@@ -138,7 +138,7 @@ def launch_bot():
             with st.spinner(st.session_state.thinking_prompt):
                 res = st.session_state.agent.chat(prompt)
                 cleaned = re.sub(r'\[\d+\]', '', res.response)
-                st.text(cleaned)
+                st.write(cleaned)
             message = {"role": "assistant", "content": cleaned, "avatar": '🤖'}
             st.session_state.messages.append(message)