update UI/UX...

app.py

@@ -337,10 +337,10 @@ def process_message(
     payload_content = f'{operation_prompt} :\n"""\n{message}\n"""'
 
     st.session_state.messages.append({"role": "user", "content": payload_content})
-    with st.chat_message("user"):
+    with st.chat_message("user", avatar="👤"):
         st.markdown(message)
 
-    with st.chat_message("assistant"):
+    with st.chat_message("assistant", avatar="👻"):
         message_placeholder = st.empty()
         full_response = ""
         try: