# Base image with CUDA and cuDNN
FROM nvidia/cuda:11.2.2-cudnn8-runtime-ubuntu20.04

# Set environment variables for CUDA and cuDNN
ENV CUDA_VERSION 11.2.2
ENV CUDNN_VERSION 8

# Install dependencies
RUN apt-get update && apt-get install -y --no-install-recommends \
    build-essential \
    curl \
    ca-certificates \
    python3-pip \
    python3-dev \
    && rm -rf /var/lib/apt/lists/*

# Install TensorFlow
RUN pip3 install --upgrade pip
RUN pip3 install tensorflow==2.8.0

# Install Streamlit and other Python dependencies
COPY requirements.txt .
RUN pip3 install -r requirements.txt

# Copy the application code
COPY . /app
WORKDIR /app

# Expose the Streamlit port
EXPOSE 8501

# Run the Streamlit application
CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]