# Use an NVIDIA CUDA base image with Python
FROM nvidia/cuda:12.2.0-devel-ubuntu22.04

# Set working directory
WORKDIR /app

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

# Upgrade pip
RUN pip3 install --upgrade pip

# Copy requirements.txt and install Python dependencies
COPY requirements.txt .
RUN pip3 install -r requirements.txt

# Copy the application code
COPY app.py .

# Expose the port used by Gradio (default is 7860)
EXPOSE 7860

# Command to run the application
CMD ["python3", "app.py"]