FROM nvidia/cuda:12.3.2-cudnn9-devel-ubuntu22.04

ENV DEBIAN_FRONTEND=noninteractive

RUN apt-get update && apt-get install -y \
    wget \
    build-essential \
    libssl-dev \
    zlib1g-dev \
    libbz2-dev \
    libreadline-dev \
    libsqlite3-dev \
    libffi-dev \
    libncursesw5-dev \
    xz-utils \
    tk-dev \
    liblzma-dev \
    git \
    ca-certificates \
    curl \
    ffmpeg \
    && rm -rf /var/lib/apt/lists/*

WORKDIR /tmp
RUN wget https://www.python.org/ftp/python/3.12.7/Python-3.12.7.tgz \
    && tar -xvf Python-3.12.7.tgz \
    && cd Python-3.12.7 \
    && ./configure --enable-optimizations --prefix=/usr/local \
    && make -j$(nproc) \
    && make altinstall \
    && cd .. \
    && rm -rf Python-3.12.7 Python-3.12.7.tgz

RUN python3.12 -m ensurepip --upgrade
RUN python3.12 -m pip install --upgrade pip

RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user
ENV PATH="$HOME/.local/bin:$PATH"

WORKDIR /app

COPY --chown=user ./requirements.txt requirements.txt
RUN python3.12 -m pip install --no-cache-dir --upgrade -r requirements.txt

COPY --chown=user . /app

CMD ["python3.12", "app.py"]