FROM docker.io/library/ubuntu:22.04

ENV DEBIAN_FRONTEND noninteractive

RUN apt-get update \
	    && apt-get install -y --no-install-recommends apt-utils build-essential g++ curl cmake zlib1g-dev libjpeg-dev xvfb xorg-dev libboost-all-dev libsdl2-dev swig python3 python3-dev python3-future python3-pip python3-setuptools python3-wheel python3-tk libatlas-base-dev cython3 \
	&& apt-get clean \
	&& rm -rf /var/lib/apt/lists/*

RUN python3 -m pip install --upgrade pip \
	&& python3 -m pip install jupyterlab keras==3.3.3 matplotlib==3.9.0 numpy==1.26.4 tensorflow==2.16.1 scikit-image==0.22.0 \
	&& python3 -m pip install "gymnasium[box2d]==0.29.1" "stable-baselines3[extra]==2.3.2"

RUN apt-get update && apt-get install -y wget

WORKDIR /work
COPY . /work
RUN /work/download-data.sh \
	&& rm /work/*_solutions.ipynb

ENV DEBIAN_FRONTEND teletype

CMD xvfb-run -s "-screen 0 1400x900x24" \
	/usr/local/bin/jupyter lab --port 8888 --ip=0.0.0.0 --allow-root