LAYA 쇼핑 검색 1 — "50만원이하 패딩 갈색" 을 필터로: 질문 여섯 개 설계

"""검색 문장 하나를 질문 여섯 개로 묻고, 답을 검색엔진 필터로 바꾼다.

python first_query.py base                  # 학습 전 다국어 체크포인트
python first_query.py checkpoints/shop-v1   # 내가 학습한 판
"""
import json
import sys

import laya

QUESTIONS = {
    "price": {"type": "choice", "instructions": "검색어의 가격 조건에 맞는 가격 구간은? 구간 경계와 다르면 그 금액을 넘는 가장 작은 구간",
              "criteria": {"under10": "10만원 이하", "under30": "30만원 이하", "under50": "50만원 이하",
                           "under100": "100만원 이하", "none": "가격 조건 없음"}},
    "category": {"type": "choice", "instructions": "찾는 상품의 대분류는?",
                 "criteria": {"top": "상의·아우터", "bottom": "하의", "shoes": "신발", "bag": "가방", "none": "분류를 알 수 없음"}},
    "item": {"type": "choice", "instructions": "찾는 상품의 품목은?",
             "criteria": {"padding": "패딩", "coat": "코트", "knit": "니트", "tshirt": "티셔츠", "jeans": "청바지",
                          "slacks": "슬랙스", "sneakers": "운동화", "boots": "부츠", "bag": "가방", "none": "품목 언급 없음"}},
    "winter": {"type": "noul", "instructions": "겨울옷(패딩·코트·니트·부츠, 또는 겨울에 쓸 상품)을 찾는가?"},
    "gender": {"type": "choice", "instructions": "누가 쓸 상품인가?",
               "criteria": {"men": "남성", "women": "여성", "unknown": "알 수 없음"}},
    "color": {"type": "choice", "instructions": "원하는 색은?",
              "criteria": {"brown": "갈색", "black": "검정", "white": "흰색", "gray": "회색", "navy": "남색",
                           "beige": "베이지", "none": "색 언급 없음"}},
}
EMPTY = {"none", "unknown"}   # "언급 없음"·"알 수 없음" 은 필터를 걸지 않는다
MIN_CONFIDENCE = 0.7          # 이보다 자신 없는 칸도 걸지 않는다 — 틀린 필터는 결과 0건을 만든다


def to_filters(answers):
    """LAYA 답 → 검색엔진 필터. 걸 칸은 true, 모르는 칸은 null 로 남긴다."""
    out = {}
    for qid, a in answers.items():
        if a["type"] == "noul":
            out["winter"] = (a["noul"] >= 0.5) if a["answer_confidence"] >= MIN_CONFIDENCE else None
        elif a["choice"] in EMPTY or a["answer_confidence"] < MIN_CONFIDENCE:
            out[qid] = None
        else:
            out[f"{qid}_{a['choice']}"] = True
    return out


target = sys.argv[1]
agent = laya.load("convaiinnovations/laya", subfolder="multilingual") if target == "base" else laya.load(target)
for query in ["50만원이하 패딩 상품중 갈색", "아빠 겨울 부츠 브라운으로", "언니 선물로 십만 원 이하 밤색 터틀넥"]:
    answers = agent.predict(query, QUESTIONS)["answers"]
    print(f"검색어: {query}")
    for qid, a in answers.items():
        value = f"{a['noul']:.2f}" if a["type"] == "noul" else a["choice"]
        print(f"  {qid:<9}{value:<10}확신 {a['answer_confidence']:.2f}")
    print("  filters:", json.dumps(to_filters(answers), ensure_ascii=False))

검색어: 50만원이하 패딩 상품중 갈색
  price    under100  확신 0.51
  category none      확신 0.76
  item     padding   확신 1.00
  winter   0.92      확신 0.92
  gender   unknown   확신 0.63
  color    brown     확신 0.99
  filters: {"price": null, "category": null, "item_padding": true, "winter": true, "gender": null, "color_brown": true}
검색어: 아빠 겨울 부츠 브라운으로
  price    none      확신 0.54
  category none      확신 0.83
  item     boots     확신 1.00
  winter   0.89      확신 0.89
  gender   unknown   확신 0.49
  color    none      확신 0.88
  filters: {"price": null, "category": null, "item_boots": true, "winter": true, "gender": null, "color": null}
검색어: 언니 선물로 십만 원 이하 밤색 터틀넥
  price    under10   확신 0.38
  category none      확신 0.45
  item     knit      확신 0.51
  winter   0.74      확신 0.74
  gender   unknown   확신 0.38
  color    none      확신 0.54
  filters: {"price": null, "category": null, "item": null, "winter": true, "gender": null, "color": null}