Update app.py

app.py

@@ -184,9 +184,9 @@ if user_input:
   # Display the user input as a chat message or an image
   with io_container:
     if input_format == "Text":
-      st.chat_message("user", user_input)
+        st.write(f"user: {user_input}")
     else:
-      st.image(user_input, caption="User input")
+        st.image(user_input, caption="User input")
 
   # Process the user input and generate a response
   # You can use your own logic or a language model here