# -------------------------------------------------------
# Relies on JUPYTER_PORT being set in docker-compose.yml
# -------------------------------------------------------
FROM pytorch/pytorch:latest


# Install packages
RUN pip install jupyterlab ipywidgets
RUN pip install pandas numpy
RUN pip install plotly
RUN pip install seaborn
RUN pip install statsmodels
#
RUN pip install autogluon
#
RUN pip install scikit-learn
RUN pip install Arch
RUN pip install Quantlib
RUN pip install Backtrader
RUN pip install pyfolio
RUN pip install XGBoost

# Install additional applications
RUN apt update
RUN apt-get install graphviz -y

RUN mkdir -p /workspace

# Set environment variables, etc.
#ENV MY_ENV_VAR=myvalue
COPY cvttpy /cvttpy

# Set the working directory
WORKDIR /workspace

# Expose the port for JupyterLab
EXPOSE ${JUPYTER_PORT}

# Command to run JupyterLab
CMD ["jupyter-lab", "--ip=0.0.0.0", "--port=8888", "--no-browser", "--allow-root", "--NotebookApp.token=''"]