Update app.py

app.py

@@ -517,6 +517,60 @@ def poster(vid: str):
         return FileResponse(str(p), media_type="image/jpeg")
     raise HTTPException(404, "Poster introuvable")
 
+# >>> B1_BEGIN warmup_routes
+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).
+    """
+    models = []
+    if payload and isinstance(payload, dict):
+        models = [str(x).strip() for x in (payload.get("models") or []) if str(x).strip()]
+    if not models:
+        models = _default_model_list()
+    if not models:
+        raise HTTPException(400, "Aucun modèle fourni (payload.models) et WARMUP_MODELS vide")
+
+    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
+        job_id = uuid.uuid4().hex[:8]
+        warmup_state["job_id"] = job_id
+        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.). """
+    with warmup_lock:
+        data = deepcopy(warmup_state)
+    data["ts"] = time.time()
+    return data
+
+@app.post("/warmup/stop", tags=["warmup"])
+def warmup_stop_api():
+    """ Demande l'arrêt propre (on termine le modèle en cours puis on stoppe). """
+    with warmup_lock:
+        running = bool(warmup_state.get("running"))
+    if running:
+        warmup_stop.set()
+        return {"ok": True, "was_running": True}
+    return {"ok": True, "was_running": False}
+# >>> B1_END warmup_routes
+
+
 # >>> A1_BEGIN window_fix
 @app.get("/window/{vid}", tags=["io"])
 def window(vid: str, center: int = 0, count: int = 21):