# syntax=m.daocloud.io/docker.io/docker/dockerfile:1.7
# ---------------------------------------------------------------------------
# 禁止「FROM python:...」简写 — 会直连 docker.io / registry-1.docker.io。
# Python 使用 DaoCloud 增加前缀形式：https://github.com/DaoCloud/public-image-mirror
# 勿用裸 tag「3.13」；须 3.13-slim-bookworm。
# ---------------------------------------------------------------------------
FROM m.daocloud.io/docker.io/library/python:3.13-slim-bookworm

# Debian bookworm: use Aliyun mirror for apt (default snapshot points at deb.debian.org).
RUN sed -i \
  -e 's|http://deb.debian.org/debian-security|https://mirrors.aliyun.com/debian-security|g' \
  -e 's|http://deb.debian.org/debian|https://mirrors.aliyun.com/debian|g' \
  /etc/apt/sources.list.d/debian.sources

# OpenCV / MediaPipe (doctor pose) need GLVND + Mesa GLES/EGL in slim images; omit X11/GUI stack.
RUN apt-get update && apt-get install -y --no-install-recommends \
    ca-certificates \
    curl \
    docker.io \
    ffmpeg \
    fontconfig \
    fonts-noto-cjk \
    fonts-wqy-microhei \
    libegl-mesa0 \
    libegl1 \
    libgbm1 \
    libgl1 \
    libgl1-mesa-dri \
    libglx-mesa0 \
    libgles2 \
    libglib2.0-0 \
    libgomp1 \
    libxcb1 \
    && fc-cache -fv \
    && ldconfig \
    && ldconfig -p | grep -F 'libGLESv2.so.2' >/dev/null \
    && rm -rf /var/lib/apt/lists/*

# ghcr.io：「增加前缀」形式（与 kindest/node 示例一致）
COPY --from=m.daocloud.io/ghcr.io/astral-sh/uv:latest /uv /usr/local/bin/uv

WORKDIR /app

ENV PYTHONUNBUFFERED=1 \
    UV_HTTP_TIMEOUT=600 \
    UV_LINK_MODE=copy \
    TORCH_HOME=/app/.cache/torch \
    YOLO_CONFIG_DIR=/app/.cache/ultralytics

COPY pyproject.toml uv.lock main.py alembic.ini ./
COPY scripts ./scripts/
COPY app ./app/
COPY alembic ./alembic/
# 离线批处理 / demo 直调 algorithm_subprocesses/5.15/main.py（含 weights/）
COPY algorithm_subprocesses ./algorithm_subprocesses/
# Bake runtime patches/assets so non-root api never writes the read-only bundle tree.
COPY app/algorithm_runner/actionformer_release/libs/utils/nms.py \
     algorithm_subprocesses/5.15/code/actionformer_release/libs/utils/nms.py
RUN mkdir -p algorithm_subprocesses/5.15/doctor_identity_package/.mediapipe_models && \
    curl -fsSL --retry 3 \
      -o algorithm_subprocesses/5.15/doctor_identity_package/.mediapipe_models/pose_landmarker_lite.task \
      "https://storage.googleapis.com/mediapipe-models/pose_landmarker/pose_landmarker_lite/float16/1/pose_landmarker_lite.task" && \
    test -s algorithm_subprocesses/5.15/doctor_identity_package/.mediapipe_models/pose_landmarker_lite.task

# uv.lock pins uv.agentsmirror.com artifact URLs. Rewrite to mainland mirrors (same /packages/... paths).
# PyPI: Tsinghua  |  PyTorch wheel index: 南大 (syncs download.pytorch.org / download-r2)
RUN sed -i \
  -e 's|https://uv.agentsmirror.com/pypi/files/files.pythonhosted.org|https://pypi.tuna.tsinghua.edu.cn|g' \
  -e 's|https://uv.agentsmirror.com/pypi/simple|https://pypi.tuna.tsinghua.edu.cn/simple|g' \
  -e 's|https://download-r2.pytorch.org|https://mirrors.nju.edu.cn/pytorch|g' \
  -e 's|https://download.pytorch.org|https://mirrors.nju.edu.cn/pytorch|g' \
  uv.lock

ENV UV_DEFAULT_INDEX=https://pypi.tuna.tsinghua.edu.cn/simple

# VideoSwin (Swin3D-T) hub weights: bake at build time so RTSP batch jobs never hit pytorch.org at runtime.
# Domestic PyPI mirrors (NJU/Tsinghua/Aliyun) only sync pip wheels, not /models/*.pth; default tries
# uv.agentsmirror.com (same ecosystem as uv.lock) then download.pytorch.org. Optional offline bake:
#   backend/weights/swin3d_t-7615ae03.pth  (see weights/.gitkeep)
# Override: --build-arg PYTORCH_MODELS_URL=... or PYTORCH_MODELS_MIRROR=...
ARG PYTORCH_MODELS_MIRROR=
ARG PYTORCH_MODELS_URL=
ENV PYTORCH_MODELS_MIRROR=${PYTORCH_MODELS_MIRROR} \
    PYTORCH_MODELS_URL=${PYTORCH_MODELS_URL}
# Optional offline weight (only .gitkeep by default; add swin3d_t-7615ae03.pth before build if needed).
COPY weights ./weights/

RUN --mount=type=cache,target=/root/.cache/uv \
    uv sync --frozen --no-dev --no-compile --refresh-package numpy --refresh-package mediapipe && \
    .venv/bin/python -c "import alembic" && \
    .venv/bin/python -c "import numpy; import numpy.lib._index_tricks_impl" && \
    .venv/bin/python -c "import mediapipe as mp; print('mediapipe', mp.__version__)" && \
    mkdir -p /app/.cache/ultralytics && \
    PYTORCH_MODELS_LOCAL_PATH=/app/weights/swin3d_t-7615ae03.pth \
      .venv/bin/python scripts/bake_torch_hub_checkpoint.py && \
    rm -rf /app/weights && \
    TORCH_HOME=/app/.cache/torch .venv/bin/python -c "from torchvision.models.video import Swin3D_T_Weights, swin3d_t; swin3d_t(weights=Swin3D_T_Weights.KINETICS400_V1); print('swin3d_t cached ok')" && \
    chmod -R a+rX /app/.venv /app/algorithm_subprocesses /app/.cache/torch /app/.cache/ultralytics

ENV PATH="/app/.venv/bin:$PATH"

EXPOSE 8000

CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]