Bổ sung tính năng tìm theo mã sản phẩm

app.py

@@ -20,6 +20,7 @@ class ConversationState:
     raw_documents: Optional[Dict[str, Any]] = None
     outputs: Optional[Dict[str, Any]] = None
     selected_model: str = "Gemini 2.0 Flash"
+    product_model_search: bool = False
     
     def reset(self):
         """Reset state to initial values"""
@@ -28,6 +29,7 @@ class ConversationState:
         self.raw_documents = None
         self.outputs = None
         self.selected_model = "Gemini 2.0 Flash"
+        self.product_model_search = False
 
 
 class StateManager:
@@ -59,6 +61,11 @@ class StateManager:
             state.selected_model = model_name
         except Exception as e:
             print(f"Warning: set_model failed: {e}")
+    
+    @staticmethod
+    def toggle_product_model_search(state: ConversationState):
+        """Toggle product model search mode"""
+        state.product_model_search = not state.product_model_search
 
 
 class ChatService:
@@ -71,7 +78,7 @@ class ChatService:
         image_path: Optional[str] = None
     ) -> str:
         """Handle chat responses with image support"""
-        print(f"🔄 === DEBUG STATE ===\n Chat request with model: {state.selected_model}")
+        print(f"🔄 === DEBUG STATE ===\n Chat request with model: {state.selected_model}, Product Model Search: {state.product_model_search}")
         start = time.perf_counter()
         
         # Call API
@@ -79,11 +86,18 @@ class ChatService:
             if image_path:
                 with open(image_path, 'rb') as f:
                     files = {"image": f}
-                    data = {"message": message}
+                    data = {
+                        "message": message,
+                        "product_model_search": str(state.product_model_search).lower()
+                    }
                     resp = requests.post(
                         f"{API_BASE_URL}/chat_with_image", files=files, data=data, timeout=120)
             else:
-                payload = {"message": message}
+                payload = {
+                    "message": message,
+                    "debug": "Normal",
+                    "product_model_search": state.product_model_search
+                }
                 resp = requests.post(f"{API_BASE_URL}/chat",
                                      json=payload, timeout=120)
 
@@ -286,24 +300,7 @@ class DisplayService:
         if not packages or packages == []:
             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"
-
         markdown_table = "\n\n".join(packages)
-
         return markdown_table
 
 
@@ -311,26 +308,34 @@ class UIService:
     """Handles UI-related operations"""
     
     @staticmethod
-    def create_action_buttons():
+    def create_action_buttons(state: ConversationState):
         """Create persistent action buttons"""
+        # Determine the product search button text and icon
+        search_status = "🔍 Tìm theo mã sản phẩm (Đang tắt)" if not state.product_model_search else "🔍 Tìm theo mã sản phẩm (Đang bật)"
+        
         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="toggle_product_search", value="toggle_search", label=search_status, payload={"action": "toggle_search"}),
             cl.Action(name="change_model", value="model", label="🔄 Đổi model", payload={"action": "model"}),
         ]
 
-    def create_start_buttons():
+    @staticmethod
+    def create_start_buttons(state: ConversationState):
         """Create start buttons"""
+        search_status = "🔍 Tìm theo mã sản phẩm (Đang tắt)" if not state.product_model_search else "🔍 Tìm theo mã sản phẩm (Đang bật)"
+        
         return [
+            cl.Action(name="toggle_product_search", value="toggle_search", label=search_status, payload={"action": "toggle_search"}),
             cl.Action(name="change_model", value="model", label="🔄 Đổi model", payload={"action": "model"}),
         ]
     
     @staticmethod
-    async def send_message_with_buttons(content: str, actions=None, author="assistant"):
+    async def send_message_with_buttons(content: str, state: ConversationState, actions=None, author="assistant"):
         """Send message with optional action buttons and author"""
         if actions is None:
-            actions = UIService.create_action_buttons()
+            actions = UIService.create_action_buttons(state)
         await cl.Message(
             content=content, 
             actions=actions,
@@ -369,9 +374,9 @@ async def on_chat_start():
     ).send()
     
     # Create initial action buttons
-    actions = UIService.create_start_buttons()
+    actions = UIService.create_start_buttons(app_state)
     await cl.Message(
-        content="Sử dụng nút bên dưới để đổi mô hình:", 
+        content="Sử dụng nút bên dưới để cấu hình:", 
         actions=actions,
         author="assistant"
     ).send()
@@ -381,21 +386,37 @@ async def on_chat_start():
 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, author="assistant")
+    await UIService.send_message_with_buttons(specs_content, app_state, 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, author="assistant")
+    await UIService.send_message_with_buttons(adv_content, app_state, 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, author="assistant")
+    await UIService.send_message_with_buttons(pkg_content, app_state, author="assistant")
+
+
+@cl.action_callback("toggle_product_search")
+async def on_toggle_product_search(action):
+    """Handle toggle product model search action"""
+    StateManager.toggle_product_model_search(app_state)
+    
+    status_message = (
+        "✅ **Đã bật tìm kiếm theo mã sản phẩm**\n\n"
+        "Khi bạn nhắc đến mã/model cụ thể trong câu hỏi, hệ thống sẽ tìm kiếm chính xác theo mã đó."
+        if app_state.product_model_search 
+        else "✅ **Đã tắt tìm kiếm theo mã sản phẩm**\n\n"
+        "Hệ thống sẽ tìm kiếm sản phẩm theo cách thông thường."
+    )
+    
+    await UIService.send_message_with_buttons(status_message, app_state, author="assistant")
 
 
 @cl.action_callback("change_model")
@@ -415,7 +436,7 @@ 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:",
+        content=f"**Model hiện tại**: {app_state.selected_model}\n**Tìm kiếm theo mã**: {'Đang bật' if app_state.product_model_search else 'Đang tắt'}\n\nChọn model mới:",
         actions=model_actions,
         author="assistant"
     ).send()
@@ -424,7 +445,7 @@ async def on_change_model(action):
 @cl.action_callback("back_to_main")
 async def on_back_to_main(action):
     """Handle back to main menu action"""
-    actions = UIService.create_action_buttons()
+    actions = UIService.create_action_buttons(app_state)
     await cl.Message(
         content="📋 **Menu chính**\n\nSử dụng các nút bên dưới để:", 
         actions=actions,
@@ -435,19 +456,19 @@ async def on_back_to_main(action):
 @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**", author="assistant")
+    await UIService.send_message_with_buttons("✅ Đã chuyển sang **Gemini 2.0 Flash**", app_state, 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**", author="assistant")
+    await UIService.send_message_with_buttons("✅ Đã chuyển sang **Gemini 2.5 Flash Lite**", app_state, 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**", author="assistant")
+    await UIService.send_message_with_buttons("✅ Đã chuyển sang **Gemini 2.0 Flash Lite**", app_state, author="assistant")
 
 
 @cl.on_message
@@ -469,6 +490,6 @@ 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.actions = UIService.create_action_buttons(app_state)
     typing_msg.author = "assistant"  # Ensure author is set
     await typing_msg.update()