FROM python:3.12-slim-bookworm

ARG YUE2_COMMIT=9c6c4b349be978b06a9d0d958471a07a6cdeff4d
ARG YUE2_WEBUI_COMMIT=8fc05609bde5dcd345d7c6d57fff3da0839164c5

RUN apt-get update \
    && apt-get install -y --no-install-recommends ca-certificates git libsndfile1 \
    && rm -rf /var/lib/apt/lists/*

RUN git clone https://github.com/multimodal-art-projection/YuE.git /opt/yue2 \
    && git -C /opt/yue2 checkout "${YUE2_COMMIT}" \
    && python -m pip install --no-cache-dir /opt/yue2

# The community console is intentionally layered onto the already verified,
# pinned YuE2 runtime.  We use only its webui/ directory; its bundled YuE2 fork
# and Windows installers never become part of the Athena image.
RUN git clone https://github.com/Ladypoly/YuE2_WebUI.git /tmp/yue2-webui \
    && git -C /tmp/yue2-webui checkout "${YUE2_WEBUI_COMMIT}" \
    && cp -a /tmp/yue2-webui/webui /opt/yue2/webui \
    && cp /tmp/yue2-webui/LICENSE /opt/yue2/YuE2_WebUI-LICENSE \
    && python -m pip install --no-cache-dir -r /opt/yue2/webui/requirements.txt \
    && rm -rf /tmp/yue2-webui

COPY community-webui-patches /tmp/community-webui-patches
RUN python /tmp/community-webui-patches/patch_webui.py /opt/yue2/webui \
    && rm -rf /tmp/community-webui-patches

COPY ui /opt/yue2-playground

WORKDIR /workspace
ENV PYTHONUNBUFFERED=1 \
    YUE2_KIT=/workspace

ENTRYPOINT ["yue2"]