"""Model-weight discovery across the new data/models layout."""
from __future__ import annotations

from pathlib import Path

from saqr.core.paths import MODELS_DIR, PROJECT_ROOT


def resolve_model_path(model_arg: str) -> Path:
    """Find weights: arg -> PROJECT_ROOT/arg -> data/models/<basename>."""
    p = Path(model_arg)
    if p.exists():
        return p
    p = PROJECT_ROOT / model_arg
    if p.exists():
        return p
    p = MODELS_DIR / Path(model_arg).name
    if p.exists():
        return p
    raise FileNotFoundError(f"Model not found: {model_arg}")