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 \
    && pixi shell-hook --frozen -e cuda-base > /entrypoint.sh \
    && echo "mednet" >> /entrypoint.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-base /app/.pixi/envs/cuda-base
COPY --from=build /entrypoint.sh /entrypoint.sh
RUN apt-get update \
    && apt-get install --no-install-recommends -y libopengl0 libegl1 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", "/entrypoint.sh"]