import asyncio
import sys
import os

# Add project root to path
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from app.ai.agent.graph import get_aida_graph

async def verify_fix():
    print("🚀 Verifying Fixes for Intent & UI...")
    
    graph = get_aida_graph()
    session_id = "verify_session_001"
    user_id = "tester"
    
    # 1. INITIAL LISTING (Should not have replace flag)
    print("\n1. sending initial listing request...")
    state_1 = await graph.ainvoke({
        "user_id": user_id,
        "session_id": session_id,
        "user_role": "landlord",
        "last_user_message": "List a 2-bed in Lagos",
        "start_new_session": True
    }, config={"configurable": {"thread_id": session_id}, "recursion_limit": 50})
    
    temp_data_1 = state_1.get("temp_data", {})
    replace_1 = temp_data_1.get("replace_last_message", False)
    intent_1 = state_1.get("intent_type")
    
    print(f"   Intent: {intent_1}")
    print(f"   Replace Flag: {replace_1} (Expected: False)")
    
    # 2. PROVIDE PRICE (Still building, replace flag false?)
    # "List a 2-bed in Lagos" -> Missing price. So it asks for price.
    
    print("\n2. Providing price...")
    state_2 = await graph.ainvoke({
        "user_id": user_id,
        "session_id": session_id,
        "user_role": "landlord",
        "last_user_message": "50k",
        # Persist state via thread_id
    }, config={"configurable": {"thread_id": session_id}, "recursion_limit": 50})
    
    # Check if we reached validate
    try:
        curr_flow = state_2.get("current_flow")
        # Handle enum if present
        if hasattr(curr_flow, 'value'):
            curr_flow = curr_flow.value
            
        print(f"   Current Flow: {curr_flow}")
    except:
        print(f"   Current Flow: {state_2.get('current_flow')}")
    
    # If we are in validate (draft shown), replace should be False (first show)
    temp_data_2 = state_2.get("temp_data", {})
    replace_2 = temp_data_2.get("replace_last_message", False)
    print(f"   Replace Flag: {replace_2} (Expected: False)")
    
    # 3. CHANGE COMMAND (The Fix!)
    print("\n3. Sending CHANGE command: 'Change price to 85k'...")
    state_3 = await graph.ainvoke({
        "user_id": user_id,
        "session_id": session_id,
        "user_role": "landlord",
        "last_user_message": "Change price to 85k"
    }, config={"configurable": {"thread_id": session_id}, "recursion_limit": 50})
    
    intent_3 = state_3.get("intent_type")
    temp_data_3 = state_3.get("temp_data", {})
    replace_3 = temp_data_3.get("replace_last_message", False)
    draft_3 = state_3.get("listing_draft", {})
    
    print(f"   Intent: {intent_3} (Expected: listing)")
    print(f"   Replace Flag: {replace_3} (Expected: True)")
    print(f"   New Price: {draft_3.get('price')} (Expected: 85000.0)")
    
    passed = True
    if intent_3 != "listing":
        print("❌ Intent check failed")
        passed = False
    if replace_3 is not True:
        print("❌ Replace flag check failed")
        passed = False
    if draft_3.get("price") != 85000.0:
        print("❌ Price update check failed")
        passed = False

    if passed:
        print("\n✅ VERIFICATION PASSED!")
    else:
        print("\n❌ VERIFICATION FAILED")

if __name__ == "__main__":
    asyncio.run(verify_fix())