Deploy public authenticated AutoML gateway

gateway_app.py

@@ -1875,38 +1875,46 @@ def _staged_visual_update(state: Any, index: int) -> Any:
     updates = _staged_visual_updates(state)
     update = updates[index]
     if (
-        index > 0
-        and isinstance(update, Mapping)
+        isinstance(update, Mapping)
         and update.get("visible") is False
-        and update.get("value") is None
+        and update.get("value") in (None, "")
     ):
-        # Gradio 6.17 can abort a chained event after applying an explicit
-        # hidden/empty Image update.  Returning ``gr.skip()`` also terminates
-        # the remaining ``then`` callbacks in the live client, so preserve the
-        # chain with a visibility-only update.  Model/package changes already
-        # run the canonical reset handler that clears prior visual values.
+        # Keep an empty warning as a visibility-only change.  Gradio 6.17 can
+        # abort a chained event when it receives an explicit hidden/empty
+        # value, while ``gr.skip()`` terminates the remaining callbacks.
         return gr.update(visible=False)
     return update
 
 
-def _staged_visual_warning(state: Any) -> Any:
-    return _staged_visual_update(state, 0)
-
-
-def _staged_visual_source(state: Any) -> Any:
-    return _staged_visual_update(state, 1)
-
+def _staged_visual_gallery(state: Any) -> Any:
+    """Present task-relevant visual artifacts in one stable, captioned gallery."""
 
-def _staged_visual_annotated(state: Any) -> Any:
-    return _staged_visual_update(state, 2)
-
-
-def _staged_visual_mask(state: Any) -> Any:
-    return _staged_visual_update(state, 3)
+    updates = _staged_visual_updates(state)
+    visible_paths: dict[int, str] = {}
+    for index in range(1, 5):
+        update = updates[index]
+        if not isinstance(update, Mapping) or update.get("visible") is not True:
+            continue
+        value = str(update.get("value") or "").strip()
+        if value:
+            visible_paths[index] = value
+
+    captions = {
+        1: "Original image",
+        2: "Annotated prediction",
+        3: "Predicted binary mask",
+        4: (
+            "Image + mask overlay"
+            if 3 in visible_paths
+            else "Instance-mask overlay"
+        ),
+    }
+    items = [(path, captions[index]) for index, path in visible_paths.items()]
+    return gr.update(value=items or None, visible=bool(items))
 
 
-def _staged_visual_overlay(state: Any) -> Any:
-    return _staged_visual_update(state, 4)
+def _staged_visual_warning(state: Any) -> Any:
+    return _staged_visual_update(state, 0)
 
 
 def _existing_payload(
@@ -2290,6 +2298,19 @@ def _guided_visual_outputs(outputs: Any) -> tuple[Any, ...]:
     return _visual_ui_updates(value, warning=warning)
 
 
+def _guided_gallery_outputs(outputs: Any) -> tuple[Any, ...]:
+    """Adapt parallel-run visual sidecars to the stable gallery presentation."""
+
+    updates = _guided_visual_outputs(outputs)
+    return (
+        _staged_visual_update(updates, 0),
+        _staged_visual_gallery(updates),
+        gr.update(value=None, visible=False),
+        gr.update(value=None, visible=False),
+        gr.update(value=None, visible=False),
+    )
+
+
 def guided_run_ui_handler(
     state_data,
     data_source,
@@ -2580,7 +2601,7 @@ def guided_parallel_plan_ui_handler(
         for value in (
             *outputs,
             *_guided_snapshot_outputs(variant_state, {}),
-            *_guided_visual_outputs(outputs),
+            *_guided_gallery_outputs(outputs),
         )
     )
 
@@ -2597,7 +2618,7 @@ def guided_parallel_plan_ui_handler(
         for value in (
             *outputs,
             *_guided_snapshot_outputs(variant_state, outputs),
-            *_guided_visual_outputs(outputs),
+            *_guided_gallery_outputs(outputs),
         )
     )
 
@@ -3244,34 +3265,34 @@ def _add_results_section(label: str = "Latest run"):
                     visible=False,
                     elem_classes=["visual-prediction-warning"],
                 )
-                with gr.Row(equal_height=True):
-                    visual_source = gr.Image(
-                        label="Original image",
-                        interactive=False,
-                        visible=False,
-                        type="filepath",
-                        elem_classes=["visual-prediction-image"],
-                    )
+                visual_source = gr.Gallery(
+                    label="Visual prediction",
+                    interactive=False,
+                    visible=False,
+                    type="filepath",
+                    columns=3,
+                    rows=1,
+                    height="auto",
+                    object_fit="contain",
+                    allow_preview=True,
+                    fit_columns=True,
+                    elem_classes=["visual-prediction-gallery"],
+                )
+                with gr.Row(visible=False):
                     visual_annotated = gr.Image(
-                        label="Annotated prediction",
                         interactive=False,
                         visible=False,
                         type="filepath",
-                        elem_classes=["visual-prediction-image"],
                     )
                     visual_mask = gr.Image(
-                        label="Predicted binary mask",
                         interactive=False,
                         visible=False,
                         type="filepath",
-                        elem_classes=["visual-prediction-image"],
                     )
                     visual_overlay = gr.Image(
-                        label="Mask overlay",
                         interactive=False,
                         visible=False,
                         type="filepath",
-                        elem_classes=["visual-prediction-image"],
                     )
                 predictions = gr.DataFrame(
                     label="Held-out prediction sample",
@@ -4152,35 +4173,34 @@ def create_interface() -> gr.Blocks:
                         predict_visual_staging_state = gr.State(
                             _staged_visual_updates(())
                         )
-                        with gr.Row(equal_height=True):
-                            predict_visual_source = gr.Image(
-                                label="Original image",
-                                interactive=False,
-                                visible=False,
-                                type="filepath",
-                                elem_classes=["visual-prediction-image"],
-                            )
+                        predict_visual_source = gr.Gallery(
+                            label="Visual prediction",
+                            interactive=False,
+                            visible=False,
+                            type="filepath",
+                            columns=3,
+                            rows=1,
+                            height="auto",
+                            object_fit="contain",
+                            allow_preview=True,
+                            fit_columns=True,
+                            elem_classes=["visual-prediction-gallery"],
+                        )
+                        with gr.Row(visible=False):
                             predict_visual_annotated = gr.Image(
-                                label="Annotated prediction",
                                 interactive=False,
                                 visible=False,
                                 type="filepath",
-                                elem_classes=["visual-prediction-image"],
                             )
-                        with gr.Row(equal_height=True):
                             predict_visual_mask = gr.Image(
-                                label="Predicted binary mask",
                                 interactive=False,
                                 visible=False,
                                 type="filepath",
-                                elem_classes=["visual-prediction-image"],
                             )
                             predict_visual_overlay = gr.Image(
-                                label="Mask overlay",
                                 interactive=False,
                                 visible=False,
                                 type="filepath",
-                                elem_classes=["visual-prediction-image"],
                             )
                         predict_frame = gr.DataFrame(
                             label="Prediction",
@@ -4822,22 +4842,18 @@ def create_interface() -> gr.Blocks:
             [*latest_outputs[:16], latest_visual_staging_state],
             api_name=False,
         )
-        for visual_fn, visual_output in zip(
-            (
-                _staged_visual_warning,
-                _staged_visual_source,
-                _staged_visual_annotated,
-                _staged_visual_mask,
-                _staged_visual_overlay,
-            ),
-            latest_outputs[16:21],
-        ):
-            primary_run_event = primary_run_event.then(
-                visual_fn,
-                latest_visual_staging_state,
-                visual_output,
-                api_name=False,
-            )
+        primary_run_event = primary_run_event.then(
+            _staged_visual_gallery,
+            latest_visual_staging_state,
+            latest_outputs[17],
+            api_name=False,
+        )
+        primary_run_event = primary_run_event.then(
+            _staged_visual_warning,
+            latest_visual_staging_state,
+            latest_outputs[16],
+            api_name=False,
+        )
         primary_run_event.then(
             _guided_run_button_ready,
             inputs=[],
@@ -4891,28 +4907,18 @@ def create_interface() -> gr.Blocks:
             ],
             api_name=False,
         )
-        for visual_fn, visual_output in zip(
-            (
-                _staged_visual_warning,
-                _staged_visual_source,
-                _staged_visual_annotated,
-                _staged_visual_mask,
-                _staged_visual_overlay,
-            ),
-            (
-                predict_visual_warning,
-                predict_visual_source,
-                predict_visual_annotated,
-                predict_visual_mask,
-                predict_visual_overlay,
-            ),
-        ):
-            predict_event = predict_event.then(
-                visual_fn,
-                predict_visual_staging_state,
-                visual_output,
-                api_name=False,
-            )
+        predict_event = predict_event.then(
+            _staged_visual_gallery,
+            predict_visual_staging_state,
+            predict_visual_source,
+            api_name=False,
+        )
+        predict_event.then(
+            _staged_visual_warning,
+            predict_visual_staging_state,
+            predict_visual_warning,
+            api_name=False,
+        )
     return demo