"""
Default paths for TherA inference (relative to this repository root).

Download model weights into the `weights/` directory — see README.md.
"""

from __future__ import annotations

import sys
from pathlib import Path

PROJECT_ROOT = Path(__file__).resolve().parent
WEIGHTS_DIR = PROJECT_ROOT / "weights"

DEFAULT_CHECKPOINT = WEIGHTS_DIR / "checkpoint"
DEFAULT_MERGED_MODEL = WEIGHTS_DIR / "merged_models"
DEFAULT_PRETRAINED_SD = WEIGHTS_DIR / "stable-diffusion"
DEFAULT_REFERENCE_CACHES = WEIGHTS_DIR / "reference_caches"


def setup_project_path() -> Path:
    """Ensure TherA root is on sys.path so local packages import correctly."""
    root = str(PROJECT_ROOT)
    if root not in sys.path:
        sys.path.insert(0, root)
    return PROJECT_ROOT