import sys
sys.path.insert(0, ".")
from app.services.rag_service import retrieve
from app.services.query_intent import extract_max_price, extract_brand_mention, is_count_query, detected_category

print("=== Q1: How many products are there in Cadbury? ===")
print("  Is Count:", is_count_query("How many products are there in Cadbury?"))
print("  Brand Mention:", extract_brand_mention("How many products are there in Cadbury?"))
print("  Category:", detected_category("How many products are there in Cadbury?"))

print("\n=== Q2: Suggest panner less than ₹100 ===")
print("  Is Count:", is_count_query("Suggest panner less than ₹100"))
print("  Brand Mention:", extract_brand_mention("Suggest panner less than ₹100"))
print("  Category:", detected_category("Suggest panner less than ₹100"))
print("  Max Price:", extract_max_price("Suggest panner less than ₹100"))
q2_prods = retrieve("Suggest panner less than ₹100")
for p in q2_prods:
    print(f"    - [{p.brand}] {p.product_name} | Price: {p.price_range} | Category: {p.category}")

print("\n=== Q3: Recommend Paneer under ₹150 ===")
print("  Is Count:", is_count_query("Recommend Paneer under ₹150"))
print("  Brand Mention:", extract_brand_mention("Recommend Paneer under ₹150"))
print("  Category:", detected_category("Recommend Paneer under ₹150"))
print("  Max Price:", extract_max_price("Recommend Paneer under ₹150"))
q3_prods = retrieve("Recommend Paneer under ₹150")
for p in q3_prods:
    print(f"    - [{p.brand}] {p.product_name} | Price: {p.price_range} | Category: {p.category}")

print("\n=== Q4: Suggest low sugar biscuits ===")
print("  Category:", detected_category("Suggest low sugar biscuits"))
q4_prods = retrieve("Suggest low sugar biscuits")
for p in q4_prods:
    print(f"    - [{p.brand}] {p.product_name} | Price: {p.price_range} | Category: {p.category}")