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

backend/hue_portal/chatbot/chatbot.py

@@ -175,6 +175,27 @@ class Chatbot(CoreChatbot):
         
         print(f"[WIZARD] Chatbot layer check - intent={intent}, wizard_stage={wizard_stage}, selected_doc_code={selected_doc_code}, selected_topic={selected_topic}, has_doc_code_in_query={has_doc_code_in_query}, query='{query[:50]}'")
         
+        # Reset wizard state if new query doesn't have document code and wizard_stage is "answer"
+        # This handles the case where user asks a new question after completing a previous wizard flow
+        if intent == "search_legal" and not has_doc_code_in_query and wizard_stage == "answer":
+            print("[WIZARD] 🔄 New query detected, resetting wizard state for fresh start")
+            selected_doc_code = None
+            selected_topic = None
+            wizard_stage = None
+            if session_id:
+                try:
+                    ConversationContext.update_session_metadata(
+                        session_id,
+                        {
+                            "selected_document_code": None,
+                            "selected_topic": None,
+                            "wizard_stage": None,
+                            "wizard_depth": 0,
+                        }
+                    )
+                except Exception as e:
+                    print(f"⚠️ Failed to reset wizard state: {e}")
+        
         # Stage 1: Choose document (if no document selected and no code in query)
         # Use Query Rewrite Strategy from slow_path_handler instead of old LLM suggestions
         if intent == "search_legal" and not selected_doc_code and not has_doc_code_in_query: