finally resolve avatar issue in handle_user_input fn

app.py

@@ -203,7 +203,7 @@ def handle_user_input(user_input):
                 response, stream_warning, stream_error = write_response()
         else:
             st.session_state["messages"].append({"role": "user", "content": user_input, "warning": None, "error": None})
-            with st.chat_message("user"):
+            with st.chat_message("user", avatar="🧑‍💻"):
                 st.markdown(user_input)
             stream = chat_completion(st.session_state["messages"])
             with st.chat_message("assistant", avatar=MODEL_AVATAR_URL):
@@ -221,7 +221,7 @@ with main:
                 avatar = MODEL_AVATAR_URL
             # else:
             #     avatar = "🧑‍💻"
-            with st.chat_message(message["role"], avatar=MODEL_AVATAR_URL if message["role"] == "assistant" else "🧑‍💻"):
+            with st.chat_message(message["role"], avatar=avatar):
                 if message["content"] is not None:
                     st.markdown(message["content"])
                 if message["error"] is not None: