backend/app/services/classifier.py

"""경이님 6-class 분류 모델 wrapper.

문장 → Category (일정/준비물/제출/비용/건강·안전/기타).
파이프라인 [4] 단계 — 윤정님 todo 각각에 대해 호출되어 AnalyzeItem.category로 들어감.

모델 모드: "auto" — KcELECTRA 체크포인트 있으면 그쪽, 없으면 simple(TF-IDF+LogReg) 폴백.
경이님 v2 KcELECTRA 학습이 끝나면 자동으로 더 정확한 모델로 업그레이드됨.
"""
import sys
from datetime import date
from pathlib import Path

from app.models.schemas import Category

_CLF_DIR = Path("/app/external_model/classification")
if str(_CLF_DIR) not in sys.path:
    sys.path.insert(0, str(_CLF_DIR))

# 외부 마운트가 없는 환경(CI/테스트)에선 import 가드 — 모듈 로드만큼은 안전하게.
try:
    from src.predict import predict_one  # noqa: E402
except ImportError as error:
    print(f"[classifier] predict_one unavailable: {error}")
    predict_one = None


def classify_category(text: str, today: date | None = None) -> Category:
    """문장 → 6-class 카테고리. 실패/미정 시 Category.other.

    model="auto"로 호출 — 경이님 predict_one이 KcELECTRA 체크포인트 존재 여부를
    감지해 자동 선택. simple은 첫 호출 시 학습 데이터로 자가 학습 (~3초).
    """
    if not text or not text.strip():
        return Category.other
    if predict_one is None:
        return Category.other  # 모델 부재 (CI 등) — 안전한 기본값

    try:
        result = predict_one(text, model="auto", today=today, explain=False)
    except Exception as error:
        print(f"[classifier] predict_one failed: {error}")
        return Category.other

    label = result.get("category", "")
    try:
        return Category(label)
    except ValueError:
        return Category.other