ARG BASE_IMAGE=runpod/worker-comfyui:5.1.0-base

FROM ${BASE_IMAGE}

# Pin ComfyUI and custom node versions.
ARG COMFYUI_VERSION=v0.8.2
ARG IMPACT_PACK_VERSION=8.28.2
ARG IMPACT_SUBPACK_VERSION=1.3.5

RUN rm -rf /comfyui \
    && /usr/bin/yes | comfy --workspace /comfyui install --version "${COMFYUI_VERSION}" --nvidia \
    && comfy set-default /comfyui

# FaceDetailer nodes (Impact Pack + Subpack).
RUN comfy-node-install \
    comfyui-impact-pack@"${IMPACT_PACK_VERSION}" \
    comfyui-impact-subpack@"${IMPACT_SUBPACK_VERSION}"

RUN set -e; \
    impact_pack_dir="/comfyui/custom_nodes/comfyui-impact-pack"; \
    impact_subpack_dir="/comfyui/custom_nodes/comfyui-impact-subpack"; \
    if [ ! -d "$impact_pack_dir" ] || [ ! -d "$impact_subpack_dir" ]; then \
      echo "Impact Pack/Subpack not found under /comfyui/custom_nodes"; \
      ls -la /comfyui/custom_nodes; \
      exit 1; \
    fi; \
    if [ -f "$impact_pack_dir/requirements.txt" ]; then \
      python -m pip install -r "$impact_pack_dir/requirements.txt"; \
    else \
      echo "No requirements.txt in $impact_pack_dir; skipping dependency install."; \
    fi; \
    if [ -f "$impact_subpack_dir/requirements.txt" ]; then \
      python -m pip install -r "$impact_subpack_dir/requirements.txt"; \
    else \
      echo "No requirements.txt in $impact_subpack_dir; skipping dependency install."; \
    fi

# Ensure ComfyUI can see models stored on the RunPod network volume.
COPY extra_model_paths.yaml /comfyui/extra_model_paths.yaml