# Usa una imagen base de Python con PyTorch si es posible, para un entorno ML robusto
FROM python:3.11

# Instala librerías de sistema necesarias (por ejemplo, para soundfile)
RUN apt-get update && apt-get install -y \
    libsndfile1 \
    git \
    && rm -rf /var/lib/apt/lists/*

# Configura variables de entorno para una instalación limpia
ENV PIP_NO_CACHE_DIR=off

# Instala todas las dependencias necesarias.
# CRUCIAL: Instalar transformers desde el código fuente (git+https) 
# asegura que tienes la definición más reciente del modelo 'qwen3_omni_moe'.
RUN pip install --no-cache-dir \
    torch \
    soundfile \
    protobuf \
    sentencepiece \
    tiktoken \
    accelerate \
    "transformers[sentencepiece] @ git+https://github.com/huggingface/transformers.git"

# El handler.py debe ser copiado al directorio /repository
# En un despliegue normal de HF, esto ocurre automáticamente si el handler.py está en la raíz del repositorio.