FROM nvcr.io/nvidia/physicsnemo/physicsnemo:25.06

# Install system dependencies
RUN apt-get update && apt-get install -y --no-install-recommends \
    wget \
    git \
    xvfb \
    && rm -rf /var/lib/apt/lists/*

# Upgrade pip
RUN python -m pip install --upgrade pip

# Create non-root user
RUN useradd -m -u 1001 user
USER user
ENV HOME=/home/user
RUN mkdir $HOME/.cache $HOME/.config $HOME/.local && chmod -R 777 $HOME
ENV PATH=/home/user/.local/bin:$PATH

# Set working directory
WORKDIR $HOME/app

# Create data directory with proper permissions
USER root
RUN mkdir /domino-demo && chown user:user /domino-demo && chmod 777 /domino-demo
# Add NGC CLI
RUN wget https://ngc.nvidia.com/downloads/ngccli_linux.zip
RUN unzip ngccli_linux.zip
RUN chmod u+x ngc-cli/ngc
RUN --mount=target=/root/on_startup.sh,source=on_startup.sh,readwrite \
	bash /root/on_startup.sh
USER user

# Install Python dependencies
RUN pip install --no-cache-dir numpy pyvista vtk matplotlib tqdm numpy-stl torchinfo

COPY --chown=user login.html $HOME/app/
COPY --chown=user on_startup.sh $HOME/app/
COPY --chown=user README.md $HOME/app/
COPY --chown=user start_server.sh $HOME/app/
COPY --chown=user domino-data-preprocessing.ipynb /domino-demo/
COPY --chown=user domino-training-test.ipynb /domino-demo/
COPY --chown=user introduction.ipynb /domino-demo/
COPY --chown=user outputs /domino-demo/outputs
COPY --chown=user README.md /domino-demo/

RUN chmod -R 777 /domino-demo/

RUN chmod +x start_server.sh

COPY --chown=user login.html /usr/local/lib/python3.12/dist-packages/jupyter_server/templates/login.html

# Expose port
EXPOSE 7860

# Start the server
CMD ["./start_server.sh"]