# src/config.py
from pathlib import Path

# Get project root
PROJECT_ROOT = Path(__file__).resolve().parent.parent

# Data Paths
DATA_DIR = PROJECT_ROOT / "data"
ARTROOM_DIR = DATA_DIR / "artroom"
BIRD_YOLO_DIR = ARTROOM_DIR / "bird" / "yolo"
BIRD_YAML = BIRD_YOLO_DIR / "bird_data.yaml"

# Model Paths
MODEL_DIR = PROJECT_ROOT / "models"
MODEL_PATHS = {
    'yolo' : MODEL_DIR / "yolov8n.pt",
    'resnet': MODEL_DIR / "resnet18.pth",
    'resnet_head': MODEL_DIR / "resnet18_head.pkl",
    'mobilenet': MODEL_DIR / "mobilenet_v3.pth",
    'mobilevit': MODEL_DIR / "mobilevit_xxs.pth",
    'mobilevit_head': MODEL_DIR / "mobilevit_head.pkl",
    'orb_ref': MODEL_DIR / "orb_reference.pkl",
    'rce_model': MODEL_DIR / "rce_model.pkl"
}

# Training Results
TRAINING_DIR = PROJECT_ROOT / "training"