Fix bug về button Thông Số Kỹ Thuật và button Ưu điểm - Phải lấy theo sản phẩm (tập sản phẩm) của câu trả lời hiện tại

app.py

@@ -4,8 +4,9 @@ import chainlit as cl
 import pandas as pd
 import requests
 import asyncio
-from typing import Dict, Any, Optional, Callable
+from typing import Dict, List, Any, Optional, Callable
 from dataclasses import dataclass, field
+import os
 
 
 API_BASE_URL = "https://sale-agent-m179.onrender.com"
@@ -15,6 +16,7 @@ API_BASE_URL = "https://sale-agent-m179.onrender.com"
 class ConversationState:
     """Data class to hold conversation state"""
     specs_advantages: Dict[str, Any] = field(default_factory=dict)
+    solution_packages: List[str] = field(default_factory=list)
     raw_documents: Optional[Dict[str, Any]] = None
     outputs: Optional[Dict[str, Any]] = None
     selected_model: str = "Gemini 2.0 Flash"
@@ -22,6 +24,7 @@ class ConversationState:
     def reset(self):
         """Reset state to initial values"""
         self.specs_advantages = {}
+        self.solution_packages = []
         self.raw_documents = None
         self.outputs = None
         self.selected_model = "Gemini 2.0 Flash"
@@ -88,25 +91,54 @@ class ChatService:
                 j = resp.json()
                 response = j.get("response", "")
                 specs_advantages = j.get("specs_advantages")
+                solution_packages = j.get("solution_packages")
                 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
+                specs_advantages, solution_packages, raw_documents, outputs = None, None, None, None
         except Exception as e:
             response = f"Error calling API: {e}"
-            specs_advantages, raw_documents, outputs = None, None, None
+            specs_advantages, solution_packages, raw_documents, outputs = None, None, None, None
 
         end = time.perf_counter()
 
         # Update state
-        if isinstance(specs_advantages, dict):
-            state.specs_advantages = specs_advantages
+        state.specs_advantages = specs_advantages
+        state.solution_packages = solution_packages
         state.raw_documents = raw_documents
         state.outputs = outputs
 
+        # Filter products based on query - FIX: Call static method correctly
+        if state.specs_advantages is not None:
+            ChatService.get_specific_product_from_query(message, state)
+
         return response + f"\n\n*Thời gian xử lí: {end - start:.6f}s*"
 
+    @staticmethod
+    def get_specific_product_from_query(query, state):
+        """Filter specs_advantages based on models found in query"""
+        specs_map = state.specs_advantages or {}
+        product_model_list = []
+        
+        for prod_id, data in specs_map.items():
+            model = data.get("model", None) 
+            if model is not None:
+                product_model_list.append(model)
+        
+        found_models = []
+        for model in product_model_list:
+            pattern = re.escape(model)
+            if re.search(pattern, query, re.IGNORECASE):
+                found_models.append(model)
+        
+        new_specs_advantages = {}
+        if found_models:  # FIX: Changed from != [] to truthy check
+            for prod_id, data in specs_map.items():
+                if data.get("model", None) in found_models:
+                    new_specs_advantages[prod_id] = data
+        
+            state.specs_advantages = new_specs_advantages
 
 class DisplayService:
     """Handles display-related operations"""
@@ -286,13 +318,6 @@ class UIService:
             cl.Action(name="change_model", value="model", label="🔄 Đổi model", payload={"action": "model"}),
         ]
     
-    @staticmethod
-    def create_start_buttons():
-        """Create persistent action buttons"""
-        return [
-            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"):
         """Send message with optional action buttons and author"""
@@ -313,7 +338,7 @@ class UIService:
         # Typing animation frames
         typing_frames = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]
         
-        for i in range(28):  # Show animation for ~7 seconds
+        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()
@@ -336,9 +361,9 @@ async def on_chat_start():
     ).send()
     
     # Create initial action buttons
-    actions = UIService.create_start_buttons()
+    actions = UIService.create_action_buttons()
     await cl.Message(
-        content="Sử dụng nút bên dưới để đổi model:", 
+        content="Sử dụng các nút bên dưới để:", 
         actions=actions,
         author="assistant"
     ).send()