#!/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_BACKEND=${OCR_BACKEND:-tesseract}   # tesseract|paddle|auto
export OCR_LANG=${OCR_LANG:-eng+rus}
export OCR_AUTO_LANG=${OCR_AUTO_LANG:-true}
export OCR_CONFIG=${OCR_CONFIG:---oem 1 --psm 6}
export PADDLE_LANG=${PADDLE_LANG:-ru}
export PADDLE_USE_GPU=${PADDLE_USE_GPU:-false}
export PADDLE_USE_ANGLE_CLS=${PADDLE_USE_ANGLE_CLS:-false}
export PADDLE_DET_MODEL_DIR=${PADDLE_DET_MODEL_DIR:-./models/paddle/PP-OCRv3_mobile_det_infer}
export PADDLE_REC_MODEL_DIR=${PADDLE_REC_MODEL_DIR:-./models/paddle/cyrillic_PP-OCRv3_mobile_rec_infer}
export PADDLE_CLS_MODEL_DIR=${PADDLE_CLS_MODEL_DIR:-}
export PADDLE_CPU_THREADS=${PADDLE_CPU_THREADS:-4}
# 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"