# Use the NVIDIA CUDA runtime as a base image
FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04

# Set environment variables
ENV DEBIAN_FRONTEND=noninteractive

# Install system dependencies
RUN apt-get update && apt-get install -y --no-install-recommends \
    git \
    cmake \
    build-essential \
    libgl1-mesa-glx \
    libglib2.0-0 \
    ffmpeg \
    python3.8 \
    python3-pip \
    python3.8-dev \
    && rm -rf /var/lib/apt/lists/*

# Set Python 3.8 as the default python and pip versions
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.8 1
RUN update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1

# Create a working directory
WORKDIR /app

# Clone the repository (adjust the URL if needed)
RUN git clone --recursive https://github.com/jnjaby/KEEP.git .

# Install Python dependencies from requirements.txt
RUN pip install --upgrade pip
RUN pip install -r requirements.txt

# Install basicsr (assuming setup.py is in basicsr directory)
RUN python basicsr/setup.py develop

# Install additional Python packages
RUN pip install dlib ffmpeg-python

# Initialize and update Git submodules (if not handled by the clone command)
RUN git submodule init && git submodule update

# Expose any required ports (if your app uses any)
# EXPOSE 8000

# Command to run your application
CMD ["python", "app.py"]