Déploiement automatique depuis GitLab CI

main.py

@@ -5,10 +5,12 @@ Cette API charge un modèle MLflow (pipeline sklearn) au démarrage et expose de
 pour effectuer des prédictions de rendement (hg/ha) à partir de variables explicatives.
 """
 
+import asyncio
 import os
 from contextlib import asynccontextmanager
 from typing import Dict, List
 
+import httpx
 import joblib
 import numpy as np
 import logfire
@@ -38,6 +40,9 @@ logfire.configure(
 
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 
+# URL du frontend à reveiller toutes les 12h (définie via variable d'environnement)
+FRONTEND_URL = os.environ.get("FRONTEND_URL", "https://huggingface.co/spaces/CedM/oc_mlops_projet_4_dashboard")
+
 # Chemin vers le modèle MLflow (pipeline sklearn complet avec préprocesseur)
 MODEL_PATH = os.path.join(BASE_DIR, MODEL_REL_PATH)
 
@@ -103,10 +108,38 @@ async def lifespan(app: FastAPI):
     pipeline = load_pipeline()
     available_items, available_areas, available_items_per_area = load_training_data()
     logfire.info("API initialisée et prête")
+
+    # Lancement de la tâche keep-alive en arrière-plan
+    task = asyncio.create_task(keep_alive_task())
+
     yield
+
+    task.cancel()
     logfire.info("API arrêtée")
 
 
+async def keep_alive_task():
+    """
+    Tâche de keep-alive : ping le frontend toutes les 12h
+    afin de le maintenir actif sur les plateformes d'hébergement avec mise en veille.
+    """
+    INTERVAL = 12 * 3600  # 12 heures en secondes
+    await asyncio.sleep(60)  # Délai initial pour laisser le temps au démarrage complet
+
+    while True:
+        if FRONTEND_URL:
+            async with httpx.AsyncClient() as client:
+                try:
+                    resp = await client.get(FRONTEND_URL, timeout=15)
+                    logfire.info("Keep-alive frontend: réponse {status}", status=resp.status_code)
+                except Exception as e:
+                    logfire.warning("Keep-alive frontend échoué: {error}", error=str(e))
+        else:
+            logfire.warning("Keep-alive: FRONTEND_URL non défini, aucun ping envoyé.")
+
+        await asyncio.sleep(INTERVAL)
+
+
 # =======================================================================================================
 # Initialisation de l'application FastAPI + instrumentation Logfire
 # =======================================================================================================