#!/usr/bin/env bash
set -euo pipefail
ROOT="${1:-.}"
MODEL_DIR="$ROOT/tools/keypose_image_sampler/models"
MODEL_PATH="$MODEL_DIR/pose_landmarker_full.task"
URL="https://storage.googleapis.com/mediapipe-models/pose_landmarker/pose_landmarker_full/float16/latest/pose_landmarker_full.task"
mkdir -p "$MODEL_DIR"
echo "Downloading MediaPipe Pose Landmarker model..."
echo "$URL"
if command -v curl >/dev/null 2>&1; then
  curl -L --fail --retry 3 --connect-timeout 20 --max-time 180 -o "$MODEL_PATH" "$URL"
elif command -v python3 >/dev/null 2>&1; then
  python3 - <<PY
from urllib.request import urlretrieve
url='$URL'
out='$MODEL_PATH'
urlretrieve(url,out)
PY
else
  echo "No curl or python3 available" >&2
  exit 2
fi
python3 "$ROOT/tools/keypose_image_sampler/v65_scripts/validate_v65_mediapipe_runtime.py" --root "$ROOT"