Update app.py

app.py

@@ -159,6 +159,7 @@ def _warmup_thread(models: List[str]):
         warmup_state.update({
             "running": True,
             "percent": 0,
+            "ok_count": 0,
             "done": False,
             "logs": [],
             "current": None,
@@ -180,6 +181,8 @@ def _warmup_thread(models: List[str]):
         if ok:
             ok_count += 1
             _log_warmup(f"OK: {repo}")
+            with warmup_lock:
+                warmup_state["ok_count"] = ok_count
         else:
             _log_warmup(f"ÉCHEC: {repo}")
 
@@ -187,6 +190,7 @@ def _warmup_thread(models: List[str]):
         warmup_state["percent"] = 100
         warmup_state["done"] = True
         warmup_state["running"] = False
+        warmup_state["ok_count"] = ok_count
     _log_warmup(f"Terminé: {ok_count}/{len(models)} modèles.")
 # <<<< HUGINFACE PATCH: WARMUP STATE+HELPERS END >>>>
 # ---------- Helpers ----------
@@ -929,12 +933,6 @@ const DEFAULT_MODELS = [
 ];
 // >>> A2B2P3_END warmup_defaults
 
-
-
-
-
-
-
 function openWarmupPopup(){ if(warmupPopup) warmupPopup.style.display = 'block'; }
 
 function closeWarmupPopup(){ if(warmupPopup) warmupPopup.style.display = 'none'; }
@@ -970,13 +968,18 @@ async function refreshWarmupUI(){
     const idx = (s.idx ?? 0) + 1;
         warmupStatusEl.textContent = `⏳ ${pct}% — ${s.current||''} (${idx}/${tot}) [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 = `✅ Terminé — cache local: ${nCache} (inst:${instanceId})`;
+        // Si le run vient de se terminer et que le backend expose ok_count, on priorise ce résumé
+    if (s.done && Number.isFinite(s.ok_count)) {
+      warmupStatusEl.textContent = `✅ Terminé — ${s.ok_count}/${tot} téléchargés (inst:${instanceId})`;
     } else {
-      warmupStatusEl.textContent = `Prêt (aucun run)`;
+      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)`;
+      }
     }
 
   }
@@ -985,31 +988,25 @@ async function refreshWarmupUI(){
     if (warmupProgressFill) warmupProgressFill.style.width = pct + '%';
     if (warmupPopupStatus)  warmupPopupStatus.textContent = running ? 'Téléchargement en cours…' : 'Terminé';
 
-
-
-
-
-    // Préambule construit dynamiquement à partir de /warmup/status
-    let prefaceNow = '';
-    if (Array.isArray(s.audit_cached)) {
-            const n = Number.isFinite(s.audit_count) ? s.audit_count : s.audit_cached.length;
-      prefaceNow =
-        `[Instance ${instanceId}]` + '\n' +
-        'Déjà en cache (' + n + '):\n' +
-        s.audit_cached.map(m => ' • ' + m).join('\n');
-
-    }
-    if (Array.isArray(window.lastRequestedModels) && window.lastRequestedModels.length) {
-      prefaceNow += (prefaceNow ? '\n\n' : '') +
-        'Demandé dans cette exécution (' + window.lastRequestedModels.length + '):\n' +
-        window.lastRequestedModels.map(m => ' • ' + m).join('\n');
-    }
+    // PRÉAMBULE FIXE (conservé pendant tout le run)
     const logsTxt = Array.isArray(s.logs) ? s.logs.join('\n') : '';
-    if (warmupLogs) warmupLogs.textContent = (prefaceNow ? (prefaceNow + '\n\n') : '') + logsTxt;
-
-
-
-
+    const fixedPreface = (warmupPreface && warmupPreface.trim().length)
+      ? warmupPreface
+      : (()=>{
+          const n = Number.isFinite(s.audit_count)
+            ? s.audit_count
+            : (Array.isArray(s.audit_cached) ? s.audit_cached.length : 0);
+          const cachedList = Array.isArray(s.audit_cached) ? s.audit_cached : [];
+          const asked = Array.isArray(window.lastRequestedModels) ? window.lastRequestedModels : [];
+          return (
+            `[Instance ${instanceId}]` + '\n' +
+            'Déjà en cache (' + n + '):\n' +
+            cachedList.map(m => ' • ' + m).join('\n') + '\n\n' +
+            'Demandé dans cette exécution (' + asked.length + '):\n' +
+            asked.map(m => ' • ' + m).join('\n')
+          );
+        })();
+    if (warmupLogs) warmupLogs.textContent = fixedPreface + '\n\n' + logsTxt;
 
 
     if (warmupStopBtn) warmupStopBtn.style.display = running ? 'inline-block' : 'none';
@@ -1092,6 +1089,14 @@ if (warmupStartBtn){
         alert('Échec démarrage: ' + r.status + ' ' + t);
         return;
       }
+      // Si un warm-up tourne déjà, on l'indique et on passe en mode "suivi"
+      try {
+        const payload = await r.json();
+        if (payload && payload.already_running) {
+          showToast("Un warm-up est déjà en cours — j'affiche l'état.");
+        }
+      } catch(e) { /* no-op */ }
+
       // Rafraîchit l’UI et démarre le polling
       await refreshWarmupUI();
       if (!warmupTimer) warmupTimer = setInterval(refreshWarmupUI, 1000);
@@ -1102,11 +1107,6 @@ if (warmupStartBtn){
   });
 }
 
-
-
-
-
-
 if (warmupLogsBtn){
   warmupLogsBtn.addEventListener('click', async ()=>{