Dockerfile

# Dockerfile_api pour Hugging Face Spaces
FROM python:3.11-slim AS builder

# Créer un utilisateur non-root (requis par HF Spaces)
RUN useradd -m -u 1000 user

# Définir le répertoire de travail
WORKDIR /app

# Copier les fichiers de dépendances
COPY --chown=user requirements_api.txt .

# Installer les dépendances
RUN pip install --no-cache-dir --upgrade pip && \
    pip install --no-cache-dir -r requirements_api.txt

# Copier le code de l'application
COPY --chown=user api.py .
COPY --chown=user functions.py .
COPY --chown=user 2_Data_transformed/train_data_sp2_subsample_1.csv ./2_Data_transformed/
COPY --chown=user 2_Data_transformed/shap_feature_importance.csv ./2_Data_transformed/
COPY --chown=user 3_Results/best_gradient_boosting_model.pkl ./3_Results/

# ============================================
# Stage de test : exécution des tests fonctionnels
# ============================================
FROM builder AS test

# Installer les dépendances de test
RUN pip install --no-cache-dir httpx

# Copier le fichier de tests fonctionnels
COPY --chown=user 5_Tests/test_functional.py .

# Exécuter le test fonctionnel sur /health (démarre le serveur et fait une vraie requête HTTP)
RUN python test_functional.py

# ============================================
# Stage de production
# ============================================
FROM builder AS production

# Changer vers l'utilisateur non-root
USER user

# Exposer le port 7860 (port par défaut de HF Spaces)
EXPOSE 7860

# Variables d'environnement
ENV PYTHONUNBUFFERED=1

# Commande de démarrage
CMD ["uvicorn", "api:app", "--host", "0.0.0.0", "--port", "7860"]