FROM nvcr.io/nvidia/pytorch:23.10-py3

RUN apt update && apt install software-properties-common -y
RUN add-apt-repository ppa:deadsnakes/ppa -y
RUN apt update && DEBIAN_FRONTEND=noninteractive apt install curl python3.11-full python3.11-distutils python3.11-venv -y

ENV POETRY_HOME="/usr/local"
RUN curl -sSL https://install.python-poetry.org | python3.11 -;

WORKDIR /app
COPY pyproject.toml /app
RUN poetry env use /usr/bin/python3.11
RUN poetry install
RUN poetry run playwright install firefox && poetry run playwright install-deps

COPY . /app

CMD /bin/sh