Dockerfile.webui

# syntax=docker/dockerfile:1.7
# Thin overlay on the existing Blackwell runtime image.
# Bakes web UI, entrypoint, inference patches, and missing runtime deps
# without rebuilding Torch / TE / flash-attn from scratch.
ARG BASE_IMAGE=unirelight-blackwell:cu129
FROM ${BASE_IMAGE}

USER root
WORKDIR /workspace/UniRelight

COPY scripts/patches /opt/unirelight-patches
COPY scripts/apply_unirelight_patches.sh /usr/local/bin/apply_unirelight_patches.sh
COPY scripts/entrypoint.sh /usr/local/bin/entrypoint.sh
COPY scripts/te_attention_compat.py /tmp/te_attention_compat.py
COPY webui /opt/unirelight-web/webui
COPY requirements.web.txt /tmp/requirements.web.txt

RUN sed -i 's/\r$//' /usr/local/bin/entrypoint.sh /usr/local/bin/apply_unirelight_patches.sh /tmp/te_attention_compat.py \
    && chmod +x /usr/local/bin/entrypoint.sh /usr/local/bin/apply_unirelight_patches.sh \
    && mkdir -p /opt/unirelight-web \
    && touch /opt/unirelight-web/webui/__init__.py \
    && bash /usr/local/bin/apply_unirelight_patches.sh \
    && pip install --no-cache-dir -r /tmp/requirements.web.txt ffmpegcv \
    && (python /tmp/te_attention_compat.py || true) \
    && (pip install --no-build-isolation "git+https://github.com/NVlabs/nvdiffrast.git" || echo "WARN: nvdiffrast skipped") \
    && mkdir -p /data/input /data/hdri /data/outputs /data/models /data/checkpoints /data/jobs /data/frames

ENV PYTHONPATH=/workspace/UniRelight:/opt/unirelight-web \
    UNIRELIGHT_DATA=/data \
    UNIRELIGHT_HOST=0.0.0.0 \
    UNIRELIGHT_PORT=8080 \
    OPENCV_IO_ENABLE_OPENEXR=1

EXPOSE 8080
VOLUME ["/root/.cache/huggingface"]
ENTRYPOINT ["/usr/local/bin/entrypoint.sh"]
CMD []