FROM python:3.11

# Definir HOME y rutas de caché para Hugging Face
WORKDIR /home/user/app
ENV HOME="/home/user"
ENV HF_HOME="/home/user/.cache/huggingface"
ENV OMP_NUM_THREADS=1

# Configuración de Git vía variables de entorno (sin .gitconfig)
ENV GIT_AUTHOR_NAME="sob111"
ENV GIT_AUTHOR_EMAIL="sob111@users.noreply.huggingface.co"
ENV GIT_COMMITTER_NAME="sob111"
ENV GIT_COMMITTER_EMAIL="sob111@users.noreply.huggingface.co"

# 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/*

# Instalar dependencias Python
COPY requirements.txt .
RUN pip install --upgrade pip \
    && pip install --no-cache-dir -r requirements.txt

# Instalar Coqui TTS en editable mode desde v0.22.0
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 --no-cache-dir -e .

# Crear directorio de salida para checkpoints
RUN mkdir -p /tmp/output_model

# Copiar el proyecto
COPY . .

# Verificar instalación
RUN pip show TTS

# Entrenamiento: usa "python -m TTS.bin.train" en lugar de tts_train
# RUN python -m TTS.bin.train --help

# Comando por defecto (se puede sobreescribir en Space)
CMD ["python", "app.py"]
# CMD ["python", "finetune_xtts_hf.py"]