# FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04

# ARG DEBIAN_FRONTEND=noninteractive


# RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
#     curl git wget libgl1 freeglut3-dev unzip ffmpeg libsm6 libxext6 libgomp1 \
#     libfontconfig1 libegl1-mesa-dev libgl1-mesa-glx libglapi-mesa libc6 \
#     libxdamage1 libxfixes3 libxcb-glx0 libxcb-dri2-0 libosmesa6-dev \
#     libxcb-dri3-0 libxcb-present0 libxcb-sync1 libxshmfence1 libxxf86vm1 \
#     libgl1-mesa-dev libgles2-mesa-dev libglib2.0-dev libxrender1 libgbm1 \
#     build-essential libeigen3-dev python3.10 python3-pip python-is-python3 \
#     python3-opencv nvidia-cuda-toolkit \
#     && rm -rf /var/lib/apt/lists/*

# # Set up a new user named "user" with user ID 1000
# RUN useradd -m -u 1000 user
# # libgl1
# # unzip
# # ffmpeg 
# # libsm6 
# # libxext6
# # libgl1-mesa-dri
# # libegl1-mesa
# # libgbm1
# # libosmesa6
# # python3-opengl

# # Switch to the "user" user
# USER user

# FROM python:3.10

# ENV PYTHONUNBUFFERED=1

# ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6"
# ENV TCNN_CUDA_ARCHITECTURES=86;80;75;70;61;60
# ENV FORCE_CUDA=1

# ENV CUDA_HOME=/usr/local/cuda
# ENV PATH=${CUDA_HOME}/bin:/home/${USER_NAME}/.local/bin:/usr/bin:${PATH}
# ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:/usr/local/lib/python3.8/site-packages/open3d:/usr/lib:/usr/lib64:${LD_LIBRARY_PATH}
# ENV LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:/usr/local/lib/python3.8/site-packages/open3d:/usr/lib:/usr/lib64:${LIBRARY_PATH}

# # Set home to the user's home directory
# ENV HOME=/home/user \
#     PATH=/home/user/.local/bin:$PATH \
#     PYTHONPATH=$HOME/app:$PYTHONPATH \
# 	PYTHONUNBUFFERED=1 \
# 	GRADIO_ALLOW_FLAGGING=never \
# 	GRADIO_NUM_PORTS=1 \
# 	GRADIO_SERVER_NAME=0.0.0.0 \
# 	GRADIO_THEME=huggingface \
# 	SYSTEM=spaces

# RUN pip install --upgrade pip setuptools ninja
# RUN pip install torch==2.3.1+cu121 -f https://download.pytorch.org/whl/cu113/torch_stable.html

# RUN python -c "import torch; print(torch.version.cuda)"
# COPY requirements.txt /tmp
# RUN cd /tmp && pip install -r requirements.txt

# # RUN pip install https://download.is.tue.mpg.de/icon/HF/kaolin-0.11.0-cp38-cp38-linux_x86_64.whl
# # RUN pip install https://download.is.tue.mpg.de/icon/HF/pytorch3d-0.7.0-cp38-cp38-linux_x86_64.whl

# RUN chmod 777 $HOME
# ENV TRANSFORMERS_CACHE=/tmp
# ENV MPLCONFIGDIR=/tmp

# # Copy the current directory contents into the container at $HOME/app setting the owner to the user
# COPY --chown=user . $HOME/app

# # Set the working directory to the user's home directory
# WORKDIR $HOME/app

# CMD ["python", "app.py"]

#----------------------------------------
FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04

ARG DEBIAN_FRONTEND=noninteractive

RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
    curl git wget libgl1 freeglut3-dev unzip ffmpeg libsm6 libxext6 libgomp1 \
    libfontconfig1 libegl1-mesa-dev libgl1-mesa-glx libglapi-mesa libc6 \
    libxdamage1 libxfixes3 libxcb-glx0 libxcb-dri2-0 libosmesa6-dev \
    libxcb-dri3-0 libxcb-present0 libxcb-sync1 libxshmfence1 libxxf86vm1 \
    libgl1-mesa-dev libgles2-mesa-dev libglib2.0-dev libxrender1 libgbm1 \
    build-essential libeigen3-dev python3.10 python3-pip python-is-python3 \
    python3-opencv \
    libegl1 \
    libegl1-mesa \
    libglvnd0 \
    libglx0 \
    libglvnd-dev \
    libglx-dev \
    libegl-dev \
    && rm -rf /var/lib/apt/lists/*

# Set up a new user named "user" with user ID 1000
RUN useradd -m -u 1000 user

# Set NVIDIA and EGL specific environment variables
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility,graphics
ENV __GLX_VENDOR_LIBRARY_NAME=nvidia
ENV NVIDIA_VISIBLE_DEVICES=all

USER user

ENV PYTHONUNBUFFERED=1
ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6"
ENV TCNN_CUDA_ARCHITECTURES=86;80;75;70;61;60
ENV FORCE_CUDA=1

ENV CUDA_HOME=/usr/local/cuda
ENV PATH=${CUDA_HOME}/bin:/home/${USER_NAME}/.local/bin:/usr/bin:${PATH}
ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:/usr/local/lib/python3.8/site-packages/open3d:/usr/lib:/usr/lib64:/usr/lib/x86_64-linux-gnu:/usr/lib/x86_64-linux-gnu/nvidia/current:${LD_LIBRARY_PATH}
ENV LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:/usr/local/lib/python3.8/site-packages/open3d:/usr/lib:/usr/lib64:${LIBRARY_PATH}

# Set home to the user's home directory
ENV HOME=/home/user \
    PATH=/home/user/.local/bin:$PATH \
    PYTHONPATH=$HOME/app:$PYTHONPATH \
    PYTHONUNBUFFERED=1 \
    GRADIO_ALLOW_FLAGGING=never \
    GRADIO_NUM_PORTS=1 \
    GRADIO_SERVER_NAME=0.0.0.0 \
    GRADIO_THEME=huggingface \
    SYSTEM=spaces

RUN pip install --upgrade pip setuptools ninja
RUN pip install torch==2.3.1+cu121 -f https://download.pytorch.org/whl/cu113/torch_stable.html

RUN python -c "import torch; print(torch.version.cuda)"
COPY requirements.txt /tmp
RUN cd /tmp && pip install -r requirements.txt

RUN chmod 777 $HOME
ENV TRANSFORMERS_CACHE=/tmp
ENV MPLCONFIGDIR=/tmp

# Copy the current directory contents into the container at $HOME/app setting the owner to the user
COPY --chown=user . $HOME/app

# Set the working directory to the user's home directory
WORKDIR $HOME/app

CMD ["python", "app.py"]