FROM python:3.10-slim

LABEL version="1.0.2"

ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
ENV TZ=America/Los_Angeles
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone

RUN apt-get update && apt-get install -y --no-install-recommends \
    curl \
    libx11-6 \
    libxrender1 \
    libxxf86vm1 \
    libxfixes3 \
    libxi6 \
    libxkbcommon0 \
    libgomp1 \
    libsm6 \
    libgl1 \
    git

RUN pip3 install --no-cache-dir \
    numpy \
    trimesh \
    Pillow \
    gradio==4.40.0 \
    bpy \
    opencv-python-headless

RUN pip3 install \
    torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

RUN pip3 install git+https://github.com/facebookresearch/segment-anything.git

# Set up a new user named "user" with user ID 1000
RUN useradd -m -u 1000 user
# Switch to the "user" user
USER user
# Set home to the user's home directory
ENV HOME=/home/user \
	PATH=/home/user/.local/bin:$PATH \
    PYTHONPATH=$HOME/app \
	PYTHONUNBUFFERED=1 \
	SYSTEM=spaces
# Set the working directory to the user's home directory
WORKDIR $HOME/app
# Copy the current directory contents into the container at $HOME/app setting the owner to the user
COPY --chown=user . $HOME/app

RUN pip3 install gradio_model3dcolor-0.0.1-py3-none-any.whl
RUN pip3 install gradio_model3dnormal-0.0.1-py3-none-any.whl

RUN mkdir ./checkpoints/ && curl https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth -o ./checkpoints/sam_vit_h_4b8939.pth

CMD ["python3", "gradio_demo.py"]