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

NETWORK=${NETWORK:-unis-ai}
API_NAME=${API_NAME:-mineru-api}
WEB_NAME=${WEB_NAME:-mindmap-web}
WEB_PORT=${WEB_PORT:-9898}
MINERU_API_URL=${MINERU_API_URL:-http://${API_NAME}:8000}
MINERU_GPU_MEMORY_UTILIZATION=${MINERU_GPU_MEMORY_UTILIZATION:-0.5}
MINERU_MODEL_DIR=${MINERU_MODEL_DIR:-/data/unis/mineru-models}
MINERU_PYTHONPATH=${MINERU_PYTHONPATH:-/app:/usr/local/corex-4.4.0/lib64/python3/dist-packages}

sudo docker network create "$NETWORK" >/dev/null 2>&1 || true
sudo docker rm -f "$API_NAME" "$WEB_NAME" >/dev/null 2>&1 || true
sudo mkdir -p "$MINERU_MODEL_DIR"/pipeline "$MINERU_MODEL_DIR"/vlm
sudo chmod -R 777 "$MINERU_MODEL_DIR"

sudo docker run -d --name "$API_NAME" --network "$NETWORK" \
  --privileged --ipc=host --shm-size=32g \
  -v "${MINERU_MODEL_DIR}:/models" \
  -e MINERU_MODEL_SOURCE="${MINERU_MODEL_SOURCE:-local}" \
  -e MINERU_MODEL_DIR=/models \
  -e MINERU_PIPELINE_MODEL_DIR=/models/pipeline \
  -e MINERU_VLM_MODEL_DIR=/models/vlm \
  -e PYTHONPATH="$MINERU_PYTHONPATH" \
  -e MINERU_GPU_MEMORY_UTILIZATION="$MINERU_GPU_MEMORY_UTILIZATION" \
  unis-mindmap-api:latest \
  mineru-api --host 0.0.0.0 --port 8000

sudo docker run -d --name "$WEB_NAME" --network "$NETWORK" \
  -p "${WEB_PORT}:9898" \
  -e MINERU_API_URL="$MINERU_API_URL" \
  unis-mindmap-web:latest