# FROM nvcr.io/nvidia/clara-train-sdk:v4.1
FROM nvcr.io/nvidia/pytorch:22.08-py3


# apt
RUN apt-get update -y \
&&  apt-get install libxrender1 openssh-server pigz sudo  -y

#conda
# RUN conda update -y -n base conda
# RUN conda install -y -n base conda-libmamba-solver
# RUN conda config -y --set solver libmamba

# pip
# RUN pip install --upgrade pip

# RUN pip install antspyx itk-elastix nipype nnunet rt_utils
RUN pip install antspyx nipype nnunet nnunetv2 simpleitk==2.3.1 tensorflow # too large .... install first
# RUN pip install itk-elastix==0.13.0
# RUN pip install itk-elastix
# RUN pip install --upgrade git+https://github.com/FabianIsensee/hiddenlayer.git@more_plotted_details#egg=hiddenlayer

# RUN pip install --upgrade git+https://github.com/FabianIsensee/hiddenlayer.git
# RUN pip install --upgrade git+https://github.com/rohitrango/fireants.git
# RUN pip install git+https://github.com/qurit/rt-utils.git@5bab9ffcc8fe19dd775e940afdc3d8f48f869150          # fix FrameOfReferenceUID

# RUN pip install git+https://github.com/adalca/neurite.git git+https://github.com/freesurfer/surfa.git git+https://github.com/voxelmorph/voxelmorph.git

# WORKDIR /workspace
WORKDIR /123
COPY requirements.txt /123
RUN pip install -r requirements.txt

RUN wget https://github.com/rordenlab/dcm2niix/releases/download/v1.0.20240202/dcm2niix_lnx.zip
RUN unzip dcm2niix_lnx.zip
RUN cp dcm2niix /usr/bin

# nnUNet

ENV nnUNet_raw_data_base="/workspace/nnUNet_raw_data_base"
ENV nnUNet_preprocessed="/workspace/nnUNet_preprocessed"
ENV RESULTS_FOLDER="/workspace/nnUNet_trained_models"

ENV nnUNet_raw="/workspace/v2/nnUNet_raw"
ENV nnUNet_preprocessed="/workspace/v2/nnUNet_preprocessed"
ENV nnUNet_results="/workspace/v2/nnUNet_results"

# SSH server
#RUN echo 'root:password' | chpasswd
#RUN echo "PasswordAuthentication yes" >> /etc/ssh/sshd_config
#RUN echo "PermitRootLogin yes" >> /etc/ssh/sshd_config
#ENTRYPOINT service ssh restart && env >> /etc/environment && bash
#EXPOSE 22

# Masonite
EXPOSE 8000

# jupyter
ENTRYPOINT jupyter-lab
EXPOSE 8888

# pynetdicom
EXPOSE 11120