Update app.py

app.py

@@ -7,10 +7,12 @@ prompt = st.chat_input("Say something")
 
 if prompt:
     try:
-        st.write_stream(prompt)
+        with st.chat_message("user"):
+            st.markdown(prompt)
     except:
         try:
-            st.write_stream(prompt)
+            with st.chat_message("user"):
+                st.markdown(prompt)
         except Exception as e:
             st.write(str(e))