Update app.py

app.py

@@ -410,6 +410,74 @@ async def upload(request: Request, file: UploadFile = File(...), redirect: Optio
 def progress(vid_stem: str):
     return progress_data.get(vid_stem, {'percent': 0, 'logs': [], 'done': False})
 
+# >>> B1_BEGIN warmup_routes
+@app.post("/warmup/start", tags=["warmup"])
+async def warmup_start(request: Request, payload: Optional[Dict[str, Any]] = Body(None)):
+    """
+    Démarre un warm-up séquentiel.
+    Sources de la liste des modèles (dans cet ordre) :
+    1) payload JSON: {"models": ["repo/a", "repo/b", ...]}
+    2) query string: ?models=repo/a,repo/b
+    3) env WARMUP_MODELS (JSON array)
+    """
+    with warmup_lock:
+        if warmup_state.get("running"):
+            return {
+                "ok": False,
+                "running": True,
+                "msg": "Warm-up déjà en cours",
+                "percent": warmup_state.get("percent", 0),
+                "current": warmup_state.get("current"),
+                "idx": warmup_state.get("idx", 0),
+                "total": warmup_state.get("total", 0),
+            }
+
+    # 1) payload JSON
+    models_list: List[str] = []
+    if isinstance(payload, dict):
+        ml = payload.get("models")
+        if isinstance(ml, list):
+            models_list = [str(x).strip() for x in ml if str(x).strip()]
+
+    # 2) query string
+    if not models_list:
+        qs_models = request.query_params.get("models") or ""
+        if qs_models:
+            models_list = [s.strip() for s in qs_models.split(",") if s.strip()]
+
+    # 3) env
+    if not models_list:
+        models_list = _default_model_list()
+
+    if not models_list:
+        raise HTTPException(400, "Aucun modèle fourni (body.models ou ?models=) et WARMUP_MODELS vide.")
+
+    t = threading.Thread(target=_warmup_thread, args=(models_list,), daemon=True)
+    t.start()
+    return {"ok": True, "started": True, "total": len(models_list)}
+
+@app.get("/warmup/status", tags=["warmup"])
+def warmup_status():
+    """
+    État courant du warm-up + 200 dernières lignes de logs.
+    """
+    with warmup_lock:
+        state = dict(warmup_state)
+        state["logs"] = warmup_state.get("logs", [])[-200:]
+    return state
+
+@app.post("/warmup/stop", tags=["warmup"])
+def warmup_stop_route():
+    """
+    Demande d'arrêt gracieux. Le modèle en cours se termine puis la boucle stoppe.
+    """
+    if not warmup_state.get("running"):
+        return {"ok": True, "was_running": False}
+    warmup_stop.set()
+    return {"ok": True, "was_running": True}
+# >>> B1_END warmup_routes
+
+
 @app.delete("/delete/{vid}", tags=["io"])
 def delete_video(vid: str):
     v = DATA_DIR / vid