Upload hue_portal/chatbot/chatbot.py with huggingface_hub

hue_portal/chatbot/chatbot.py

@@ -177,7 +177,15 @@ class Chatbot(CoreChatbot):
         
         # 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":
+        # CRITICAL: Check conditions and reset BEFORE Stage 1 check
+        should_reset = (
+            intent == "search_legal" 
+            and not has_doc_code_in_query 
+            and wizard_stage == "answer"
+        )
+        print(f"[WIZARD] Reset check - intent={intent}, has_doc_code={has_doc_code_in_query}, wizard_stage={wizard_stage}, should_reset={should_reset}")
+        
+        if should_reset:
             print("[WIZARD] 🔄 New query detected, resetting wizard state for fresh start")
             selected_doc_code = None
             selected_topic = None
@@ -197,6 +205,12 @@ class Chatbot(CoreChatbot):
                     print("[WIZARD] ✅ Wizard state reset in session metadata")
                 except Exception as e:
                     print(f"⚠️ Failed to reset wizard state: {e}")
+            # Also update session_metadata dict for current function scope
+            if session_metadata:
+                session_metadata["selected_document_code"] = None
+                session_metadata["selected_topic"] = None
+                session_metadata["wizard_stage"] = None
+                session_metadata["wizard_depth"] = 0
         
         # 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