FROM python:3.10-slim

# System dependencies
RUN apt-get update && apt-get install -y \
    git wget libgl1 libglib2.0-0 libgl1-mesa-glx ffmpeg \
    && apt-get clean

# Install ComfyUI
WORKDIR /app
RUN git clone https://github.com/comfyanonymous/ComfyUI.git .
RUN pip install --upgrade pip
RUN pip install -r requirements.txt

# Models folder structure
RUN mkdir -p models/checkpoints models/loras models/controlnet models/upscale_models

# Expose port
EXPOSE 7860

CMD ["python", "main.py", "--listen", "--port", "7860"]