Dockerfile.runpod

# RunPod Serverless ComfyUI with Wan Animate
FROM runpod/pytorch:2.1.0-py3.10-cuda11.8.0-devel-ubuntu22.04

# Install system dependencies
RUN apt-get update && apt-get install -y \
    git \
    wget \
    ffmpeg \
    libgl1-mesa-glx \
    libglib2.0-0 \
    && rm -rf /var/lib/apt/lists/*

# Set working directory
WORKDIR /comfyui

# Clone ComfyUI
RUN git clone https://github.com/comfyanonymous/ComfyUI.git . && \
    pip install -r requirements.txt

# Install ComfyUI Manager
RUN cd custom_nodes && \
    git clone https://github.com/ltdrdata/ComfyUI-Manager.git

# Install Video Helper Suite (for VHS_LoadVideo)
RUN cd custom_nodes && \
    git clone https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite.git && \
    cd ComfyUI-VideoHelperSuite && \
    pip install -r requirements.txt

# Install Wan Animate nodes (you'll need to add the correct repo)
# RUN cd custom_nodes && \
#     git clone <WAN_ANIMATE_REPO_URL> && \
#     cd <WAN_ANIMATE_DIR> && \
#     pip install -r requirements.txt

# Download required models
RUN mkdir -p models/checkpoints models/vae models/clip_vision models/unet

# Create input/output directories
RUN mkdir -p input output temp

# Copy handler for RunPod serverless
COPY handler.py /comfyui/

# Set entrypoint
CMD ["python", "handler.py"]