# Use an appropriate base image with Ubuntu
FROM nvidia/cuda:11.2.2-base-ubuntu20.04

# Set environment variables for CUDA
ENV CUDA_HOME=/usr/local/cuda
ENV PATH=$CUDA_HOME/bin:$PATH
ENV LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH

# Set DEBIAN_FRONTEND to noninteractive to prevent interactive prompts
ENV DEBIAN_FRONTEND=noninteractive

# # Install NVIDIA driver
# RUN apt-get update && apt-get install -y --no-install-recommends nvidia-driver-460

# Install curl
RUN apt-get update \
    && apt-get install -y curl

# Install git
RUN apt-get -y update \
    && apt-get install -y git

# Install ffmpeg
RUN apt-get -y update \
    && apt-get install -y ffmpeg

# Install yt-dlp
RUN apt-get -y update \
    && apt-get install -y yt-dlp

# Install Python Dependencies
RUN apt-get -y update \
    && apt-get install -y software-properties-common

# Adding PPA for Python 3.10
RUN apt-get -y update \
    && add-apt-repository ppa:deadsnakes/ppa

# Install Python 3.10
RUN apt-get -y update \
    && apt-get install -y python3.10 python3.10-distutils

# Setting Python 3.10 as the default
RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.8 1
RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.10 2
RUN update-alternatives --config python3

# Download and install pip for Python 3.10
RUN curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py \
    && python3.10 get-pip.py

# Check Python and pip versions
RUN python3.10 --version \
    && pip --version

# Install Cryptodome
RUN apt-get -y update \
    && apt-get install -y python-pycryptodome

# # FROM python:3.10

# Switch back to the root user to install Python packages
USER root

# Copy your requirements.txt and install Python packages
COPY ./requirements.txt /code/requirements.txt
RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt

# Create a non-root user and set environment variables
RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user \
    PATH=/home/user/.local/bin:$PATH

# Set the working directory and copy your application files
WORKDIR $HOME/app
COPY --chown=user . $HOME/app

# Specify the command to run your application
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]