#!/usr/bin/env bash
set -euo pipefail

MODEL_DIR="${1:-/var/vigilar/models}"
mkdir -p "$MODEL_DIR"

PB_URL="https://raw.githubusercontent.com/opencv/opencv_extra/master/testdata/dnn/ssd_mobilenet_v2_coco_2018_03_29.pb"
PBTXT_URL="https://raw.githubusercontent.com/opencv/opencv_extra/master/testdata/dnn/ssd_mobilenet_v2_coco_2018_03_29.pbtxt"

echo "Downloading MobileNet-SSD v2 model..."
if command -v curl &>/dev/null; then
    curl -fSL -o "$MODEL_DIR/mobilenet_ssd_v2.pb" "$PB_URL"
    curl -fSL -o "$MODEL_DIR/mobilenet_ssd_v2.pbtxt" "$PBTXT_URL"
elif command -v wget &>/dev/null; then
    wget -q -O "$MODEL_DIR/mobilenet_ssd_v2.pb" "$PB_URL"
    wget -q -O "$MODEL_DIR/mobilenet_ssd_v2.pbtxt" "$PBTXT_URL"
else
    echo "ERROR: curl or wget required"
    exit 1
fi

echo "Model downloaded to $MODEL_DIR"
ls -lh "$MODEL_DIR"/mobilenet_ssd_v2.*