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

INDEX_CSV="${1:?Usage: $0 INDEX_CSV OUTPUT_DIR [GPU]}"
OUTPUT_DIR="${2:?Usage: $0 INDEX_CSV OUTPUT_DIR [GPU]}"
GPU="${3:-0}"

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
BUNDLE="$(cd "${SCRIPT_DIR}/.." && pwd)"
PYTHON="${PYTHON:-python}"
SAM3_CKPT="${SAM3_CHECKPOINT:-${BUNDLE}/model/sam3_base.pt}"

cd "${BUNDLE}"
CUDA_VISIBLE_DEVICES="${GPU}" PYTHONNOUSERSITE=1 "${PYTHON}" -u \
  "${SCRIPT_DIR}/evaluate_sam3_test_text_prompt_segmentation.py" \
  --index_csv "${INDEX_CSV}" \
  --sam3_model_path "${SAM3_CKPT}" \
  --checkpoint_path "${BUNDLE}/model/best_model.pt" \
  --encoder_trainable lora \
  --lora_rank 8 \
  --lora_alpha 16 \
  --prompts "breast lesion" \
  --output_dir "${OUTPUT_DIR}" \
  --overwrite