#R-Shiny template https://huggingface.co/spaces/posit/shiny-for-r-template/tree/main
#Py-shiny template https://huggingface.co/spaces/posit/gptneox-chat

#############################
# RSTUDIO-SERVER            #
# VOLUME DOCKERFILE         #
#***************************#
# Set up a jupyter-lab for  #
# developing using deepvats #
#############################

##############
# Base image #
##############
# Adds rstudio server, tiyverse, devtools to rocker/cuda
FROM misantamaria/dvats-rstudio:0

SHELL [ "/bin/bash", "--login", "-c" ]

###############
# Ensure vars #
###############
RUN echo "--> ARGS"
ARG USER_NAME=user \
    USER_ID=1000 \
    GROUP_ID \
    WANDB_ENTITY \
    WANDB_PROJECT \


RUN echo "--> ARGS -> ENV"
ENV USER_NAME=$USER_NAME \
    USER_ID=$USER_ID \
    GROUP_ID=$GROUP_ID \
    WANDB_ENTITY=$WANDB_ENTITY \
    WANDB_PROJECT=$WANDB_PROJECT

RUN echo "--> Setup ENV variables"
ENV RETICULATE_PYTHON_ENV=/home/env \
    ROOT=TRUE \
    RETICULATE_MINICONDA_PATH=/home//miniconda \
    CUDA_VISIBLE_DEVICES=0,1,2 \
    ENV_VARS=WANDB_ENTITY,WANDB_PROJECT,USER_NAME,USER_ID,GROUP_ID,PASSWORD,ROOT,CUDA_VISIBLE_DEVICES\
    HOME="/home/"


### Copy deepvats
RUN echo "---> Setup volumes" 
#--- Volumes
COPY --chown=${USER_ID}:${GROUP_ID} ../r_shiny_app $HOME/${USER_NAME}app
COPY --chown=${USER_ID}:${GROUP_ID} ${LOCAL_DATA_PATH} $HOME/${USER_NAME}data/
COPY --chown=${USER_ID}:${GROUP_ID} ../dvats ${HOME}/${USER_NAME}/dvats

# Install deepvats

RUN pip install -e  /home/work
RUN echo "--> Go!"

CMD ["R", "--quiet", "-e", "shiny::runApp(host='0.0.0.0', port=${RSTUDIO_PORT:-3838}, launch.browser = FALSE, appDir = '/home/${USER_NAME}/app')"]