FROM nvidia/cuda:11.7.1-devel-ubuntu22.04

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

# Create a group and user
ENV APPUSER="appuser"
ENV HOME=/home/$APPUSER
RUN useradd -m -u 1000 $APPUSER

USER $APPUSER
WORKDIR $HOME

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

COPY --chown=$APPUSER ./requirements.txt $HOME/app/requirements.txt
WORKDIR $HOME/app

RUN pip install --no-cache-dir --user -r requirements.txt

COPY --chown=$APPUSER . $HOME/app

# Expose port for web service
ENV PORT=7860
EXPOSE $PORT

# Run streamlit app under conda environment
# CMD ["sh", "-c", "streamlit run --server.port=$PORT --server.address=0.0.0.0 app.py"]

CMD ["python3", "main.py"]