Update app.py

app.py

@@ -106,7 +106,8 @@ if user_prompt and uploaded_file:
     # Trigger assistant's response retrieval and update UI
     with st.spinner("Assistant is typing..."):
         response = handle_query(user_prompt)
-        st.write(response)
+        with st.chat_message("user", avatar="👽"):
+            st.write(response)
         st.session_state.messages.append({'role': 'assistant', "content": response})