Fix: Cải thiện xử lý follow-up questions và off-topic questions

backend/hue_portal/chatbot/chatbot.py

@@ -113,14 +113,105 @@ class Chatbot(CoreChatbot):
             }
         
         elif route_decision.route == IntentRoute.SMALL_TALK:
-            response = {
-                "message": "Tôi có thể giúp bạn tra cứu các văn bản quy định pháp luật về xử lí kỷ luật cán bộ đảng viên. Bạn muốn tìm gì?",
-                "intent": intent,
-                "confidence": confidence,
-                "results": [],
-                "count": 0,
-                "routing": "small_talk"
-            }
+            # Xử lý follow-up questions trong context
+            follow_up_keywords = ["có điều khoản", "liên quan", "khác", "nữa", "thêm", "tóm tắt", "tải file"]
+            query_lower = query.lower()
+            is_follow_up = any(kw in query_lower for kw in follow_up_keywords)
+            
+            response = None
+            
+            # Nếu là follow-up question, thử tìm context từ conversation trước
+            if is_follow_up and session_id:
+                try:
+                    recent_messages = ConversationContext.get_recent_messages(session_id, limit=5)
+                    # Tìm message bot cuối cùng có results
+                    for msg in reversed(recent_messages):
+                        if msg.role == "bot" and msg.intent == "search_legal":
+                            # Có context về legal query trước đó, thử search lại với query mới
+                            enhanced_query = f"{query} {msg.content[:100]}"
+                            search_result = self.search_by_intent("search_legal", enhanced_query, limit=3)
+                            if search_result["count"] > 0:
+                                # Tìm thấy results, trả về
+                                top_result = search_result["results"][0]
+                                top_data = top_result.get("data", {})
+                                doc_code = top_data.get("document_code", "")
+                                doc_title = top_data.get("document_title", "văn bản pháp luật")
+                                section_code = top_data.get("section_code", "")
+                                section_title = top_data.get("section_title", "")
+                                content = top_data.get("content", "") or top_data.get("excerpt", "")
+                                
+                                if "tóm tắt" in query_lower:
+                                    content_preview = content[:400] + "..." if len(content) > 400 else content
+                                    message = (
+                                        f"**Tóm tắt {section_code}**: {section_title or 'Nội dung chính'}\n\n"
+                                        f"{content_preview}\n\n"
+                                        f"Nguồn: {doc_title}" + (f" ({doc_code})" if doc_code else "")
+                                    )
+                                elif "tải" in query_lower or "download" in query_lower:
+                                    message = (
+                                        f"Bạn có thể tải file gốc của {doc_title}" + (f" ({doc_code})" if doc_code else "") +
+                                        f" từ link download trong kết quả tìm kiếm."
+                                    )
+                                else:
+                                    # Câu hỏi "có điều khoản liên quan nào khác không?"
+                                    if search_result["count"] > 1:
+                                        message = (
+                                            f"Có, tôi tìm thấy {search_result['count']} điều khoản liên quan:\n\n"
+                                        )
+                                        for i, result in enumerate(search_result["results"][:3], 1):
+                                            data = result.get("data", {})
+                                            sec_code = data.get("section_code", "")
+                                            sec_title = data.get("section_title", "")
+                                            message += f"{i}. **{sec_code}**: {sec_title or 'Nội dung liên quan'}\n"
+                                        message += f"\nNguồn: {doc_title}" + (f" ({doc_code})" if doc_code else "")
+                                    else:
+                                        message = (
+                                            f"Tôi đã tìm thấy điều khoản liên quan:\n\n"
+                                            f"**{section_code}**: {section_title or 'Nội dung liên quan'}\n\n"
+                                            f"{content[:300]}...\n\n"
+                                            f"Nguồn: {doc_title}" + (f" ({doc_code})" if doc_code else "")
+                                        )
+                                
+                                response = {
+                                    "message": message,
+                                    "intent": "search_legal",
+                                    "confidence": 0.85,
+                                    "results": search_result["results"][:3],
+                                    "count": search_result["count"],
+                                    "routing": "follow_up"
+                                }
+                                break
+                except Exception as e:
+                    logger.warning(f"[FOLLOW_UP] Failed to process follow-up: {e}")
+            
+            # Nếu không phải follow-up hoặc không tìm thấy context, trả về message thân thiện
+            if response is None:
+                # Detect off-topic questions (nấu ăn, chả trứng, etc.)
+                off_topic_keywords = ["nấu", "nau", "chả trứng", "cha trung", "món ăn", "mon an", "công thức", "cong thuc", 
+                                     "cách làm", "cach lam", "đổ chả", "do cha", "trứng", "trung"]
+                is_off_topic = any(kw in query_lower for kw in off_topic_keywords)
+                
+                if is_off_topic:
+                    message = (
+                        "Xin lỗi, tôi là chatbot chuyên về tra cứu các văn bản quy định pháp luật "
+                        "về xử lí kỷ luật cán bộ đảng viên của Phòng Thanh Tra - Công An Thành Phố Huế.\n\n"
+                        "Tôi không thể trả lời các câu hỏi về nấu ăn, công thức nấu ăn hay các chủ đề khác ngoài phạm vi pháp luật.\n\n"
+                        "Bạn có muốn tra cứu thông tin về:\n"
+                        "- Các quy định về xử lí kỷ luật cán bộ đảng viên\n"
+                        "- Các điều khoản trong Thông tư 02 về xử lý điều lệnh trong CAND\n"
+                        "- Hoặc các văn bản pháp luật liên quan khác?"
+                    )
+                else:
+                    message = "Tôi có thể giúp bạn tra cứu các văn bản quy định pháp luật về xử lí kỷ luật cán bộ đảng viên. Bạn muốn tìm gì?"
+                
+                response = {
+                    "message": message,
+                    "intent": intent,
+                    "confidence": confidence,
+                    "results": [],
+                    "count": 0,
+                    "routing": "small_talk"
+                }
         
         else:  # IntentRoute.SEARCH
             # Use core chatbot search for other intents