FROM pytorch/pytorch:2.5.1-cuda12.1-cudnn9-runtime

WORKDIR /app

# Upgrade pip first to avoid dependency resolution issues
RUN pip install --no-cache-dir --upgrade pip setuptools wheel

# Install system dependencies needed for soundfile/audio
RUN apt-get update && apt-get install -y --no-install-recommends \
    libsndfile1 \
    build-essential \
    && rm -rf /var/lib/apt/lists/*

# Pin transformers to compatible version (BeamSearchScorer removed in 4.44+)
RUN pip install --no-cache-dir "transformers<4.44"

# Install Python packages
RUN pip install --no-cache-dir TTS==0.22.0 gradio==5.25.0

# Copy pre-download script first (to cache in layer even if download fails)
COPY predownload.py /app/predownload.py

# Pre-download XTTS v2.0 model files (cached in Docker layer)
RUN python /app/predownload.py

COPY . /app

# Rebuild 14:05 - generate noticiero
CMD ["python", "app.py"]