from __future__ import annotations

from pathlib import Path

import mediapipe as mp

BaseOptions = mp.tasks.BaseOptions

_MODELS_DIR = Path(__file__).resolve().parent.parent.parent / "pose_models"

POSE_LANDMARKER_FULL = _MODELS_DIR / "pose_landmarker_full.task"
POSE_LANDMARKER_LITE = _MODELS_DIR / "pose_landmarker_lite.task"

DEFAULT_MODEL_PATH = str(POSE_LANDMARKER_FULL) if POSE_LANDMARKER_FULL.exists() else str(POSE_LANDMARKER_LITE)