Update app.py

app.py

@@ -572,9 +572,17 @@ def conditioner():
     return Client(CONDITIONER_SPACE)
 
 
-def encode_remote(prompt, image_path, last_image_path, canvas, num_frames, rewrite_prompt=False):
+def encode_remote(
+    prompt, image_path, last_image_path, canvas, num_frames, rewrite_prompt=False,
+    semantic_bridge=False, semantic_bridge_alpha=0.10, semantic_bridge_magnitude_match="per_token",
+):
     """`/encode` on the conditioner Space: a safetensors file holding `prompt_embeds` + `text_token_tags`, with the
-    resolved `height` / `width` / `num_frames` in its metadata, plus the plan. `canvas` is the label."""
+    resolved `height` / `width` / `num_frames` in its metadata, plus the plan. `canvas` is the label.
+
+    `semantic_bridge*` are forwarded as-is to the conditioner's own `encode()` — this Space never sees the
+    adapter itself, only whichever `prompt_embeds` comes back, bridged or not per the plan's `semantic_bridge`
+    field.
+    """
     from gradio_client import handle_file
     from safetensors import safe_open
 
@@ -585,6 +593,9 @@ def encode_remote(prompt, image_path, last_image_path, canvas, num_frames, rewri
         canvas=canvas,
         num_frames=num_frames,
         rewrite_prompt=bool(rewrite_prompt),
+        semantic_bridge=bool(semantic_bridge),
+        semantic_bridge_alpha=float(semantic_bridge_alpha),
+        semantic_bridge_magnitude_match=semantic_bridge_magnitude_match,
         api_name="/encode",
     )
     with safe_open(path, framework="pt") as handle:
@@ -867,6 +878,9 @@ def generate(
     chunk_stop=DEFAULT_CHUNK_STOP,
     chunk_state=None,
     momentum=DEFAULT_MOMENTUM,
+    semantic_bridge=False,
+    semantic_bridge_alpha=0.10,
+    semantic_bridge_magnitude_match="per_token",
     progress=gr.Progress(track_tqdm=True),
     *,
     advance: bool = False,
@@ -972,7 +986,9 @@ def generate(
         )
         conditioned = time.time()
         prompt_embeds, text_token_tags, metadata, plan = encode_remote(
-            prompt, effective_first_frame, effective_last_frame, canvas, num_frames, rewrite_prompt=upsample
+            prompt, effective_first_frame, effective_last_frame, canvas, num_frames, rewrite_prompt=upsample,
+            semantic_bridge=semantic_bridge, semantic_bridge_alpha=semantic_bridge_alpha,
+            semantic_bridge_magnitude_match=semantic_bridge_magnitude_match,
         )
         condition_seconds = time.time() - conditioned
         height, width, num_frames = (int(metadata[key]) for key in ("height", "width", "num_frames"))
@@ -1049,6 +1065,8 @@ def generate(
         *post,
         f"seed {int(seed)}",
     ]
+    if plan.get("semantic_bridge"):
+        info.append(f"Semantic Bridge {float(semantic_bridge_alpha):.2f}")
     if stage_enabled:
         info.append(f"staged {steps_done_after}/{int(target_steps)} steps")
     if lora_text:
@@ -1374,6 +1392,23 @@ with gr.Blocks(title="PlagueKind · MiniMax-H3") as demo:
                     value=False,
                     info="Rewrite the prompt on the conditioner Space first, MiniMax's Context-IR style.",
                 )
+                with gr.Accordion("Semantic Bridge", open=False):
+                    gr.Markdown(
+                        "A small, independent adapter that nudges text conditioning toward better constraint "
+                        "adherence — left/right ordering, which hand holds what, material and reflection "
+                        "correctness — without touching the video transformer. Runs on the conditioner Space. "
+                        "Experimental; may help some prompts, do little on others."
+                    )
+                    semantic_bridge = gr.Checkbox(label="Enable Semantic Bridge", value=False)
+                    semantic_bridge_alpha = gr.Slider(
+                        label="alpha", minimum=0.0, maximum=1.0, step=0.01, value=0.10, visible=False
+                    )
+                    semantic_bridge_magnitude_match = gr.Dropdown(
+                        label="magnitude_match",
+                        choices=["per_token", "global", "none"],
+                        value="per_token",
+                        visible=False,
+                    )
                 maximize_gpu = gr.Checkbox(
                     label="Maximize Free Tier ZeroGPU (150 seconds)",
                     value=False,
@@ -1568,6 +1603,12 @@ with gr.Blocks(title="PlagueKind · MiniMax-H3") as demo:
         [chunk_start, chunk_stop, momentum, continue_btn],
         api_name=False,
     )
+    semantic_bridge.change(
+        lambda enabled: tuple(gr.update(visible=enabled) for _ in range(2)),
+        semantic_bridge,
+        [semantic_bridge_alpha, semantic_bridge_magnitude_match],
+        api_name=False,
+    )
 
     controls = [
         prompt,
@@ -1607,6 +1648,9 @@ with gr.Blocks(title="PlagueKind · MiniMax-H3") as demo:
         chunk_stop,
         chunk_state,
         momentum,
+        semantic_bridge,
+        semantic_bridge_alpha,
+        semantic_bridge_magnitude_match,
     ]
 
     # `functools.partial` binds `advance`/`chunk_advance` by keyword regardless of their position in `generate`'s