Update app.py

app.py

@@ -575,6 +575,7 @@ def conditioner():
 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",
+    semantic_bridge_adapter="semantic_bridge_v1",
 ):
     """`/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.
@@ -596,6 +597,7 @@ def encode_remote(
         semantic_bridge=bool(semantic_bridge),
         semantic_bridge_alpha=float(semantic_bridge_alpha),
         semantic_bridge_magnitude_match=semantic_bridge_magnitude_match,
+        semantic_bridge_adapter=semantic_bridge_adapter,
         api_name="/encode",
     )
     with safe_open(path, framework="pt") as handle:
@@ -881,6 +883,7 @@ def generate(
     semantic_bridge=False,
     semantic_bridge_alpha=0.10,
     semantic_bridge_magnitude_match="per_token",
+    semantic_bridge_adapter="semantic_bridge_v1",
     progress=gr.Progress(track_tqdm=True),
     *,
     advance: bool = False,
@@ -989,6 +992,7 @@ def generate(
             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,
+            semantic_bridge_adapter=semantic_bridge_adapter,
         )
         condition_seconds = time.time() - conditioned
         height, width, num_frames = (int(metadata[key]) for key in ("height", "width", "num_frames"))
@@ -1066,7 +1070,7 @@ def generate(
         f"seed {int(seed)}",
     ]
     if plan.get("semantic_bridge"):
-        info.append(f"Semantic Bridge {float(semantic_bridge_alpha):.2f}")
+        info.append(f"Semantic Bridge `{plan.get('semantic_bridge_adapter')}` {float(semantic_bridge_alpha):.2f}")
     if stage_enabled:
         info.append(f"staged {steps_done_after}/{int(target_steps)} steps")
     if lora_text:
@@ -1409,6 +1413,14 @@ with gr.Blocks(title="PlagueKind · MiniMax-H3") as demo:
                         value="per_token",
                         visible=False,
                     )
+                    semantic_bridge_adapter = gr.Dropdown(
+                        label="adapter",
+                        choices=["semantic_bridge_v1", "bunny_action_logic_v1"],
+                        value="semantic_bridge_v1",
+                        visible=False,
+                        info="semantic_bridge_v1: broad (composition, spatial, materials). "
+                        "bunny_action_logic_v1: action/combat, character-action ownership in multi-character scenes.",
+                    )
                 maximize_gpu = gr.Checkbox(
                     label="Maximize Free Tier ZeroGPU (150 seconds)",
                     value=False,
@@ -1604,9 +1616,9 @@ with gr.Blocks(title="PlagueKind · MiniMax-H3") as demo:
         api_name=False,
     )
     semantic_bridge.change(
-        lambda enabled: tuple(gr.update(visible=enabled) for _ in range(2)),
+        lambda enabled: tuple(gr.update(visible=enabled) for _ in range(3)),
         semantic_bridge,
-        [semantic_bridge_alpha, semantic_bridge_magnitude_match],
+        [semantic_bridge_alpha, semantic_bridge_magnitude_match, semantic_bridge_adapter],
         api_name=False,
     )
 
@@ -1651,6 +1663,7 @@ with gr.Blocks(title="PlagueKind · MiniMax-H3") as demo:
         semantic_bridge,
         semantic_bridge_alpha,
         semantic_bridge_magnitude_match,
+        semantic_bridge_adapter,
     ]
 
     # `functools.partial` binds `advance`/`chunk_advance` by keyword regardless of their position in `generate`'s