Update app.py

app.py

@@ -116,6 +116,19 @@ warmup_state: Dict[str, Any] = {
 
 warmup_lock = threading.Lock()
 warmup_stop = threading.Event()
+# --- Shim anti-avertissement Pylance (défini si absent) ----------------------
+if "_hf_cache_dir" not in globals():
+    def _hf_cache_dir() -> str:
+        return os.path.expanduser(os.getenv("HF_HOME", "/home/user/.cache/huggingface"))
+
+if "_is_repo_cached" not in globals():
+    def _is_repo_cached(repo_id: str) -> bool:
+        local_dir = os.path.join(_hf_cache_dir(), "models", repo_id.replace("/", "__"))
+        try:
+            return os.path.isdir(local_dir) and any(os.scandir(local_dir))
+        except FileNotFoundError:
+            return False
+# ---------------------------------------------------------------------------
 
 def _default_model_list() -> List[str]:
     """
@@ -518,11 +531,14 @@ from copy import deepcopy
 
 @app.post("/warmup/start", tags=["warmup"])
 async def warmup_start(payload: Optional[Dict[str, Any]] = Body(None)):
-        """
+    """
     Démarre le téléchargement séquentiel des modèles.
     Corps JSON optionnel: { "models": ["repo1","repo2", ...] }
     Si vide: lit la variable d'env WARMUP_MODELS (JSON).
     """
+    # Nettoie un éventuel stop précédent pour permettre un nouveau run
+    warmup_stop.clear()
+
     models = []
     if payload and isinstance(payload, dict):
         models = [str(x).strip() for x in (payload.get("models") or []) if str(x).strip()]
@@ -561,7 +577,6 @@ async def warmup_start(payload: Optional[Dict[str, Any]] = Body(None)):
 
     return {"ok": True, "started": True, "total": len(models), "job_id": job_id}
 
-
 @app.get("/warmup/status", tags=["warmup"])
 def warmup_status():
     """Retourne l'état courant du warm-up (progression, logs, etc.)."""
@@ -585,8 +600,10 @@ def warmup_stop_api():
         running = bool(warmup_state.get("running"))
     if running:
         warmup_stop.set()
+        _log_warmup("[STOP] Demande d'arrêt reçue — arrêt après le dépôt en cours")
         return {"ok": True, "was_running": True}
     return {"ok": True, "was_running": False}
+
 # >>> B1_END warmup_routes
 
 
@@ -1059,8 +1076,12 @@ const DEFAULT_MODELS = [
 
 function openWarmupPopup(){ if(warmupPopup) warmupPopup.style.display = 'block'; }
 
-function closeWarmupPopup(){ if(warmupPopup) warmupPopup.style.display = 'none'; }
+function closeWarmupPopup(){ if(warmupPopup) warmupPopup.style.display = 'none'; } 
+let userClosedWarmupPopup = false;
+
 if (warmupCloseBtn) warmupCloseBtn.addEventListener('click', closeWarmupPopup);
+if (warmupCloseBtn) warmupCloseBtn.addEventListener('click', ()=>{ userClosedWarmupPopup = true; });
+
 
 // Rafraîchissement d’état
 let warmupTimer = null;
@@ -1134,14 +1155,14 @@ async function refreshWarmupUI(){
       warmupLogs.scrollTop = warmupLogs.scrollHeight;
     }
 
-    if (warmupStopBtn) warmupStopBtn.style.display = running ? 'inline-block' : 'none';
-
-    if (running) {
-      openWarmupPopup();
+        if (running) {
+      if (!userClosedWarmupPopup) openWarmupPopup();
       if (!warmupTimer) warmupTimer = setInterval(refreshWarmupUI, 1000);
     } else {
       if (warmupTimer) { clearInterval(warmupTimer); warmupTimer = null; }
+      userClosedWarmupPopup = false;
     }
+
   }catch(e){ /* no-op */ }
 }
 
@@ -1187,13 +1208,11 @@ if (warmupStartBtn){
     } catch(e) {
       warmupPreface = '';
     }
-
-
-
-
     // Ouvre la popup et affiche immédiatement la liste demandée
+    userClosedWarmupPopup = false;
     openWarmupPopup();
-    if (warmupPopupStatus) warmupPopupStatus.textContent = 'Préparation en cours…';
+    if (warmupPopupStatus) warmupPopupStatus.textContent = 'Téléchargement en cours…';
+
         if (warmupLogs) {
       warmupLogs.textContent = (warmupPreface ? warmupPreface + '\n' : '') + '— démarrage…';
       warmupLogs.scrollTop = warmupLogs.scrollHeight; // auto-scroll vers le bas
@@ -1319,6 +1338,7 @@ if (launchSelectedBtn){
       .filter(cb=>cb.checked).map(cb=>cb.value);
     if(!picks.length){ alert('Sélectionne au moins un modèle.'); return; }
     window.lastRequestedModels = picks.slice();
+        userClosedWarmupPopup = false;
     if (typeof openWarmupPopup === 'function') openWarmupPopup();
     if (typeof refreshWarmupUI === 'function') await refreshWarmupUI();
     try{