# Use an NVIDIA PyTorch container with cuDNN 9.1 support
FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04

# Basic setup
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update && apt-get install -y \
    python3 python3-pip ffmpeg git libsndfile1 \
    && rm -rf /var/lib/apt/lists/*

# Set up Python environment
COPY requirements.txt .
RUN pip install --upgrade pip && pip install -r requirements.txt

# Copy app files
COPY . /app
WORKDIR /app

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