# Base image
FROM nvcr.io/nvidia/jax:25.01-py3

RUN apt-get update && apt-get install -y \
    libx11-dev \
    libgl1-mesa-dev \
    libglib2.0-0

RUN pip install --upgrade pip

# create a directory for the code
RUN mkdir -p /app/PointObjRep

# root directory
WORKDIR /app/PointObjRep

# copy the requirements file
COPY req.txt req.txt

# install the requirements
RUN pip install -r req.txt

# copy the code
COPY . .

# set the environment variable
RUN echo "OBJECT_PATH=/app/PointObjRep/object_dataset" >> .env
RUN echo "REP_CKPT=checkpoints/rep_ckpt/fps64_ccd_grad_v4" >> .env
