Update app.py

app.py

@@ -589,7 +589,33 @@ def warmup_audit():
         "present_default": present_default,
         "missing_default": missing_default,
     }
-# >>> B2_END warmup_audit
+@app.get("/warmup/catalog", tags=["warmup"])
+def warmup_catalog():
+    try:
+        env = (os.getenv("WARMUP_MODELS") or "").strip()
+        base = []
+        if env:
+            try:
+                lst = json.loads(env)
+                if isinstance(lst, list) and lst:
+                    base = [str(x).strip() for x in lst if str(x).strip()]
+            except Exception:
+                base = []
+    except Exception:
+        base = []
+
+    if not base:
+        base = [
+            "runwayml/stable-diffusion-v1-5",
+            "facebook/sam-vit-base",
+            "stabilityai/sd-vae-ft-mse",
+            "lixiaowen/diffuEraser",
+            "facebook/sam2-hiera-base",
+        ]
+
+    seen = set()
+    out = [m for m in base if not (m in seen or seen.add(m))]
+    return {"count": len(out), "models": out}
 
 
 # >>> A1_BEGIN window_fix
@@ -755,6 +781,11 @@ HTML_TEMPLATE = r"""
  #portionBand{position:absolute;top:0;height:calc(var(--thumbH,110px) + 24px);background:rgba(37,99,235,.12);pointer-events:none;display:none}
  #inHandle,#outHandle{position:absolute;top:0;height:calc(var(--thumbH,110px) + 24px);width:6px;background:rgba(37,99,235,.9);border-radius:2px;cursor:ew-resize;display:none;pointer-events:auto}
  .playhead{position:absolute;top:0;bottom:0;width:2px;background:var(--active-border);opacity:.9;pointer-events:none;display:block}
+ .repo-list{max-height:320px; overflow:auto; border:1px solid var(--b); border-radius:8px; padding:6px}
+ .repo-item{display:flex; align-items:center; gap:8px; padding:6px; border-bottom:1px solid #f3f4f6}
+ .repo-item:last-child{border-bottom:none}
+ .repo-id{font-family:ui-monospace, SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", monospace; font-size:12px}
+
 </style>
 <h1>🎬 Video Editor</h1>
 <div class="topbar card">
@@ -776,6 +807,8 @@ HTML_TEMPLATE = r"""
 <div style="display:flex;gap:8px;align-items:center">
   <button id="warmupStartBtn" class="btn" title="Télécharger les modèles (séquentiel)">🔥 Préparer modèles</button>
   <button id="warmupStopBtn" class="btn" title="Arrêter le warm-up en cours" style="display:none">⏹️ Stop warm-up</button>
+  <button id="warmupSelectBtn" class="btn" title="Sélectionner les repos à préparer">🧩 Choisir modèles</button>
+
   <button id="warmupLogsBtn" class="btn" title="Voir les logs du warm-up">📜 Logs warm-up</button>
   <span id="warmupStatus" class="muted"></span>
 </div>
@@ -867,49 +900,16 @@ HTML_TEMPLATE = r"""
     <button id="warmupCloseBtn" class="btn">Fermer</button>
   </div>
 </div>
-<!-- >>> A2B2_END warmup_popup -->
-
-
-<div id="toast"></div>
-<script>
-const serverVid = "__VID__";
-const serverMsg = "__MSG__";
-document.getElementById('msg').textContent = serverMsg;
-
-// Elements
-const statusEl = document.getElementById('status');
-const player = document.getElementById('player');
-const srcEl = document.getElementById('vidsrc');
-const canvas = document.getElementById('editCanvas');
-const ctx = canvas.getContext('2d');
-const modeLabel = document.getElementById('modeLabel');
-const btnEdit = document.getElementById('btnEdit');
-const btnBack = document.getElementById('btnBack');
-const btnSave = document.getElementById('btnSave');
-const btnClear= document.getElementById('btnClear');
-const posInfo = document.getElementById('posInfo');
-const goFrame = document.getElementById('goFrame');
-const palette = document.getElementById('palette');
-const fileList= document.getElementById('fileList');
-const tlBox = document.getElementById('timeline');
-const tlNote = document.getElementById('tlNote');
-const playerWrap = document.getElementById('playerWrap');
-const loadingInd = document.getElementById('loading-indicator');
-const isolerBoucle = document.getElementById('isolerBoucle');
-const resetFull = document.getElementById('resetFull');
-const endPortion = document.getElementById('endPortion');
-const popup = document.getElementById('popup');
-const popupLogs = document.getElementById('popup-logs');
-const tlProgressFill = document.getElementById('tl-progress-fill');
-const popupProgressFill = document.getElementById('popup-progress-fill');
-const btnFollow = document.getElementById('btnFollow');
-const btnFilterMasked = document.getElementById('btnFilterMasked');
-const zoomSlider = document.getElementById('zoomSlider');
-const maskedCount = document.getElementById('maskedCount');
-const hud = document.getElementById('hud');
-const toastWrap = document.getElementById('toast');
-const gotoInput = document.getElementById('gotoInput');
-
+<div id="warmupSelectPopup" style="display:none; position:fixed; top:24%; left:50%; transform:translate(-50%,-50%); background:#fff; padding:20px; border-radius:8px; box-shadow:0 0 10px rgba(0,0,0,0.2); z-index:1001; min-width:420px; max-width:720px; width:90%">
+  <h3>Sélection des modèles à préparer</h3>
+  <div id="repoList" class="repo-list"></div>
+  <div style="margin-top:12px; display:flex; gap:8px; justify-content:flex-end; flex-wrap:wrap">
+    <button id="selectAllBtn" class="btn">Tout cocher</button>
+    <button id="selectNoneBtn" class="btn">Tout décocher</button>
+    <button id="selectCancelBtn" class="btn">Annuler</button>
+    <button id="launchSelectedBtn" class="btn">Lancer la préparation</button>
+  </div>
+</div>
 
 
 // >>> A2B2_BEGIN warmup_dom
@@ -1162,6 +1162,90 @@ if (warmupStopBtn){
 refreshWarmupUI();
 // >>> A2B2_END warmup_dom
 
+const warmupSelectBtn = document.getElementById('warmupSelectBtn');
+const warmupSelectPopup = document.getElementById('warmupSelectPopup');
+const repoListEl = document.getElementById('repoList');
+const selectCancelBtn = document.getElementById('selectCancelBtn');
+const selectAllBtn = document.getElementById('selectAllBtn');
+const selectNoneBtn = document.getElementById('selectNoneBtn');
+const launchSelectedBtn = document.getElementById('launchSelectedBtn');
+
+function openSelect(){ if(warmupSelectPopup) warmupSelectPopup.style.display='block'; }
+function closeSelect(){ if(warmupSelectPopup) warmupSelectPopup.style.display='none'; }
+
+async function loadCatalog(){
+  if(!repoListEl) return;
+  repoListEl.innerHTML = '<div class="muted">Chargement…</div>';
+  try{
+    const r = await fetch('/warmup/catalog');
+    if(!r.ok){ repoListEl.innerHTML = '<div class="muted">Erreur de chargement.</div>'; return; }
+    const d = await r.json();
+    const items = Array.isArray(d.models) ? d.models : [];
+    if(!items.length){ repoListEl.innerHTML = '<div class="muted">Aucun modèle proposé.</div>'; return; }
+    repoListEl.innerHTML = '';
+    items.forEach(id=>{
+      const row = document.createElement('div'); row.className='repo-item';
+      const cb  = document.createElement('input'); cb.type='checkbox'; cb.value=id;
+      const label = document.createElement('span'); label.className='repo-id'; label.textContent = id;
+      row.appendChild(cb); row.appendChild(label);
+      repoListEl.appendChild(row);
+    });
+  }catch(e){
+    repoListEl.innerHTML = '<div class="muted">Erreur réseau.</div>';
+  }
+}
+
+if (warmupSelectBtn){
+  warmupSelectBtn.addEventListener('click', async ()=>{
+    await loadCatalog();
+    openSelect();
+  });
+}
+if (selectCancelBtn){ selectCancelBtn.addEventListener('click', closeSelect); }
+if (selectAllBtn){
+  selectAllBtn.addEventListener('click', ()=>{
+    repoListEl.querySelectorAll('input[type="checkbox"]').forEach(cb=> cb.checked=true);
+  });
+}
+if (selectNoneBtn){
+  selectNoneBtn.addEventListener('click', ()=>{
+    repoListEl.querySelectorAll('input[type="checkbox"]').forEach(cb=> cb.checked=false);
+  });
+}
+if (launchSelectedBtn){
+  launchSelectedBtn.addEventListener('click', async ()=>{
+    const picks = Array.from(repoListEl.querySelectorAll('input[type="checkbox"]'))
+      .filter(cb=>cb.checked).map(cb=>cb.value);
+    if(!picks.length){ alert('Sélectionne au moins un modèle.'); return; }
+    window.lastRequestedModels = picks.slice();
+    if (typeof openWarmupPopup === 'function') openWarmupPopup();
+    if (typeof refreshWarmupUI === 'function') await refreshWarmupUI();
+    try{
+      const r = await fetch('/warmup/start', {
+        method:'POST',
+        headers:{ 'Content-Type':'application/json' },
+        body: JSON.stringify({ models: picks })
+      });
+      if(!r.ok){
+        const t = await r.text();
+        alert('Échec démarrage: ' + r.status + ' ' + t);
+        return;
+      }
+      if (typeof refreshWarmupUI === 'function') {
+        const id = setInterval(async ()=>{
+          await refreshWarmupUI();
+          try{
+            const s = await (await fetch('/warmup/status')).json();
+            if (!s.running) clearInterval(id);
+          }catch(e){ clearInterval(id); }
+        }, 1000);
+      }
+      closeSelect();
+    }catch(e){
+      alert('Erreur réseau.');
+    }
+  });
+}