Update app.py

app.py

@@ -1,10 +1,19 @@
+import base64
 import streamlit as st
 from chatbot.core import get_chat_response
 
-# Cấu hình giao diện Streamlit
-st.set_page_config(page_title="Explorer Chatbot", layout="wide")
+# Helper function to load and convert image to base64
+def load_image_as_base64(image_path):
+    with open(image_path, "rb") as img_file:
+        img_bytes = img_file.read()
+    encoded = base64.b64encode(img_bytes).decode("utf-8")
+    return f"data:image/png;base64,{encoded}"
+
+# Convert your local avatar image (make sure it's in the same directory or adjust the path)
+avatar_base64 = load_image_as_base64("avatar.png")
+
+st.set_page_config(page_title="Kumiko v1 Assistant", layout="wide")
 
-# Tiêu đề ứng dụng
 st.markdown(
     """
     <style>
@@ -19,34 +28,28 @@ st.markdown(
     unsafe_allow_html=True
 )
 
-# Khởi tạo session state để lưu lịch sử chat
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-# Hiển thị lịch sử chat
+# Display chat history
 for message in st.session_state.messages:
-    # 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"
+        avatar = avatar_base64  # Use the base64 avatar for assistant messages
     else:
         avatar = None
     with st.chat_message(message["role"], avatar=avatar):
         st.markdown(message["content"])
 
-# Ô nhập liệu chat
+# Chat input
 user_input = st.chat_input("Nhập tin nhắn của bạn...")
 
 if user_input:
-    # Hiển thị tin nhắn của user
     st.session_state.messages.append({"role": "user", "content": user_input})
     with st.chat_message("user"):
         st.markdown(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 với avatar
+    
     st.session_state.messages.append({"role": "assistant", "content": response})
-    with st.chat_message("assistant", avatar="avatar.png"):
+    with st.chat_message("assistant", avatar=avatar_base64):
         st.markdown(response)