FROM nvidia/cuda:12.3.2-devel-ubuntu22.04 AS base

# Set environment variables
ENV DEBIAN_FRONTEND=noninteractive
WORKDIR /app

# Install dependencies
RUN apt-get update && apt-get install -y \
    git \
    curl \
    wget \
    python3 \
    python3-pip \
    && rm -rf /var/lib/apt/lists/*

COPY . /app/

RUN if ! update-alternatives --query python > /dev/null 2>&1; then \
        update-alternatives --install /usr/bin/python python /usr/bin/python3 1; \
    fi

RUN pip install -r /app/requirements.txt
USER root
RUN chmod +x /app/start.sh
EXPOSE 7860
CMD ["python", "/app/setup.py"]