Upload backend/hue_portal/chatbot/chatbot.py with huggingface_hub

backend/hue_portal/chatbot/chatbot.py

@@ -6,9 +6,11 @@ import copy
 import logging
 import json
 import time
+import unicodedata
+import re
 from typing import Dict, Any, Optional
 from hue_portal.core.chatbot import Chatbot as CoreChatbot, get_chatbot as get_core_chatbot
-from hue_portal.chatbot.router import decide_route, IntentRoute, RouteDecision
+from hue_portal.chatbot.router import decide_route, IntentRoute, RouteDecision, DOCUMENT_CODE_PATTERNS
 from hue_portal.chatbot.context_manager import ConversationContext
 from hue_portal.chatbot.llm_integration import LLMGenerator
 from hue_portal.core.models import LegalSection
@@ -158,6 +160,68 @@ class Chatbot(CoreChatbot):
                 except Exception as e:
                     print(f"⚠️ Failed to save cached bot message: {e}")
             return cached_response
+
+        # Wizard / option-first ngay tại chatbot layer:
+        # Nếu là câu hỏi search_legal chung, chưa chọn văn bản, không có mã văn bản trong câu hỏi
+        # => trả về danh sách văn bản để người dùng chọn, không sinh câu trả lời chi tiết.
+        if (
+            intent == "search_legal"
+            and not selected_doc_code
+            and not self._query_has_document_code(query)
+        ):
+            intro_message = (
+                "Tôi tìm thấy một số nhóm văn bản có thể liên quan đến câu hỏi của bạn.\n\n"
+                "Bạn hãy chọn văn bản muốn tra cứu trước, sau đó tôi sẽ trả lời chi tiết hơn:"
+            )
+            clarification_options = [
+                {
+                    "code": "264-QD-TW",
+                    "title": "Quyết định 264-QĐ/TW về kỷ luật đảng viên",
+                    "reason": "Quy định chung về xử lý kỷ luật đối với đảng viên vi phạm.",
+                },
+                {
+                    "code": "QD-69-TW",
+                    "title": "Quy định 69-QĐ/TW về kỷ luật tổ chức đảng, đảng viên",
+                    "reason": "Quy định chi tiết về các hành vi vi phạm và hình thức kỷ luật.",
+                },
+                {
+                    "code": "TT-02-CAND",
+                    "title": "Thông tư 02/2021/TT-BCA về điều lệnh CAND",
+                    "reason": "Quy định về điều lệnh, lễ tiết, tác phong trong CAND.",
+                },
+                {
+                    "code": "__other__",
+                    "title": "Khác",
+                    "reason": "Tôi muốn hỏi văn bản hoặc chủ đề pháp luật khác.",
+                },
+            ]
+            response = {
+                "message": intro_message,
+                "intent": intent,
+                "confidence": confidence,
+                "results": [],
+                "count": 0,
+                "routing": "legal_wizard",
+                "type": "options",
+                "wizard_stage": "choose_document",
+                "clarification": {
+                    "message": intro_message,
+                    "options": clarification_options,
+                },
+                "options": clarification_options,
+            }
+            if session_id:
+                response["session_id"] = session_id
+                try:
+                    ConversationContext.add_message(
+                        session_id=session_id,
+                        role="bot",
+                        content=intro_message,
+                        intent=intent,
+                    )
+                except Exception as e:
+                    print(f"⚠️ Failed to save wizard bot message: {e}")
+            return response
         
         # Always send legal intent through Slow Path RAG
         if intent == "search_legal":
@@ -501,6 +565,25 @@ class Chatbot(CoreChatbot):
         if not response.get("results"):
             return False
         return True
+
+    def _query_has_document_code(self, query: str) -> bool:
+        """
+        Check if the raw query string explicitly contains a known document code pattern
+        (ví dụ: '264/QĐ-TW', 'QD-69-TW', 'TT-02-CAND').
+        """
+        if not query:
+            return False
+        # Remove accents để regex đơn giản hơn
+        normalized = unicodedata.normalize("NFD", query)
+        normalized = "".join(ch for ch in normalized if unicodedata.category(ch) != "Mn")
+        normalized = normalized.upper()
+        for pattern in DOCUMENT_CODE_PATTERNS:
+            try:
+                if re.search(pattern, normalized):
+                    return True
+            except re.error:
+                continue
+        return False
     
     def _handle_legal_query(self, query: str, session_id: Optional[str] = None) -> Dict[str, Any]:
         """