#!/usr/bin/env bash
# Download DROID/LIBERO UWM checkpoints from Google Drive.
# See README: https://drive.google.com/drive/folders/1M4AuVLMRpSwOf_YAp56bV9AqyZI9ul6g
set -e
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
REPO_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
CHECKPOINT_DIR="${REPO_ROOT}/checkpoints"
FOLDER_ID="1M4AuVLMRpSwOf_YAp56bV9AqyZI9ul6g"

mkdir -p "$CHECKPOINT_DIR"
cd "$CHECKPOINT_DIR"

if command -v gdown &>/dev/null; then
  echo "Downloading checkpoints with gdown..."
  gdown --folder "https://drive.google.com/drive/folders/${FOLDER_ID}" --remaining-ok || true
else
  echo "Install gdown: pip install gdown"
  echo "Then run: gdown --folder https://drive.google.com/drive/folders/${FOLDER_ID} -O ${CHECKPOINT_DIR}"
  echo "Or download manually from: https://drive.google.com/drive/folders/${FOLDER_ID}"
fi

echo "Checkpoints (if downloaded) are in: $CHECKPOINT_DIR"