from utils.animal_rag import get_animal_rag

rag = get_animal_rag()

queries = [
    "스킨카솔",
    "센텔라",
    "피부 재생",
    "피부 보호",
    "피부 진정",
    "상처 회복",
    "피부 케어",
    "습진",
    "아토피",
    "티트리오일",
]

for query in queries:
    results = rag.search(query, n_results=3)
    has_skincasol = any("skincasol" in r["source"].lower() for r in results)
    mark = "O" if has_skincasol else "X"
    print(f"[{mark}] {query}")
    if has_skincasol:
        for r in results:
            if "skincasol" in r["source"].lower():
                print(f"    -> {r['score']:.0%} {r['section']}")