FROM nvidia/cuda:12.8.1-base-ubuntu22.04@sha256:001469ea0f3dec85a1ca929aeea3b58ae369d4c11228b10aec1f642bb6ca7a6f

ENV DEBIAN_FRONTEND=noninteractive
RUN rm -f /etc/apt/sources.list.d/cuda*.list \
    && sed -i 's|http://archive.ubuntu.com|https://archive.ubuntu.com|g; s|http://security.ubuntu.com|https://security.ubuntu.com|g' /etc/apt/sources.list \
    && apt-get -o Acquire::ForceIPv4=true -o Acquire::Retries=2 -o Acquire::https::Timeout=45 update \
    && apt-get -o Acquire::ForceIPv4=true -o Acquire::Retries=2 -o Acquire::https::Timeout=45 install -y --no-install-recommends \
    ca-certificates libosmesa6 libgl1 libglu1-mesa libglib2.0-0 \
    libsndfile1 libgomp1 libffi8 ffmpeg fontconfig fonts-dejavu-core \
    && rm -rf /var/lib/apt/lists/*

ADD environment.tar /opt/conda/envs/hybrid/
COPY environment-inventory.json requirements.snapshot.txt /opt/codiffact/
COPY smoke_test.py /opt/codiffact/smoke_test.py
RUN ln -sfn /usr/lib/x86_64-linux-gnu/libstdc++.so.6 /opt/conda/envs/hybrid/lib/libstdc++.so.6 \
    && ln -s /opt/conda/envs/hybrid/lib/python3.12/site-packages/imageio_ffmpeg/binaries/ffmpeg-linux64-v4.2.2 /usr/local/bin/ffmpeg \
    && mkdir -p /app /models /data /outputs /jizhi/jizhi2/worker/trainer/repo \
    && ln -s /app /jizhi/jizhi2/worker/trainer/repo/hybrid-motion

ENV PATH=/usr/local/bin:/opt/conda/envs/hybrid/bin:/usr/local/sbin:/usr/sbin:/usr/bin:/sbin:/bin \
    PYOPENGL_PLATFORM=osmesa \
    PYTHONNOUSERSITE=1 \
    PYTHONDONTWRITEBYTECODE=1 \
    HF_HOME=/models/hf_cache \
    HF_HUB_OFFLINE=1 \
    TOKENIZERS_PARALLELISM=false \
    OMP_NUM_THREADS=4 \
    MKL_NUM_THREADS=4 \
    PYTHONPATH=/app/src:/app/third_party:/app/third_party/HY-Motion-1.0:/app

WORKDIR /app
RUN apt-get -o Acquire::ForceIPv4=true -o Acquire::Retries=2 -o Acquire::https::Timeout=45 update \
    && apt-get -o Acquire::ForceIPv4=true -o Acquire::Retries=2 -o Acquire::https::Timeout=45 install -y --no-install-recommends git git-lfs zip unzip \
    && rm -rf /var/lib/apt/lists/*
RUN python /opt/codiffact/smoke_test.py
LABEL org.opencontainers.image.title="CoDiffAct environment only" \
      org.opencontainers.image.description="Python 3.12 / PyTorch 2.7.1 cu128; no project, pretrained weights, datasets or credentials"
CMD ["bash"]