FROM nvcr.io/nvidia/pytorch:25.04-py3

# Set up environment variables
ENV DEBIAN_FRONTEND=noninteractive \
    PYTHONUNBUFFERED=1 \
    PIP_NO_CACHE_DIR=1


RUN apt-get update && apt-get install -y --no-install-recommends \
    libgl1 \
    libglib2.0-0 \
    && rm -rf /var/lib/apt/lists/*


# Install system dependencies and ffmpeg in one layer
RUN set -eux; \
    apt-get update && apt-get install -y --no-install-recommends \
        curl \
        ca-certificates \
        xz-utils \
    && rm -rf /var/lib/apt/lists/* \
    && cd /tmp \
    && curl -L https://johnvansickle.com/ffmpeg/releases/ffmpeg-release-amd64-static.tar.xz -o ffmpeg.tar.xz \
    && tar -xf ffmpeg.tar.xz \
    && cd ffmpeg-*-static \
    && install -m 0755 ffmpeg /usr/local/bin/ffmpeg \
    && install -m 0755 ffprobe /usr/local/bin/ffprobe \
    && cd / \
    && rm -rf /tmp/ffmpeg* \
    && ffprobe -version

# Copy requirements file first (for better caching)
COPY requirements.txt /tmp/requirements.txt

# Install Python packages in optimized order
RUN pip install --no-cache-dir --upgrade pip setuptools wheel && \
    pip install --no-cache-dir --extra-index-url https://pypi.nvidia.com nvidia-dali-cuda110 && \
    pip install --no-cache-dir -r /tmp/requirements.txt && \
    rm /tmp/requirements.txt

# Set default command
CMD ["/bin/bash"]