#!/usr/bin/env python3
"""Quick test for the preference extractor."""

import sys
sys.path.insert(0, "/projects/bfqt/users/yurenh2/ml-projects/personalization-user-model/src")

from personalization.config.registry import get_preference_extractor

print("="*60)
print("PREFERENCE EXTRACTOR TEST")
print("="*60)

print("\nLoading extractor (qwen3_0_6b_sft)...")
extractor = get_preference_extractor("qwen3_0_6b_sft")
print("Extractor loaded successfully!")

# Test extraction with various queries
test_queries = [
    "I prefer Python over Java for scripting tasks",
    "Please use bullet points instead of numbered lists",
    "Can you explain this in simpler terms? I'm a beginner.",
    "I like concise answers, not long explanations",
    "Always show code examples when explaining programming concepts",
]

print("\n" + "="*60)
print("EXTRACTION TESTS")
print("="*60)

for i, query in enumerate(test_queries, 1):
    print(f"\n--- Test {i} ---")
    print(f"Query: {query}")
    result = extractor.extract_preferences(query)
    print(f"Extracted: {result}")

    if result.get("preferences"):
        for pref in result["preferences"]:
            print(f"  - condition: {pref.get('condition', 'N/A')}")
            print(f"    action: {pref.get('action', 'N/A')}")
            print(f"    confidence: {pref.get('confidence', 'N/A')}")
    else:
        print("  (No preferences extracted)")

print("\n" + "="*60)
print("TEST COMPLETE")
print("="*60)