#!/usr/bin/env bash
# Runner for image2text service on Jetson Nano
# Adjust env vars below as needed.

set -euo pipefail

# ---- Configurable env ----
export MODEL_EDGETPU_PATH=${MODEL_EDGETPU_PATH:-./models/ssd_mobilenet_v2_coco_quant_postprocess_edgetpu.tflite}
export MODEL_LABELS_PATH=${MODEL_LABELS_PATH:-./models/coco_labels.txt}
export MODEL_EAST_PATH=${MODEL_EAST_PATH:-./models/frozen_east_text_detection.pb}
export OCR_LANG=${OCR_LANG:-eng+rus}
export OCR_AUTO_LANG=${OCR_AUTO_LANG:-true}
export OCR_CONFIG=${OCR_CONFIG:---oem 1 --psm 6}
# Point to tessdata_best if installed
export TESSDATA_PREFIX=${TESSDATA_PREFIX:-}
export MAX_IMAGE_MB=${MAX_IMAGE_MB:-10}
export CONF_THRESHOLD=${CONF_THRESHOLD:-0.4}
export RETURN_BOXES=${RETURN_BOXES:-true}
# Enable CUDA DNN for EAST if your OpenCV is built with CUDA
export USE_CUDA_DNN=${USE_CUDA_DNN:-false}

HOST=${HOST:-0.0.0.0}
PORT=${PORT:-8000}

python3 app.py --host "$HOST" --port "$PORT"