FROM python:3.11-slim

WORKDIR /app

RUN apt-get update && apt-get install -y \
    git \
    wget \
    curl \
    ffmpeg \
    && rm -rf /var/lib/apt/lists/*

RUN git clone https://github.com/comfyanonymous/ComfyUI.git
RUN git clone https://github.com/Comfy-Org/ComfyUI-Manager.git custom_nodes/ComfyUI-Manager

WORKDIR /app/ComfyUI

# Force CPU PyTorch
RUN pip install --no-cache-dir \
    torch torchvision torchaudio \
    --index-url https://download.pytorch.org/whl/cpu

RUN pip install --no-cache-dir -r requirements.txt

RUN mkdir -p models/checkpoints
RUN mkdir -p models/text_encoders
RUN ls -R /data || true

CMD sh -c "cp -f /data/models/checkpoints/* /app/ComfyUI/models/checkpoints/ 2>/dev/null || true; cp -f /data/models/text_encoders/* /app/ComfyUI/models/text_encoders/ 2>/dev/null || true; cp -f /data/models/loras/* /app/ComfyUI/models/loras/ 2>/dev/null || true; cp -f /data/custom_nodes/* /app/ComfyUI/custom_nodes/ 2>/dev/null || true; python main.py --cpu --listen 0.0.0.0 --port 7860"