Rename chainlit_demo.py to app.py

chainlit_demo.py → app.py

@@ -1,419 +1,419 @@
-import re
-import time
-import chainlit as cl
-import pandas as pd
-import requests
-import asyncio
-from typing import Dict, Any, Optional, Callable
-from dataclasses import dataclass, field
-
-
-API_BASE_URL = "http://127.0.0.1:7861"
-
-
-@dataclass
-class ConversationState:
-    """Data class to hold conversation state"""
-    specs_advantages: Dict[str, Any] = field(default_factory=dict)
-    raw_documents: Optional[Dict[str, Any]] = None
-    outputs: Optional[Dict[str, Any]] = None
-    selected_model: str = "Gemini 2.0 Flash"
-    
-    def reset(self):
-        """Reset state to initial values"""
-        self.specs_advantages = {}
-        self.raw_documents = None
-        self.outputs = None
-        self.selected_model = "Gemini 2.0 Flash"
-
-
-class StateManager:
-    """Manages conversation state operations"""
-    
-    @staticmethod
-    def create_initial_state() -> ConversationState:
-        """Create initial conversation state"""
-        return ConversationState()
-    
-    @staticmethod
-    async def clear_chat_state(state: ConversationState):
-        """Clear all conversation history and reset state via API"""
-        try:
-            requests.post(f"{API_BASE_URL}/clear_memory",
-                          json={"reset_cache": True, "reset_model": False})
-        except Exception as e:
-            print(f"Warning: clear_memory failed: {e}")
-        
-        # Reset state
-        state.reset()
-    
-    @staticmethod
-    def change_model(state: ConversationState, model_name: str):
-        """Change the selected model"""
-        try:
-            requests.post(f"{API_BASE_URL}/set_model",
-                          json={"model_name": model_name})
-            state.selected_model = model_name
-        except Exception as e:
-            print(f"Warning: set_model failed: {e}")
-
-
-class ChatService:
-    """Handles chat-related operations"""
-    
-    @staticmethod
-    async def respond_to_chat(
-        state: ConversationState, 
-        message: str, 
-        image_path: Optional[str] = None
-    ) -> str:
-        """Handle chat responses with image support"""
-        print(f"🔄 === DEBUG STATE ===\n Chat request with model: {state.selected_model}")
-        start = time.perf_counter()
-        
-        # Call API
-        try:
-            if image_path:
-                with open(image_path, 'rb') as f:
-                    files = {"image": f}
-                    data = {"message": message}
-                    resp = requests.post(
-                        f"{API_BASE_URL}/chat_with_image", files=files, data=data, timeout=120)
-            else:
-                payload = {"message": message}
-                resp = requests.post(f"{API_BASE_URL}/chat",
-                                     json=payload, timeout=120)
-
-            if resp.status_code == 200:
-                j = resp.json()
-                response = j.get("response", "")
-                specs_advantages = j.get("specs_advantages")
-                raw_documents = j.get("raw_documents")
-                outputs = j.get("outputs")
-            else:
-                response = f"Error: API status {resp.status_code}"
-                specs_advantages, raw_documents, outputs = None, None, None
-        except Exception as e:
-            response = f"Error calling API: {e}"
-            specs_advantages, raw_documents, outputs = None, None, None
-
-        end = time.perf_counter()
-
-        # Update state
-        if isinstance(specs_advantages, dict):
-            state.specs_advantages = specs_advantages
-        state.raw_documents = raw_documents
-        state.outputs = outputs
-
-        return response + f"\n\n*Thời gian xử lí: {end - start:.6f}s*"
-
-
-class DisplayService:
-    """Handles display-related operations"""
-    
-    @staticmethod
-    def show_specs(state: ConversationState) -> str:
-        """Generate specifications table"""
-        specs_map = state.specs_advantages
-        columns = ["Thông số"]
-        raw_data = []
-
-        if not specs_map:
-            return "📄 **Thông số kỹ thuật**\n\nKhông có thông số kỹ thuật nào."
-
-        print(specs_map)
-        for prod_id, data in specs_map.items():
-            spec = data.get("specification", None)
-            model = data.get("model", "")
-            url = data.get("url", "")
-
-            # Handle both products and solution packages
-            if url:
-                full_name = f"**[{data['name']} {model}]({url})**"
-            else:
-                full_name = f"**{data['name']} {model}**"
-
-            if full_name not in columns:
-                columns.append(full_name)
-
-            if spec:
-                # Check if this is a solution package (contains markdown table)
-                if "### 📦" in spec:
-                    # For solution packages, parse the markdown table properly
-                    lines = spec.split('\n')
-                    in_table = False
-                    headers = []
-
-                    for line in lines:
-                        line = line.strip()
-                        if '|' in line and '---' not in line and line.startswith('|') and line.endswith('|'):
-                            cells = [cell.strip()
-                                     for cell in line.split('|')[1:-1]]
-
-                            if not in_table:
-                                # This is the header row
-                                headers = cells
-                                in_table = True
-                                continue
-
-                            # This is a data row
-                            if len(cells) >= len(headers):
-                                for i, header in enumerate(headers):
-                                    if i < len(cells):
-                                        param_name = header
-                                        param_value = cells[i]
-
-                                        existing_row = None
-                                        for row in raw_data:
-                                            if row["Thông số"] == param_name:
-                                                existing_row = row
-                                                break
-
-                                        if existing_row:
-                                            existing_row[full_name] = param_value
-                                        else:
-                                            new_row = {"Thông số": param_name}
-                                            for col in columns[1:]:
-                                                new_row[col] = ""
-                                            new_row[full_name] = param_value
-                                            raw_data.append(new_row)
-                        elif in_table and (not line or not line.startswith('|')):
-                            in_table = False
-                else:
-                    # For products, parse specification items
-                    items = re.split(r';|\n', spec)
-                    for item in items:
-                        if ":" in item:
-                            key, value = item.split(':', 1)
-                            spec_key = key.strip().capitalize()
-                            if spec_key == "Vậtl iệu":
-                                spec_key = "Vật liệu"
-
-                            existing_row = None
-                            for row in raw_data:
-                                if row["Thông số"] == spec_key:
-                                    existing_row = row
-                                    break
-
-                            if existing_row:
-                                existing_row[full_name] = value.strip() if value else ""
-                            else:
-                                new_row = {"Thông số": spec_key}
-                                for col in columns[1:]:
-                                    new_row[col] = ""
-                                new_row[full_name] = value.strip() if value else ""
-                                raw_data.append(new_row)
-
-        if raw_data:
-            df = pd.DataFrame(raw_data, columns=columns)
-            df = df.fillna("").replace("None", "").replace("nan", "")
-        else:
-            df = pd.DataFrame(
-                [["Không có thông số kỹ thuật", "", ""]], columns=columns)
-
-        markdown_table = df.to_markdown(index=False)
-        return f"📄 **Thông số kỹ thuật**\n\n{markdown_table}"
-    
-    @staticmethod
-    def show_advantages(state: ConversationState) -> str:
-        """Generate advantages table"""
-        specs_map = state.specs_advantages
-        columns = ["Tên", "Ưu điểm nổi trội"]
-        table_data = []
-
-        if not specs_map:
-            return "💡 **Ưu điểm nổi trội**\n\nKhông có ưu điểm nào."
-
-        for prod_id, data in specs_map.items():
-            adv = data.get("advantages", "Không có ưu điểm")
-            model = data.get("model", "")
-            url = data.get("url", "")
-
-            # Handle both products and solution packages
-            if url:
-                full_name = f"**[{data['name']} {model}]({url})**"
-            else:
-                full_name = f"**{data['name']} {model}**"
-
-            if adv not in ["Không có ưu điểm", "", None]:
-                table_data.append([full_name, adv])
-
-        if table_data:
-            df = pd.DataFrame(table_data, columns=columns)
-        else:
-            df = pd.DataFrame([["Không có ưu điểm", ""]], columns=columns)
-
-        markdown_table = df.to_markdown(index=False)
-        return f"💡 **Ưu điểm nổi trội**\n\n{markdown_table}"
-    
-    @staticmethod
-    def show_solution_packages(state: ConversationState) -> str:
-        """Show solution packages in a structured format"""
-        specs_map = state.specs_advantages
-
-        if not specs_map:
-            return "📦 **Gói sản phẩm**\n\nKhông có gói sản phẩm nào"
-
-        # Filter only solution packages
-        solution_packages = {}
-        for key, data in specs_map.items():
-            if data.get("model") == "Gói giải pháp":
-                solution_packages[key] = data
-
-        if not solution_packages:
-            return "📦 **Gói sản phẩm**\n\nKhông có gói sản phẩm nào"
-
-        # Build markdown content for each package
-        markdown_content = "## 📦 Gói sản phẩm\n\n"
-
-        for key, data in solution_packages.items():
-            spec_content = data.get("specification", "")
-            markdown_content += spec_content + "\n\n"
-
-        return markdown_content
-
-
-class UIService:
-    """Handles UI-related operations"""
-    
-    @staticmethod
-    def create_action_buttons():
-        """Create persistent action buttons"""
-        return [
-            cl.Action(name="show_specs", value="specs", label="📄 Thông số kỹ thuật", payload={"action": "specs"}),
-            cl.Action(name="show_advantages", value="advantages", label="💡 Ưu điểm nổi trội", payload={"action": "advantages"}),
-            cl.Action(name="show_packages", value="packages", label="📦 Gói sản phẩm", payload={"action": "packages"}),
-            cl.Action(name="change_model", value="model", label="🔄 Đổi model", payload={"action": "model"}),
-        ]
-    
-    @staticmethod
-    async def send_message_with_buttons(content: str, actions=None):
-        """Send message with optional action buttons"""
-        if actions is None:
-            actions = UIService.create_action_buttons()
-        await cl.Message(content=content, actions=actions).send()
-    
-    @staticmethod
-    async def create_typing_animation():
-        """Create typing animation effect"""
-        msg = cl.Message(content="")
-        await msg.send()
-        
-        # Typing animation frames
-        typing_frames = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]
-        
-        for i in range(8):  # Show animation for ~2 seconds
-            frame = typing_frames[i % len(typing_frames)]
-            msg.content = f"{frame} Đang suy nghĩ..."
-            await msg.update()
-            await asyncio.sleep(0.25)
-        
-        return msg
-
-
-# Application state - using dependency injection pattern
-app_state = StateManager.create_initial_state()
-
-
-@cl.on_chat_start
-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"
-    ).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()
-
-
-@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)
-
-
-@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)
-
-
-@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)
-
-
-@cl.action_callback("change_model")
-async def on_change_model(action):
-    """Handle model change action"""
-    models = ["Gemini 2.0 Flash", "Gemini 2.5 Flash Lite", "Gemini 2.0 Flash Lite"]
-    
-    # Create model selection actions
-    model_actions = [
-        cl.Action(name=f"select_model_{i}", value=model, label=model, payload={"model": model})
-        for i, model in enumerate(models)
-    ]
-    
-    # Add back button to return to main actions
-    model_actions.append(
-        cl.Action(name="back_to_main", value="back", label="🔙 Quay lại", payload={"action": "back"})
-    )
-    
-    await cl.Message(
-        content=f"**Model hiện tại**: {app_state.selected_model}\n\nChọn model mới:",
-        actions=model_actions
-    ).send()
-
-
-@cl.action_callback("back_to_main")
-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()
-
-
-@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**")
-
-
-@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**")
-
-
-@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**")
-
-
-@cl.on_message
-async def main(message: cl.Message):
-    """Main message handler"""
-    # Handle images if present
-    image_path = None
-    if message.elements:
-        for element in message.elements:
-            if isinstance(element, cl.Image):
-                image_path = element.path
-                break
-    
-    # Show typing animation
-    typing_msg = await UIService.create_typing_animation()
-    
-    # Get response from API - injecting state as dependency
-    response = await ChatService.respond_to_chat(app_state, message.content, image_path)
-    
-    # Update the typing message with final response and buttons
-    typing_msg.content = response
-    typing_msg.actions = UIService.create_action_buttons()
+import re
+import time
+import chainlit as cl
+import pandas as pd
+import requests
+import asyncio
+from typing import Dict, Any, Optional, Callable
+from dataclasses import dataclass, field
+
+
+API_BASE_URL = "https://sale-agent-m179.onrender.com"
+
+
+@dataclass
+class ConversationState:
+    """Data class to hold conversation state"""
+    specs_advantages: Dict[str, Any] = field(default_factory=dict)
+    raw_documents: Optional[Dict[str, Any]] = None
+    outputs: Optional[Dict[str, Any]] = None
+    selected_model: str = "Gemini 2.0 Flash"
+    
+    def reset(self):
+        """Reset state to initial values"""
+        self.specs_advantages = {}
+        self.raw_documents = None
+        self.outputs = None
+        self.selected_model = "Gemini 2.0 Flash"
+
+
+class StateManager:
+    """Manages conversation state operations"""
+    
+    @staticmethod
+    def create_initial_state() -> ConversationState:
+        """Create initial conversation state"""
+        return ConversationState()
+    
+    @staticmethod
+    async def clear_chat_state(state: ConversationState):
+        """Clear all conversation history and reset state via API"""
+        try:
+            requests.post(f"{API_BASE_URL}/clear_memory",
+                          json={"reset_cache": True, "reset_model": False})
+        except Exception as e:
+            print(f"Warning: clear_memory failed: {e}")
+        
+        # Reset state
+        state.reset()
+    
+    @staticmethod
+    def change_model(state: ConversationState, model_name: str):
+        """Change the selected model"""
+        try:
+            requests.post(f"{API_BASE_URL}/set_model",
+                          json={"model_name": model_name})
+            state.selected_model = model_name
+        except Exception as e:
+            print(f"Warning: set_model failed: {e}")
+
+
+class ChatService:
+    """Handles chat-related operations"""
+    
+    @staticmethod
+    async def respond_to_chat(
+        state: ConversationState, 
+        message: str, 
+        image_path: Optional[str] = None
+    ) -> str:
+        """Handle chat responses with image support"""
+        print(f"🔄 === DEBUG STATE ===\n Chat request with model: {state.selected_model}")
+        start = time.perf_counter()
+        
+        # Call API
+        try:
+            if image_path:
+                with open(image_path, 'rb') as f:
+                    files = {"image": f}
+                    data = {"message": message}
+                    resp = requests.post(
+                        f"{API_BASE_URL}/chat_with_image", files=files, data=data, timeout=120)
+            else:
+                payload = {"message": message}
+                resp = requests.post(f"{API_BASE_URL}/chat",
+                                     json=payload, timeout=120)
+
+            if resp.status_code == 200:
+                j = resp.json()
+                response = j.get("response", "")
+                specs_advantages = j.get("specs_advantages")
+                raw_documents = j.get("raw_documents")
+                outputs = j.get("outputs")
+            else:
+                response = f"Error: API status {resp.status_code}"
+                specs_advantages, raw_documents, outputs = None, None, None
+        except Exception as e:
+            response = f"Error calling API: {e}"
+            specs_advantages, raw_documents, outputs = None, None, None
+
+        end = time.perf_counter()
+
+        # Update state
+        if isinstance(specs_advantages, dict):
+            state.specs_advantages = specs_advantages
+        state.raw_documents = raw_documents
+        state.outputs = outputs
+
+        return response + f"\n\n*Thời gian xử lí: {end - start:.6f}s*"
+
+
+class DisplayService:
+    """Handles display-related operations"""
+    
+    @staticmethod
+    def show_specs(state: ConversationState) -> str:
+        """Generate specifications table"""
+        specs_map = state.specs_advantages
+        columns = ["Thông số"]
+        raw_data = []
+
+        if not specs_map:
+            return "📄 **Thông số kỹ thuật**\n\nKhông có thông số kỹ thuật nào."
+
+        print(specs_map)
+        for prod_id, data in specs_map.items():
+            spec = data.get("specification", None)
+            model = data.get("model", "")
+            url = data.get("url", "")
+
+            # Handle both products and solution packages
+            if url:
+                full_name = f"**[{data['name']} {model}]({url})**"
+            else:
+                full_name = f"**{data['name']} {model}**"
+
+            if full_name not in columns:
+                columns.append(full_name)
+
+            if spec:
+                # Check if this is a solution package (contains markdown table)
+                if "### 📦" in spec:
+                    # For solution packages, parse the markdown table properly
+                    lines = spec.split('\n')
+                    in_table = False
+                    headers = []
+
+                    for line in lines:
+                        line = line.strip()
+                        if '|' in line and '---' not in line and line.startswith('|') and line.endswith('|'):
+                            cells = [cell.strip()
+                                     for cell in line.split('|')[1:-1]]
+
+                            if not in_table:
+                                # This is the header row
+                                headers = cells
+                                in_table = True
+                                continue
+
+                            # This is a data row
+                            if len(cells) >= len(headers):
+                                for i, header in enumerate(headers):
+                                    if i < len(cells):
+                                        param_name = header
+                                        param_value = cells[i]
+
+                                        existing_row = None
+                                        for row in raw_data:
+                                            if row["Thông số"] == param_name:
+                                                existing_row = row
+                                                break
+
+                                        if existing_row:
+                                            existing_row[full_name] = param_value
+                                        else:
+                                            new_row = {"Thông số": param_name}
+                                            for col in columns[1:]:
+                                                new_row[col] = ""
+                                            new_row[full_name] = param_value
+                                            raw_data.append(new_row)
+                        elif in_table and (not line or not line.startswith('|')):
+                            in_table = False
+                else:
+                    # For products, parse specification items
+                    items = re.split(r';|\n', spec)
+                    for item in items:
+                        if ":" in item:
+                            key, value = item.split(':', 1)
+                            spec_key = key.strip().capitalize()
+                            if spec_key == "Vậtl iệu":
+                                spec_key = "Vật liệu"
+
+                            existing_row = None
+                            for row in raw_data:
+                                if row["Thông số"] == spec_key:
+                                    existing_row = row
+                                    break
+
+                            if existing_row:
+                                existing_row[full_name] = value.strip() if value else ""
+                            else:
+                                new_row = {"Thông số": spec_key}
+                                for col in columns[1:]:
+                                    new_row[col] = ""
+                                new_row[full_name] = value.strip() if value else ""
+                                raw_data.append(new_row)
+
+        if raw_data:
+            df = pd.DataFrame(raw_data, columns=columns)
+            df = df.fillna("").replace("None", "").replace("nan", "")
+        else:
+            df = pd.DataFrame(
+                [["Không có thông số kỹ thuật", "", ""]], columns=columns)
+
+        markdown_table = df.to_markdown(index=False)
+        return f"📄 **Thông số kỹ thuật**\n\n{markdown_table}"
+    
+    @staticmethod
+    def show_advantages(state: ConversationState) -> str:
+        """Generate advantages table"""
+        specs_map = state.specs_advantages
+        columns = ["Tên", "Ưu điểm nổi trội"]
+        table_data = []
+
+        if not specs_map:
+            return "💡 **Ưu điểm nổi trội**\n\nKhông có ưu điểm nào."
+
+        for prod_id, data in specs_map.items():
+            adv = data.get("advantages", "Không có ưu điểm")
+            model = data.get("model", "")
+            url = data.get("url", "")
+
+            # Handle both products and solution packages
+            if url:
+                full_name = f"**[{data['name']} {model}]({url})**"
+            else:
+                full_name = f"**{data['name']} {model}**"
+
+            if adv not in ["Không có ưu điểm", "", None]:
+                table_data.append([full_name, adv])
+
+        if table_data:
+            df = pd.DataFrame(table_data, columns=columns)
+        else:
+            df = pd.DataFrame([["Không có ưu điểm", ""]], columns=columns)
+
+        markdown_table = df.to_markdown(index=False)
+        return f"💡 **Ưu điểm nổi trội**\n\n{markdown_table}"
+    
+    @staticmethod
+    def show_solution_packages(state: ConversationState) -> str:
+        """Show solution packages in a structured format"""
+        specs_map = state.specs_advantages
+
+        if not specs_map:
+            return "📦 **Gói sản phẩm**\n\nKhông có gói sản phẩm nào"
+
+        # Filter only solution packages
+        solution_packages = {}
+        for key, data in specs_map.items():
+            if data.get("model") == "Gói giải pháp":
+                solution_packages[key] = data
+
+        if not solution_packages:
+            return "📦 **Gói sản phẩm**\n\nKhông có gói sản phẩm nào"
+
+        # Build markdown content for each package
+        markdown_content = "## 📦 Gói sản phẩm\n\n"
+
+        for key, data in solution_packages.items():
+            spec_content = data.get("specification", "")
+            markdown_content += spec_content + "\n\n"
+
+        return markdown_content
+
+
+class UIService:
+    """Handles UI-related operations"""
+    
+    @staticmethod
+    def create_action_buttons():
+        """Create persistent action buttons"""
+        return [
+            cl.Action(name="show_specs", value="specs", label="📄 Thông số kỹ thuật", payload={"action": "specs"}),
+            cl.Action(name="show_advantages", value="advantages", label="💡 Ưu điểm nổi trội", payload={"action": "advantages"}),
+            cl.Action(name="show_packages", value="packages", label="📦 Gói sản phẩm", payload={"action": "packages"}),
+            cl.Action(name="change_model", value="model", label="🔄 Đổi model", payload={"action": "model"}),
+        ]
+    
+    @staticmethod
+    async def send_message_with_buttons(content: str, actions=None):
+        """Send message with optional action buttons"""
+        if actions is None:
+            actions = UIService.create_action_buttons()
+        await cl.Message(content=content, actions=actions).send()
+    
+    @staticmethod
+    async def create_typing_animation():
+        """Create typing animation effect"""
+        msg = cl.Message(content="")
+        await msg.send()
+        
+        # Typing animation frames
+        typing_frames = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]
+        
+        for i in range(8):  # Show animation for ~2 seconds
+            frame = typing_frames[i % len(typing_frames)]
+            msg.content = f"{frame} Đang suy nghĩ..."
+            await msg.update()
+            await asyncio.sleep(0.25)
+        
+        return msg
+
+
+# Application state - using dependency injection pattern
+app_state = StateManager.create_initial_state()
+
+
+@cl.on_chat_start
+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"
+    ).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()
+
+
+@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)
+
+
+@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)
+
+
+@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)
+
+
+@cl.action_callback("change_model")
+async def on_change_model(action):
+    """Handle model change action"""
+    models = ["Gemini 2.0 Flash", "Gemini 2.5 Flash Lite", "Gemini 2.0 Flash Lite"]
+    
+    # Create model selection actions
+    model_actions = [
+        cl.Action(name=f"select_model_{i}", value=model, label=model, payload={"model": model})
+        for i, model in enumerate(models)
+    ]
+    
+    # Add back button to return to main actions
+    model_actions.append(
+        cl.Action(name="back_to_main", value="back", label="🔙 Quay lại", payload={"action": "back"})
+    )
+    
+    await cl.Message(
+        content=f"**Model hiện tại**: {app_state.selected_model}\n\nChọn model mới:",
+        actions=model_actions
+    ).send()
+
+
+@cl.action_callback("back_to_main")
+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()
+
+
+@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**")
+
+
+@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**")
+
+
+@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**")
+
+
+@cl.on_message
+async def main(message: cl.Message):
+    """Main message handler"""
+    # Handle images if present
+    image_path = None
+    if message.elements:
+        for element in message.elements:
+            if isinstance(element, cl.Image):
+                image_path = element.path
+                break
+    
+    # Show typing animation
+    typing_msg = await UIService.create_typing_animation()
+    
+    # Get response from API - injecting state as dependency
+    response = await ChatService.respond_to_chat(app_state, message.content, image_path)
+    
+    # Update the typing message with final response and buttons
+    typing_msg.content = response
+    typing_msg.actions = UIService.create_action_buttons()
     await typing_msg.update()