Dockerfile.modal

# syntax=docker/dockerfile:1.5
ARG CUDA_VERSION=12.8.1
ARG CUDA_FLAVOR=devel
ARG CUDA_OS=ubuntu22.04
FROM nvidia/cuda:${CUDA_VERSION}-${CUDA_FLAVOR}-${CUDA_OS}

ENV DEBIAN_FRONTEND=noninteractive \
    UV_PROJECT_DIR=/app \
    PYTHONUNBUFFERED=1 \
    PIP_NO_CACHE_DIR=1 \
    PYTHONPATH="/app" \
    PATH="/app/.venv/bin:$PATH"

RUN apt-get update && \
    apt-get install -y --no-install-recommends \
        software-properties-common \
        curl \
        ca-certificates && \
    add-apt-repository ppa:deadsnakes/ppa && \
    apt-get update && \
    apt-get install -y --no-install-recommends \
        python3.11 \
        python3.11-venv \
        python3.11-dev \
        python3-pip \
        build-essential \
        git \
        ffmpeg \
        libglib2.0-0 \
        libgl1 && \
    ln -sf /usr/bin/python3.11 /usr/bin/python3 && \
    ln -sf /usr/bin/python3.11 /usr/bin/python && \
    python3.11 -m pip install --upgrade pip && \
    rm -rf /var/lib/apt/lists/*

RUN curl -LsSf https://astral.sh/uv/install.sh | sh && \
    mv /root/.local/bin/uv /usr/local/bin/uv && \
    rm -rf /root/.local

WORKDIR ${UV_PROJECT_DIR}

COPY pyproject.toml ./pyproject.toml
COPY uv.lock ./uv.lock

RUN --mount=type=cache,target=/root/.cache/uv \
    uv sync --extra gpu --no-editable --frozen

COPY main.py ./main.py
COPY config.py ./config.py
COPY inference.py ./inference.py

WORKDIR /workspace