FROM ubuntu:24.04

# ENV PIP_ROOT_USER_ACTION=ignore
ARG USERNAME=mcp_user
ARG USER_UID=1000
ARG APP_WORKDIR=app_srv

# Update OS and install packages
WORKDIR /tmp/
COPY requirements.txt packages.txt ./
RUN apt-get -y update && apt-get -y upgrade 
# RUN xargs apt -y install < packages.txt
RUN apt-get -y install ffmpeg git git-lfs htop iotop libxml2 libopenblas-dev libssl-dev python3-pip python3-wheel python3-setuptools python-is-python3 wget zlib1g net-tools curl

# Install CUDA 12.8 from Internet
RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb \
    && dpkg -i cuda-keyring_1.1-1_all.deb \
    && apt-get update \
    && apt-get -y install cuda-toolkit-12-8
# https://docs.nvidia.com/cuda/cuda-installation-guide-linux/#post-installation-actions
ENV PATH=/usr/local/cuda-12.8/bin:${PATH}									
ENV LD_LIBRARY_PATH=/usr/local/cuda-12.8/lib64:/usr/lib/x86_64-linux-gnu/

# Install cuDNN for CUDA 12
RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb \
	&& dpkg -i cuda-keyring_1.1-1_all.deb \
	&& apt-get update \
	&& apt-get -y install cudnn-cuda-12

# Remove EXTERNALLY-MANAGED
RUN rm -f /usr/lib/python3.12/EXTERNALLY-MANAGED

# Install Pytorch for CUDA 12.8
RUN pip3 install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128

# Install other python packages
RUN pip3 install --no-cache-dir -r requirements.txt

# Final image cleanup
RUN apt-get clean
RUN rm -rf /tmp/*

# Preparing an image for HuggingFace Space
## Delete exist user with UID 1000
RUN userdel -r ubuntu || true

## Set up a new user with user ID 1000
## https://huggingface.co/docs/hub/spaces-sdks-docker#permissions
RUN useradd -s /bin/bash -m -u $USER_UID $USERNAME

## Switch to the "user" user
USER $USERNAME

## Set home to the user's home directory
ENV HOME=/home/$USERNAME \
	PATH=/home/$USERNAME/.local/bin:$PATH

## Copy the contents of the application directory to the /home/mcp_user/app_srv container directory, setting mcp_user as the user owner
RUN mkdir -p $HOME/$APP_WORKDIR
COPY --chown=$USERNAME ./$APP_WORKDIR/* $HOME/$APP_WORKDIR

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

# Gradio network configuration 
EXPOSE 7860
ENV GRADIO_SERVER_NAME="0.0.0.0"

# Temporarily set an environment variable for unbuffered output. The goal is to test the application.
ENV PYTHONUNBUFFERED=1

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