FROM docker.io/ubuntu:noble

RUN apt update && \
    apt full-upgrade --auto-remove --purge --yes && \
    apt install --no-install-recommends --yes \
    build-essential \
    git \
    libatomic1 \
    libegl-dev \
    libegl1 \
    libgbm1 \
    libgl1 \
    libgl1-mesa-dev \
    libgl1-mesa-dri \
    libgles2 \
    libglvnd0 \
    libglx0 \
    libnuma-dev \
    libopengl0 \
    mesa-utils \
    mesa-utils-extra \
    python3 \
    python3-dev \
    python3-venv && \
    rm -rf /var/lib/apt/lists/*

ENV PYTHONDONTWRITEBYTECODE=1
ENV PYTHONUNBUFFERED=1

RUN mkdir -p /app/ComfyUI && chown -R ubuntu: /app

USER ubuntu

RUN python3 -m venv /app/venv

RUN /app/venv/bin/pip3 install \
    --index-url https://download.pytorch.org/whl/rocm7.14 \
    torch \
    torchvision
    #torchaudio \

WORKDIR /app/ComfyUI

RUN git clone https://github.com/Comfy-Org/ComfyUI.git . && \
    git checkout v0.43.0

RUN /app/venv/bin/pip install \
  --pre \
  --upgrade \
  -r requirements.txt \
  comfy-aimdo \
  comfyui-manager \
  gguf \
  PyOpenGL-accelerate

ENTRYPOINT ["/app/venv/bin/python3", "main.py"]