Thêm author để custom avatar

app.py

@@ -287,16 +287,20 @@ class UIService:
         ]
     
     @staticmethod
-    async def send_message_with_buttons(content: str, actions=None):
-        """Send message with optional action buttons"""
+    async def send_message_with_buttons(content: str, actions=None, author="assistant"):
+        """Send message with optional action buttons and author"""
         if actions is None:
             actions = UIService.create_action_buttons()
-        await cl.Message(content=content, actions=actions).send()
+        await cl.Message(
+            content=content, 
+            actions=actions,
+            author=author
+        ).send()
     
     @staticmethod
     async def create_typing_animation():
         """Create typing animation effect"""
-        msg = cl.Message(content="")
+        msg = cl.Message(content="", author="assistant")
         await msg.send()
         
         # Typing animation frames
@@ -320,33 +324,38 @@ async def on_chat_start():
     """Initialize the chat session"""
     await StateManager.clear_chat_state(app_state)
     await cl.Message(
-        content="🛍️ **RangDong Sales Agent**\n\nXin chào! Tôi có thể giúp bạn tìm kiếm và tư vấn sản phẩm RangDong. Hãy thử các câu hỏi mẫu:\n\n- Tìm sản phẩm bình giữ nhiệt dung tích dưới 2 lít\n- Tìm sản phẩm ổ cắm thông minh\n- Tư vấn cho tôi đèn học chống cận cho con gái của tôi học lớp 6"
+        content="🛍️ **RangDong Sales Agent**\n\nXin chào! Tôi có thể giúp bạn tìm kiếm và tư vấn sản phẩm RangDong. Hãy thử các câu hỏi mẫu:\n\n- Tìm sản phẩm bình giữ nhiệt dung tích dưới 2 lít\n- Tìm sản phẩm ổ cắm thông minh\n- Tư vấn cho tôi đèn học chống cận cho con gái của tôi học lớp 6",
+        author="assistant"
     ).send()
     
     # Create initial action buttons
     actions = UIService.create_action_buttons()
-    await cl.Message(content="Sử dụng các nút bên dưới để:", actions=actions).send()
+    await cl.Message(
+        content="Sử dụng các nút bên dưới để:", 
+        actions=actions,
+        author="assistant"
+    ).send()
 
 
 @cl.action_callback("show_specs")
 async def on_show_specs(action):
     """Handle show specifications action"""
     specs_content = DisplayService.show_specs(app_state)
-    await UIService.send_message_with_buttons(specs_content)
+    await UIService.send_message_with_buttons(specs_content, author="assistant")
 
 
 @cl.action_callback("show_advantages")
 async def on_show_advantages(action):
     """Handle show advantages action"""
     adv_content = DisplayService.show_advantages(app_state)
-    await UIService.send_message_with_buttons(adv_content)
+    await UIService.send_message_with_buttons(adv_content, author="assistant")
 
 
 @cl.action_callback("show_packages")
 async def on_show_packages(action):
     """Handle show packages action"""
     pkg_content = DisplayService.show_solution_packages(app_state)
-    await UIService.send_message_with_buttons(pkg_content)
+    await UIService.send_message_with_buttons(pkg_content, author="assistant")
 
 
 @cl.action_callback("change_model")
@@ -367,7 +376,8 @@ async def on_change_model(action):
     
     await cl.Message(
         content=f"**Model hiện tại**: {app_state.selected_model}\n\nChọn model mới:",
-        actions=model_actions
+        actions=model_actions,
+        author="assistant"
     ).send()
 
 
@@ -375,25 +385,29 @@ async def on_change_model(action):
 async def on_back_to_main(action):
     """Handle back to main menu action"""
     actions = UIService.create_action_buttons()
-    await cl.Message(content="📋 **Menu chính**\n\nSử dụng các nút bên dưới để:", actions=actions).send()
+    await cl.Message(
+        content="📋 **Menu chính**\n\nSử dụng các nút bên dưới để:", 
+        actions=actions,
+        author="assistant"
+    ).send()
 
 
 @cl.action_callback("select_model_0")
 async def on_select_model_0(action):
     StateManager.change_model(app_state, "Gemini 2.0 Flash")
-    await UIService.send_message_with_buttons("✅ Đã chuyển sang **Gemini 2.0 Flash**")
+    await UIService.send_message_with_buttons("✅ Đã chuyển sang **Gemini 2.0 Flash**", author="assistant")
 
 
 @cl.action_callback("select_model_1")
 async def on_select_model_1(action):
     StateManager.change_model(app_state, "Gemini 2.5 Flash Lite")
-    await UIService.send_message_with_buttons("✅ Đã chuyển sang **Gemini 2.5 Flash Lite**")
+    await UIService.send_message_with_buttons("✅ Đã chuyển sang **Gemini 2.5 Flash Lite**", author="assistant")
 
 
 @cl.action_callback("select_model_2")
 async def on_select_model_2(action):
     StateManager.change_model(app_state, "Gemini 2.0 Flash Lite")
-    await UIService.send_message_with_buttons("✅ Đã chuyển sang **Gemini 2.0 Flash Lite**")
+    await UIService.send_message_with_buttons("✅ Đã chuyển sang **Gemini 2.0 Flash Lite**", author="assistant")
 
 
 @cl.on_message
@@ -416,4 +430,5 @@ async def main(message: cl.Message):
     # Update the typing message with final response and buttons
     typing_msg.content = response
     typing_msg.actions = UIService.create_action_buttons()
+    typing_msg.author = "assistant"  # Ensure author is set
     await typing_msg.update()