Load mode toggle JS via demo.load instead of stripped <script> tag

app.py

@@ -33,7 +33,7 @@ torch.backends.cudnn.allow_tf32 = True
 print("[startup] TF32 enabled", flush=True)
 
 print("[startup] importing dimensions...", flush=True)
-from dimensions import compute_output_dimensions, MAX_OUTPUT_DIM_FAST
+from dimensions import compute_output_dimensions, max_dim_for_mode
 print("[startup] importing diffusers...", flush=True)
 from diffusers import FlowMatchEulerDiscreteScheduler
 print("[startup] importing QwenImageEditPlusPipeline...", flush=True)
@@ -316,6 +316,9 @@ with open("static/wire_outputs.js") as _f:
 with open("static/run_preprocess.js") as _f:
     run_preprocess_js = _f.read()
 
+with open("static/mode_toggle.js") as _f:
+    mode_toggle_js = _f.read()
+
 with open("static/negative_prompt.txt") as _f:
     negative_prompt = _f.read().strip()
 
@@ -359,8 +362,7 @@ def infer(images_b64_json, prompt, seed, randomize_seed, guidance_scale, steps,
 
     seed = _resolve_seed(seed, randomize_seed)
     generator = torch.Generator(device=device).manual_seed(seed)
-    max_dim = MAX_OUTPUT_DIM_FAST if mode == "fast" else MAX_OUTPUT_DIM
-    width, height = update_dimensions_on_upload(pil_images[0], max_dim)
+    width, height = update_dimensions_on_upload(pil_images[0], max_dim_for_mode(mode))
     print(f"[infer] input={pil_images[0].size}, output={width}x{height}, seed={seed}")
 
     # Per-step callback: syncs the GPU then records a CUDA event so elapsed_time()
@@ -433,6 +435,7 @@ with gr.Blocks() as demo:
 
     demo.load(fn=None, js=gallery_js)
     demo.load(fn=None, js=wire_outputs_js)
+    demo.load(fn=None, js=mode_toggle_js)
 
     run_btn.click(
         fn=infer,

static/mode_toggle.js

@@ -0,0 +1,8 @@
+window.__selectedMode = 'fast';
+window.__setMode = function(m) {
+    window.__selectedMode = m;
+    var fast = document.getElementById('mode-btn-fast');
+    var hd   = document.getElementById('mode-btn-hd');
+    if (fast) fast.classList.toggle('mode-btn-active', m === 'fast');
+    if (hd)   hd.classList.toggle('mode-btn-active',   m === 'high_detail');
+};

templates/app.html

@@ -179,12 +179,3 @@
     </details>
 
 </div>
-
-<script>
-window.__selectedMode = 'fast';
-window.__setMode = function(m) {{
-    window.__selectedMode = m;
-    document.getElementById('mode-btn-fast').classList.toggle('mode-btn-active', m === 'fast');
-    document.getElementById('mode-btn-hd').classList.toggle('mode-btn-active', m === 'high_detail');
-}};
-</script>