Update app.py

app.py

@@ -133,24 +133,42 @@ def _log_warmup(msg: str):
         if len(warmup_state["logs"]) > 400:
             warmup_state["logs"] = warmup_state["logs"][-400:]
 
+def _dir_size_bytes(path: str) -> int:
+    try:
+        total = 0
+        for root, _, files in os.walk(path):
+            for f in files:
+                try:
+                    total += os.path.getsize(os.path.join(root, f))
+                except Exception:
+                    pass
+        return total
+    except Exception:
+        return 0
+
 def _download_one(repo_id: str, tries: int = 3) -> bool:
     """
-    Télécharge un repo HF en réessayant si besoin.
+    Télécharge un repo HF en réessayant si besoin, avec logs précis.
     """
     cache_home = os.path.expanduser(os.getenv("HF_HOME", "/home/user/.cache/huggingface"))
     local_dir = os.path.join(cache_home, "models", repo_id.replace("/", "__"))
-    # Déjà présent localement ? On log et on saute proprement.
+
+    # Cache déjà présent ?
     try:
         if _is_repo_cached(repo_id):
-            _log_warmup(f"Déjà en cache: {repo_id} (skip)")
+            sz = _dir_size_bytes(local_dir)
+            _log_warmup(f"[CACHE] {repo_id} • {local_dir} • {sz/1e6:.1f} MB (skip)")
             return True
     except Exception:
-        # En cas d'erreur d'audit, on continue le process normal
         pass
 
     for attempt in range(1, tries + 1):
         if warmup_stop.is_set():
+            _log_warmup(f"[STOP] Abandon demandé avant téléchargement de {repo_id}")
             return False
+
+        t0 = time.time()
+        _log_warmup(f"[START] {repo_id} • tentative {attempt}/{tries} • {local_dir}")
         try:
             snapshot_download(
                 repo_id,
@@ -158,9 +176,13 @@ def _download_one(repo_id: str, tries: int = 3) -> bool:
                 local_dir_use_symlinks=False,
                 resume_download=True,
             )
+            dt = time.time() - t0
+            sz = _dir_size_bytes(local_dir)
+            _log_warmup(f"[DONE]  {repo_id} • {dt:.1f}s • {sz/1e6:.1f} MB")
             return True
         except Exception as e:
-            _log_warmup(f"{repo_id} -> tentative {attempt}/{tries} échouée: {e}")
+            dt = time.time() - t0
+            _log_warmup(f"[FAIL]  {repo_id} • {dt:.1f}s • {type(e).__name__}: {e}")
             time.sleep(min(10, 2 * attempt))
     return False
 
@@ -178,33 +200,39 @@ def _warmup_thread(models: List[str]):
         })
     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("Arrêt demandé par l’utilisateur.")
+            _log_warmup("[STOP] Arrêt demandé par l’utilisateur — fin du job après ce point")
             break
+
         with warmup_lock:
             warmup_state["idx"] = i
             warmup_state["current"] = repo
             warmup_state["percent"] = int((i / max(1, len(models))) * 100)
-        _log_warmup(f"Téléchargement: {repo}")
+
         ok = _download_one(repo)
         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}")
-        # Met à jour la progression globale après ce repo (ok ou échec)
+        # progression après dépôt (OK ou FAIL)
         with warmup_lock:
             warmup_state["percent"] = int(((i + 1) / max(1, len(models))) * 100)
 
+        if warmup_stop.is_set():
+            _log_warmup("[STOP] Fin anticipée — demande reçue pendant le run")
+            break
+
     with warmup_lock:
         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.")
+    _log_warmup(f"[JOB] done • {ok_count}/{len(models)} • {time.time()-t_global:.1f}s")
+
 # <<<< HUGINFACE PATCH: WARMUP STATE+HELPERS END >>>>
 # ---------- Helpers ----------