FROM ghcr.io/prefix-dev/pixi:latest AS build

COPY config.toml /root/.config/pixi/config.toml
COPY pyproject.toml pixi.lock /app/
WORKDIR /app
RUN pixi install --frozen -e cuda-base
COPY dist/*.whl /app/
RUN /app/.pixi/envs/cuda-base/bin/pip install --no-dependencies --no-build-isolation mednet-*.whl
RUN pixi shell-hook --frozen -e cuda-base > /run
RUN echo "mednet" >> /run.sh

FROM nvidia/cuda:12.3.1-runtime-ubuntu22.04 AS production

# only copy the production environment into prod container
COPY --from=build /app/.pixi/envs/cuda /app/.pixi/envs/cuda
COPY --from=build /shell-hook /shell-hook
RUN apt update \
    && apt install -y libgl1-mesa-glx > /dev/null \
    && apt-get clean && apt-get autoclean && apt-get autoremove \
    && rm -rf /var/lib/apt/lists/*
WORKDIR /app
CMD ["/bin/bash", "/run.sh"]