# For more information, please refer to https://aka.ms/vscode-docker-python
#FROM python:3.10-slim
FROM nvidia/cuda:12.6.3-cudnn-devel-ubuntu24.04

ENV DEBIAN_FRONTEND=noninteractive

# Where we'll copy the code
WORKDIR /code

# Copy the current directory contents into the container at /code
COPY ./requirements.txt /code/requirements.txt
# Install pip requirements
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
RUN pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

# Creates a non-root user with an explicit UID
# For more info, please refer to https://aka.ms/vscode-docker-python-configure-containers
RUN useradd -m -u 1000 user
USER user

# Set home to the user's home directory
ENV HOME=/home/user \ 
    PATH=/home/user/.local/bin:$PATH
 
# 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

# During debugging, this entry point will be overridden. For more information, please refer to https://aka.ms/vscode-docker-python-debug
#CMD ["app:app", "--host", "0.0.0.0", "--port", "7860"]

# Expose port 7860
EXPOSE 7860
ENV GRADIO_SERVER_NAME="0.0.0.0"
CMD ["python", "app.py"]

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

# ARG DEBIAN_FRONTEND=noninteractive

# ENV PYTHONUNBUFFERED=1

# RUN apt-get update && apt-get install --no-install-recommends -y \
#   build-essential \
#   python3.9 \
#   python3-pip \
#   git \
#   ffmpeg \
#   && apt-get clean && rm -rf /var/lib/apt/lists/*

# WORKDIR /code

# COPY ./requirements.txt /code/requirements.txt

# # 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 \
# 	GRADIO_ALLOW_FLAGGING=never \
# 	GRADIO_NUM_PORTS=1 \
# 	GRADIO_SERVER_NAME=0.0.0.0 \
# 	GRADIO_THEME=huggingface \
# 	SYSTEM=spaces

# RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt

# # 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

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