Dockerfile

FROM python:3.11-slim

# Configurar paths
WORKDIR /home/user/app
ENV HF_HOME="/home/user/.cache/huggingface"
ENV PYTHONUNBUFFERED=1
ENV NUMBA_CACHE_DIR=/tmp/numba_cache
ENV NUMBA_DISABLE_CACHING=1
ENV OMP_NUM_THREADS=1

# Instalar dependencias del sistema
RUN apt-get update && apt-get install -y \
    git wget ffmpeg libsndfile1 unzip build-essential \
    && rm -rf /var/lib/apt/lists/*

# Clonar e instalar Coqui TTS v0.22 en modo editable (con train y recipes)
RUN git clone --branch v0.22.0 https://github.com/coqui-ai/TTS.git /home/user/TTS \
    && cd /home/user/TTS \
    && git submodule update --init --recursive \
    && pip install --upgrade pip \
    && pip install --no-cache-dir -e .

# Copiar requirements adicionales
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# Crear carpeta de salida
RUN mkdir -p /tmp/output_model

# Copiar el proyecto
COPY . .

# Script de entrenamiento
CMD ["python", "finetune_xtts_hf.py"]