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

ENV DEBIAN_FRONTEND=noninteractive \
    TZ=UTC \
    DISPLAY=:1 \
    VSCODE_PASSWORD=changeme \
    JUPYTER_TOKEN=changeme \
    PATH=/home/user/.local/bin:$PATH

RUN apt-get update && apt-get install -y --no-install-recommends \
    bash \
    curl \
    wget \
    ca-certificates \
    procps \
    git \
    git-lfs \
    python3 \
    python3-pip \
    python3-venv \
    nginx \
    supervisor \
    xvfb \
    x11vnc \
    novnc \
    websockify \
    xfce4 \
    xfce4-terminal \
    dbus-x11 \
    xterm \
    fonts-dejavu \
 && rm -rf /var/lib/apt/lists/* \
 && git lfs install

RUN curl -fsSL https://code-server.dev/install.sh | sh

RUN useradd -m -u 1000 -s /bin/bash user \
 && mkdir -p /app /home/user/.config /home/user/.local/share \
 && chown -R user:user /app /home/user

WORKDIR /app

COPY --chown=user requirements.txt /app/requirements.txt
RUN pip3 install --no-cache-dir --upgrade pip \
 && pip3 install --no-cache-dir -r /app/requirements.txt

COPY --chown=user app.py /app/app.py
COPY --chown=user index.html /app/index.html
COPY nginx.conf /etc/nginx/nginx.conf
COPY supervisord.conf /etc/supervisor/conf.d/supervisord.conf

EXPOSE 7860

CMD ["/usr/bin/supervisord", "-c", "/etc/supervisor/conf.d/supervisord.conf"]