Update app.py

app.py

@@ -358,8 +358,9 @@ def _gen_thumbs_background(video: Path, vid_stem: str):
 def root():
     return {
         "ok": True,
-        "routes": ["/", "/health", "/files", "/upload", "/meta/{vid}", "/frame_idx", "/poster/{vid}", "/window/{vid}",
-                   "/mask", "/mask/{vid}", "/mask/rename", "/mask/delete", "/progress/{vid_stem}", "/ui"]
+                "routes": ["/", "/health", "/files", "/upload", "/meta/{vid}", "/frame_idx", "/poster/{vid}", "/window/{vid}",
+                   "/mask", "/mask/{vid}", "/mask/rename", "/mask/delete", "/progress/{vid_stem}", "/ui",
+                   "/warmup/start", "/warmup/status", "/warmup/stop", "/warmup/audit"]
     }
 
 @app.get("/health", tags=["meta"])
@@ -504,6 +505,47 @@ def warmup_stop_api():
     return {"ok": True, "was_running": False}
 # >>> B1_END warmup_routes
 
+# >>> B2_BEGIN warmup_audit
+def _hf_cache_dir() -> str:
+    return os.path.expanduser(os.getenv("HF_HOME", "/home/user/.cache/huggingface"))
+
+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 _list_cached_repos() -> List[str]:
+    base = os.path.join(_hf_cache_dir(), "models")
+    out: List[str] = []
+    try:
+        for e in os.scandir(base):
+            if e.is_dir():
+                out.append(e.name.replace("__", "/"))
+    except FileNotFoundError:
+        pass
+    return sorted(out)
+
+@app.get("/warmup/audit", tags=["warmup"])
+def warmup_audit():
+    """
+    Retourne l'état du cache HF côté serveur.
+    - cached: tous les repos détectés localement
+    - present_default / missing_default: croisement avec WARMUP_MODELS (si défini)
+    """
+    cached = _list_cached_repos()
+    defaults = _default_model_list()
+    present_default = [r for r in defaults if r in cached]
+    missing_default = [r for r in defaults if r not in cached]
+    return {
+        "count": len(cached),
+        "cached": cached,
+        "present_default": present_default,
+        "missing_default": missing_default,
+    }
+# >>> B2_END warmup_audit
+
 
 # >>> A1_BEGIN window_fix
 @app.get("/window/{vid}", tags=["io"])
@@ -841,6 +883,9 @@ const warmupCloseBtn     = document.getElementById('warmupCloseBtn');
 // Helpers popup
 
 
+// >>> C2_BEGIN warmup_preface
+let warmupPreface = '';
+// >>> C2_END warmup_preface
 
 
 
@@ -886,7 +931,20 @@ async function refreshWarmupUI(){
     }
     if (warmupProgressFill) warmupProgressFill.style.width = pct + '%';
     if (warmupPopupStatus)  warmupPopupStatus.textContent = running ? 'Téléchargement en cours…' : 'Terminé';
-    if (warmupLogs && Array.isArray(s.logs)) warmupLogs.textContent = s.logs.join('\n');
+
+
+
+
+
+
+
+    const logsTxt = Array.isArray(s.logs) ? s.logs.join('\n') : '';
+    if (warmupLogs) warmupLogs.textContent = (warmupPreface ? (warmupPreface + '\n') : '') + logsTxt;
+
+
+
+
+
     if (warmupStopBtn) warmupStopBtn.style.display = running ? 'inline-block' : 'none';
 
     if (running) {
@@ -923,14 +981,33 @@ if (warmupStartBtn){
       models = DEFAULT_MODELS;
     }
 
+
+    // Préface (audit du cache serveur, pour afficher le cumul)
+    try {
+      const ra = await fetch('/warmup/audit');
+      const audit = ra.ok ? await ra.json() : null;
+      const cached = (audit && Array.isArray(audit.cached)) ? audit.cached : [];
+      warmupPreface =
+        'Déjà en cache (' + cached.length + '):\n' +
+        cached.map(m => ' • ' + m).join('\n') + '\n\n' +
+        'Demandé dans cette exécution (' + models.length + '):\n' +
+        models.map(m => ' • ' + m).join('\n');
+    } catch(e) {
+      warmupPreface = '';
+    }
+
+
+
+
     // Ouvre la popup et affiche immédiatement la liste demandée
     openWarmupPopup();
     if (warmupPopupStatus) warmupPopupStatus.textContent = 'Préparation en cours…';
-    if (warmupLogs) {
-      warmupLogs.textContent = 'Warm-up demandé pour :\n' + models.map(m=>' • '+m).join('\n');
+        if (warmupLogs) {
+      warmupLogs.textContent = (warmupPreface ? warmupPreface + '\n' : '') + '— démarrage…';
       warmupLogs.scrollTop = warmupLogs.scrollHeight; // auto-scroll vers le bas
     }
 
+
     // Sécurité : évite 2 timers concurrents
     if (warmupTimer) { clearInterval(warmupTimer); warmupTimer = null; }
 
@@ -962,6 +1039,20 @@ if (warmupStartBtn){
 
 if (warmupLogsBtn){
   warmupLogsBtn.addEventListener('click', async ()=>{
+
+    // Recalcule une préface à la volée (hors exécution)
+    try{
+      const ra = await fetch('/warmup/audit');
+      const audit = ra.ok ? await ra.json() : null;
+      const cached = (audit && Array.isArray(audit.cached)) ? audit.cached : [];
+      if (!warmupPreface) {
+        warmupPreface =
+          'Déjà en cache (' + cached.length + '):\n' +
+          cached.map(m => ' • ' + m).join('\n');
+      }
+    }catch(e){}
+
+
     openWarmupPopup();
     await refreshWarmupUI();
     if (!warmupTimer) warmupTimer = setInterval(refreshWarmupUI, 1000);