FROM python:3.10-slim

WORKDIR /app

# On ajoute 'curl' qui est parfois nécessaire pour les téléchargements de modèles
RUN apt-get update && apt-get install -y \
    build-essential \
    git \
    curl \
    && rm -rf /var/lib/apt/lists/*

# On upgrade pip AVANT d'installer le reste
RUN pip install --no-cache-dir --upgrade pip setuptools wheel

COPY . /app

# On installe le requirements
RUN pip install --no-cache-dir -r requirements.txt

# Pré-téléchargement du modèle Hugging Face pour accélérer le démarrage et permettre le fonctionnement hors-ligne
RUN python -c "from transformers import AutoTokenizer, AutoModel; \
    AutoTokenizer.from_pretrained('distilbert-base-multilingual-cased'); \
    AutoModel.from_pretrained('distilbert-base-multilingual-cased')"

EXPOSE 7860

CMD ["streamlit", "run", "app.py", "--server.port=7860", "--server.address=0.0.0.0"]