FROM git.sandbox.iuk.hdm-stuttgart.de/grosse/jupyterlab-datascience:4.0.12f

ARG NB_USER="jovyan"
ARG NB_UID="1000"
ARG NB_GID="100"
ENV CONDA_CUDA_OVERRIDE="12.1"  
ENV NUM_GPUS=1
USER root
RUN curl -fsSL https://ollama.com/install.sh | sh
RUN pip install git+https://git.sandbox.iuk.hdm-stuttgart.de/grosse/jupyterlab_ollama_launcher
USER $NB_UID
ENV OLLAMA_MODELS=/tmp/.ollama
ENV PGPT_PROFILES=ollama
RUN pip install open-webui==0.3.7