Déploiement Dashboard depuis GitLab CI - 2026-05-06 11:39:02

streamlit_app.py

@@ -1,5 +1,7 @@
 import os
+import time
 import tempfile
+import threading
 import requests
 import pandas as pd
 import json
@@ -22,7 +24,42 @@ from huggingface_hub import hf_hub_download
 from functions import most_important_features_min_max
 features_min_max = most_important_features_min_max()
 
+# -----------------------------------------------------------------------
+# Configuration globale – URLs des services HF Spaces
+# -----------------------------------------------------------------------
+API_URL = os.environ.get("API_URL", "https://cedm-oc-mlops-projet-2.hf.space")
+KEEP_ALIVE_INTERVAL = 12 * 3600  # 12 heures en secondes
+
+
+# -----------------------------------------------------------------------
+# Keep-alive : maintien des containers HF Spaces en état "running"
+# Utilise @st.cache_resource pour ne démarrer le thread qu'une seule fois.
+# -----------------------------------------------------------------------
 @st.cache_resource(show_spinner=False)
+def start_keep_alive():
+    """
+    Démarre un thread démon qui envoie des requêtes croisées
+    entre le dashboard et l'API toutes les 12 h, afin de maintenir
+    les deux containers HF Spaces actifs.
+    """
+    def _loop():
+        while True:
+            time.sleep(KEEP_ALIVE_INTERVAL)
+            # Ping de l'API depuis le dashboard (ping croisé suffisant)
+            try:
+                r = requests.get(f"{API_URL}/health", timeout=30)
+                print(f"[keep-alive] Ping API → HTTP {r.status_code}")
+            except Exception as e:
+                print(f"[keep-alive] Ping API échoué: {e}")
+
+    thread = threading.Thread(target=_loop, daemon=True, name="keep-alive-api")
+    thread.start()
+    return thread
+
+
+start_keep_alive()
+
+# -----------------------------------------------------------------------
 def load_shap_explainer():
     """Charge l'explainer SHAP depuis le fichier pickle."""
     with open("./hgb_shap_explainer.pkl", "rb") as f:
@@ -117,8 +154,6 @@ with tab1:
                                        help="Moyenne (ratio) du nombre de demandes précédentes faites suite à une visite en agence (walk-in)",
                                        key='prev_name_product_type_walk_in_mean')
 
-    # URL de l'API
-    API_URL = "https://cedm-oc-mlops-projet-2.hf.space"
 
     if st.button("🔮 Lancer la prédiction", type="primary"):
         # Bouton pour lancer la prédiction