#!/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}

sudo docker network create "$NETWORK" >/dev/null 2>&1 || true
sudo docker rm -f "$API_NAME" "$WEB_NAME" >/dev/null 2>&1 || true

sudo docker run -d --name "$API_NAME" --network "$NETWORK" \
  --privileged --ipc=host --shm-size=32g \
  -e MINERU_MODEL_SOURCE=local \
  -e PYTHONPATH=/app:/usr/local/corex-4.4.0/lib64/python3/dist-packages \
  -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