Update app.py

app.py

@@ -25,7 +25,13 @@ if "messages" not in st.session_state:
 
 # Hiển thị lịch sử chat
 for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
+    # Check role to optionally assign an avatar
+    if message["role"] == "assistant":
+        # Provide the URL of your chatbot avatar
+        avatar = "https://example.com/path/to/chatbot_avatar.png"
+    else:
+        avatar = None
+    with st.chat_message(message["role"], avatar=avatar):
         st.markdown(message["content"])
 
 # Ô nhập liệu chat
@@ -40,7 +46,7 @@ if user_input:
     # Gọi chatbot để lấy phản hồi
     response = get_chat_response(user_input)
 
-    # Hiển thị phản hồi của chatbot
+    # Hiển thị phản hồi của chatbot với avatar
     st.session_state.messages.append({"role": "assistant", "content": response})
-    with st.chat_message("assistant"):
+    with st.chat_message("assistant", avatar="avatar.png"):
         st.markdown(response)