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

if [[ $# -lt 2 ]]; then
  echo "Usage: $0 INPUT OUTPUT.txt [extra mlx_ocr_pipeline.py options...]" >&2
  exit 2
fi

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
PYTHON_BIN="${PYTHON_BIN:-$ROOT_DIR/.venv/bin/python}"
MODEL_DIR="${ABJAD_MODEL_DIR:-$ROOT_DIR/models/Abjad_0.2}"
INPUT_PATH="$1"
OUTPUT_PATH="$2"
shift 2

exec "$PYTHON_BIN" "$ROOT_DIR/mlx_ocr_pipeline.py" "$INPUT_PATH" \
  -o "$OUTPUT_PATH" \
  --model "$MODEL_DIR" \
  --no-adapter \
  --preserve-digit-shapes \
  "$@"