Update app.py

app.py

@@ -106,7 +106,14 @@ warmup_state: Dict[str, Any] = {
     "total": 0,
     "idx": 0,
     "job_id": "default",
+    # NOUVEAUX CHAMPS POUR RÉCAP
+    "asked": [],           # liste demandée au lancement
+    "ok_repos": [],        # liste des repos téléchargés OK
+    "failed_repos": [],    # liste des repos en échec
+    "started_at": 0.0,     # timestamp début
+    "finished_at": 0.0,    # timestamp fin
 }
+
 warmup_lock = threading.Lock()
 warmup_stop = threading.Event()
 
@@ -187,25 +194,13 @@ def _download_one(repo_id: str, tries: int = 3) -> bool:
     return False
 
 def _warmup_thread(models: List[str]):
-    with warmup_lock:
-        warmup_state.update({
-            "running": True,
-            "percent": 0,
-            "ok_count": 0,
-            "done": False,
-            "logs": [],
-            "current": None,
-            "total": len(models),
-            "idx": 0,
-        })
-    warmup_stop.clear()
     ok_count = 0
     t_global = time.time()
     _log_warmup(f"[JOB] start • {len(models)} dépôts")
 
     for i, repo in enumerate(models):
         if warmup_stop.is_set():
-            _log_warmup("[STOP] Arrêt demandé par l’utilisateur — fin du job après ce point")
+            _log_warmup("[STOP] Arrêt demandé — fin du job après ce point")
             break
 
         with warmup_lock:
@@ -214,12 +209,14 @@ def _warmup_thread(models: List[str]):
             warmup_state["percent"] = int((i / max(1, len(models))) * 100)
 
         ok = _download_one(repo)
-        if ok:
-            ok_count += 1
-            with warmup_lock:
-                warmup_state["ok_count"] = ok_count
-        # progression après dépôt (OK ou FAIL)
         with warmup_lock:
+            if ok:
+                ok_count += 1
+                warmup_state["ok_count"] = ok_count
+                warmup_state["ok_repos"].append(repo)
+            else:
+                warmup_state["failed_repos"].append(repo)
+
             warmup_state["percent"] = int(((i + 1) / max(1, len(models))) * 100)
 
         if warmup_stop.is_set():
@@ -231,8 +228,10 @@ def _warmup_thread(models: List[str]):
         warmup_state["done"] = True
         warmup_state["running"] = False
         warmup_state["ok_count"] = ok_count
+        warmup_state["finished_at"] = time.time()
     _log_warmup(f"[JOB] done • {ok_count}/{len(models)} • {time.time()-t_global:.1f}s")
 
+
 # <<<< HUGINFACE PATCH: WARMUP STATE+HELPERS END >>>>
 # ---------- Helpers ----------
 
@@ -519,7 +518,7 @@ 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).
@@ -538,20 +537,31 @@ async def warmup_start(payload: Optional[Dict[str, Any]] = Body(None)):
 
     with warmup_lock:
         if warmup_state.get("running"):
-            # Déjà en cours : on renvoie l'état actuel pour info
-            return {
-                "ok": False,
-                "already_running": True,
-                "status": deepcopy(warmup_state),
-            }
-        # Lancer un nouveau thread
+            return {"ok": False, "already_running": True, "status": deepcopy(warmup_state)}
+        # RESET d’un nouveau job
         job_id = uuid.uuid4().hex[:8]
-        warmup_state["job_id"] = job_id
+        warmup_state.update({
+            "job_id": job_id,
+            "asked": models[:],
+            "ok_repos": [],
+            "failed_repos": [],
+            "ok_count": 0,
+            "logs": [],
+            "idx": 0,
+            "total": len(models),
+            "percent": 0,
+            "running": True,
+            "done": False,
+            "current": None,
+            "started_at": time.time(),
+            "finished_at": 0.0,
+        })
         th = threading.Thread(target=_warmup_thread, args=(models,), daemon=True)
         th.start()
 
     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.)."""
@@ -835,12 +845,12 @@ HTML_TEMPLATE = r"""
 
 <!-- >>> A2B2_BEGIN warmup_topbar -->
 <div style="display:flex;gap:8px;align-items:center">
-  <button id="warmupStartBtn" class="btn" title="Télécharger les modèles (séquentiel)">🔥 Préparer modèles</button>
-  <button id="warmupStopBtn" class="btn" title="Arrêter le warm-up en cours" style="display:none">⏹️ Stop warm-up</button>
-  <button id="warmupSelectBtn" class="btn" title="Sélectionner les repos à préparer">🧩 Choisir modèles</button>
-
-  <button id="warmupLogsBtn" class="btn" title="Voir les logs du warm-up">📜 Logs warm-up</button>
+    <button id="warmupStartBtn" class="btn" title="Télécharger les modèles (séquentiel)">🔥 Télécharger modèles</button>
+  <button id="warmupStopBtn" class="btn" title="Arrêter le téléchargement en cours" style="display:none">⏹️ Stop téléchargement</button>
+  <button id="warmupSelectBtn" class="btn" title="Sélectionner les repos à télécharger">🧩 Choisir modèles</button>
+  <button id="warmupLogsBtn" class="btn" title="Voir les logs du téléchargement">📜 Logs téléchargement</button>
   <span id="warmupStatus" class="muted"></span>
+
 </div>
 <!-- >>> A2B2_END warmup_topbar -->
 
@@ -920,7 +930,8 @@ HTML_TEMPLATE = r"""
 
 <!-- >>> A2B2_BEGIN warmup_popup -->
 <div id="warmupPopup" style="display:none; position:fixed; top:22%; left:50%; transform:translate(-50%,-50%); background:#fff; padding:20px; border-radius:8px; box-shadow:0 0 10px rgba(0,0,0,0.2); z-index:1001; min-width:360px">
-  <h3>Préparation des modèles (warm-up)</h3>
+    <h3>Téléchargement des modèles (warm-up)</h3>
+
   <div class="muted" id="warmupPopupStatus">—</div>
   <div style="background:#f3f4f6;border-radius:4px;height:8px;margin:10px 0">
     <div id="warmup-progress-fill" style="background:#2563eb;height:100%;width:0;border-radius:4px"></div>
@@ -937,7 +948,8 @@ HTML_TEMPLATE = r"""
     <button id="selectAllBtn" class="btn">Tout cocher</button>
     <button id="selectNoneBtn" class="btn">Tout décocher</button>
     <button id="selectCancelBtn" class="btn">Annuler</button>
-    <button id="launchSelectedBtn" class="btn">Lancer la préparation</button>
+        <button id="launchSelectedBtn" class="btn">Lancer le téléchargement</button>
+
   </div>
 </div>
 <!-- >>> A2B2_END warmup_popup -->
@@ -1001,6 +1013,32 @@ window._lastInstanceId = null;
 const warmupCloseBtn     = document.getElementById('warmupCloseBtn');
 
 // Helpers popup
+function buildFinalSummary(s){
+  const asked  = Array.isArray(s.asked) ? s.asked : [];
+  const okL    = Array.isArray(s.ok_repos) ? s.ok_repos : [];
+  const failL  = Array.isArray(s.failed_repos) ? s.failed_repos : [];
+  const cached = Array.isArray(s.audit_cached) ? s.audit_cached : [];
+  const tot    = s.total ?? asked.length ?? 0;
+
+  const lines = [];
+  lines.push("RÉCAP FINAL");
+  lines.push(`Demandés : ${asked.length}`);
+  if (asked.length) lines.push(asked.map(m=>' • '+m).join('\n'));
+
+  lines.push("");
+  lines.push(`✅ Téléchargés : ${okL.length}/${tot}`);
+  if (okL.length) lines.push(okL.map(m=>' • '+m).join('\n'));
+
+  lines.push("");
+  lines.push(`❌ Échecs : ${failL.length}`);
+  if (failL.length) lines.push(failL.map(m=>' • '+m).join('\n'));
+
+  lines.push("");
+  lines.push(`📦 En cache maintenant : ${cached.length}`);
+  if (cached.length) lines.push(cached.map(m=>' • '+m).join('\n'));
+
+  return lines.join('\n');
+}
 
 
 // >>> C2_BEGIN warmup_preface
@@ -1048,32 +1086,25 @@ async function refreshWarmupUI(){
     if (warmupStatusEl) {
   const tot = (s.total ?? 0);
        if (running) {
-        const idx = (s.idx ?? 0) + 1;
+                const idx = (s.idx ?? 0) + 1;
         warmupStatusEl.textContent = `⏳ ${pct}% — ${s.current||''} (${idx}/${tot}) [inst:${instanceId}]`;
       } else {
-
-    // Détection fin de run (par job_id) => toast récap une seule fois
-    if (s.done && s.job_id && window._lastNotifiedJobId !== s.job_id) {
-      const tot2 = (s.total ?? 0);
-      const ok2  = Number.isFinite(s.ok_count) ? s.ok_count : '—';
-      showToast(`Warm-up terminé — ${ok2}/${tot2} téléchargés (inst:${instanceId})`);
-      window._lastNotifiedJobId = s.job_id;
-    }
-
-    if (s.done && Number.isFinite(s.ok_count)) {
-      warmupStatusEl.textContent = `✅ Terminé — ${s.ok_count}/${tot} téléchargés (inst:${instanceId})`;
-    } else {
-      const nCache = Number.isFinite(s.audit_count)
-        ? s.audit_count
-        : (Array.isArray(s.audit_cached) ? s.audit_cached.length : 0);
-      if (nCache > 0) {
-        warmupStatusEl.textContent = `✅ Prêt — cache local: ${nCache} (inst:${instanceId})`;
-      } else {
-        warmupStatusEl.textContent = `Prêt (aucun run)`;
+        if (s.done) {
+          const ok  = Number.isFinite(s.ok_count) ? s.ok_count : 0;
+          const fail = Array.isArray(s.failed_repos) ? s.failed_repos.length : Math.max(0, tot - ok);
+          warmupStatusEl.textContent = `✅ Terminé — ${ok}/${tot} téléchargés, ${fail} échec(s) [inst:${instanceId}]`;
+          if (warmupPopupStatus) warmupPopupStatus.textContent = 'Terminé';
+          if (warmupLogs) warmupLogs.textContent = buildFinalSummary(s);
+        } else {
+          const nCache = Number.isFinite(s.audit_count)
+            ? s.audit_count
+            : (Array.isArray(s.audit_cached) ? s.audit_cached.length : 0);
+          warmupStatusEl.textContent = nCache > 0
+            ? `✅ Prêt — cache local: ${nCache} [inst:${instanceId}]`
+            : `Prêt (aucun run)`;
+        }
       }
-    }
 
-  }
 }
 
     if (warmupProgressFill) warmupProgressFill.style.width = pct + '%';