#!/bin/bash

set -e

cd "$(dirname "$0")"

echo "Initializing dust3r submodule..."
if [ ! -d "dust3r/.git" ]; then
  git submodule add -f https://github.com/camenduru/dust3r.git dust3r 2>/dev/null || true
  git submodule update --init --recursive
fi

echo "Installing Python dependencies..."
python -m pip install -r requirements.txt

if [[ "$(uname)" == "Linux" ]]; then
  python -m pip install https://github.com/camenduru/wheels/releases/download/colab/curope-0.0.0-cp310-cp310-linux_x86_64.whl
fi

mkdir -p dust3r/checkpoints

WEIGHTS=dust3r/checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth
if [ ! -f "$WEIGHTS" ]; then
  echo "Downloading model weights..."
  wget https://huggingface.co/camenduru/dust3r/resolve/main/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth \
    -P dust3r/checkpoints
fi

echo "Setup complete."