Deploy Space app files

app.py

@@ -169,25 +169,26 @@ def load_instance_frames(
     return frames
 
 
-def render_frame_gallery(frame_items: list[dict[str, Any]], frame_index: float) -> list[tuple[Any, str]]:
-    """Render RGB/Mask/Depth/Normal gallery for one selected frame index (1-based)."""
+def render_frame_images(frame_items: list[dict[str, Any]], frame_index: float) -> list[Any | None]:
+    """Render RGB/Mask/Depth/Normal images for one selected frame index (1-based)."""
     if not frame_items:
-        return []
+        return [
+            gr.update(value=None, label="RGB"),
+            gr.update(value=None, label="Mask"),
+            gr.update(value=None, label="Depth"),
+            gr.update(value=None, label="Normal"),
+        ]
 
     idx = int(round(frame_index)) - 1
     idx = max(0, min(idx, len(frame_items) - 1))
     selected = frame_items[idx]
     frame_id = int(selected["frame_id"])
-
-    gallery: list[tuple[Any, str]] = []
-    for key, label in [
-        ("rgb", "RGB"),
-        ("mask", "Mask"),
-        ("depth_preview", "Depth"),
-        ("normal_preview", "Normal"),
-    ]:
-        gallery.append((selected[key], f"{label} frame_{frame_id:05d}"))
-    return gallery
+    return [
+        gr.update(value=selected["rgb"], label=f"RGB frame_{frame_id:05d}"),
+        gr.update(value=selected["mask"], label=f"Mask frame_{frame_id:05d}"),
+        gr.update(value=selected["depth_preview"], label=f"Depth frame_{frame_id:05d}"),
+        gr.update(value=selected["normal_preview"], label=f"Normal frame_{frame_id:05d}"),
+    ]
 
 
 # ---------------------------------------------------------------------------
@@ -265,11 +266,12 @@ def build_app(instance_df: pd.DataFrame, preview_df: pd.DataFrame) -> gr.Blocks:
         summary = f"Matched **{len(filtered)}** preview instances, showing **{len(rows)}**."
         gallery_items = build_instance_gallery_items(rows, preview_df)
         slider_empty = gr.update(minimum=1, maximum=1, step=1, value=1, interactive=False)
-        return summary, gallery_items, rows, {}, None, [], slider_empty, []
+        return summary, gallery_items, rows, {}, None, None, None, None, None, slider_empty, []
 
     def on_instance_select(rows: list[dict[str, Any]], evt: gr.SelectData):
         if not rows:
-            return {}, None, [], gr.update(minimum=1, maximum=1, step=1, value=1, interactive=False), []
+            slider_empty = gr.update(minimum=1, maximum=1, step=1, value=1, interactive=False)
+            return {}, None, None, None, None, None, slider_empty, []
         idx = evt.index[0] if isinstance(evt.index, tuple) else evt.index
         if not isinstance(idx, int) or idx < 0 or idx >= len(rows):
             raise IndexError(f"Selected gallery index is out of range: {evt.index!r}")
@@ -287,10 +289,10 @@ def build_app(instance_df: pd.DataFrame, preview_df: pd.DataFrame) -> gr.Blocks:
             value=1,
             interactive=True,
         )
-        return row, download_glb(row["glb_path"]), render_frame_gallery(frame_items, 1), slider_ready, frame_items
+        return row, download_glb(row["glb_path"]), *render_frame_images(frame_items, 1), slider_ready, frame_items
 
     def on_frame_change(frame_idx: float, frame_items: list[dict[str, Any]]):
-        return render_frame_gallery(frame_items, frame_idx)
+        return render_frame_images(frame_items, frame_idx)
 
     initial_rows = instance_df.head(MAX_RESULTS).to_dict(orient="records")
     initial_gallery = build_instance_gallery_items(initial_rows, preview_df)
@@ -311,7 +313,6 @@ def build_app(instance_df: pd.DataFrame, preview_df: pd.DataFrame) -> gr.Blocks:
             is_generated = gr.Dropdown(label="isGenerated", choices=BOOL_FILTER_CHOICES, value=FILTER_ALL)
             transparent = gr.Dropdown(label="transparent", choices=BOOL_FILTER_CHOICES, value=FILTER_ALL)
             near_light = gr.Dropdown(label="near_light", choices=BOOL_FILTER_CHOICES, value=FILTER_ALL)
-            filter_btn = gr.Button("Apply Filters", variant="primary")
 
         summary = gr.Markdown(initial_summary)
         instance_gallery = gr.Gallery(
@@ -330,7 +331,11 @@ def build_app(instance_df: pd.DataFrame, preview_df: pd.DataFrame) -> gr.Blocks:
                 camera_position=(35, 70, 3.5),
             )
 
-        frame_gallery = gr.Gallery(label="RGB / Mask / Depth / Normal", columns=4, rows=1, object_fit="contain", height="auto")
+        with gr.Row():
+            rgb_image = gr.Image(label="RGB", height=360, interactive=False, scale=1, min_width=160)
+            mask_image = gr.Image(label="Mask", height=360, interactive=False, scale=1, min_width=160)
+            depth_image = gr.Image(label="Depth", height=360, interactive=False, scale=1, min_width=160)
+            normal_image = gr.Image(label="Normal", height=360, interactive=False, scale=1, min_width=160)
         frame_slider = gr.Slider(
             label="Frame",
             minimum=1,
@@ -343,37 +348,52 @@ def build_app(instance_df: pd.DataFrame, preview_df: pd.DataFrame) -> gr.Blocks:
         instance_state = gr.State(initial_rows)
         frame_state = gr.State([])
 
-        filter_btn.click(
-            fn=filter_gallery,
-            inputs=[
-                model_name,
-                material_name,
-                env_name,
-                has_glass,
-                is_generated,
-                transparent,
-                near_light,
-            ],
+        filter_inputs = [
+            model_name,
+            material_name,
+            env_name,
+            has_glass,
+            is_generated,
+            transparent,
+            near_light,
+        ]
+        filter_outputs = [
+            summary,
+            instance_gallery,
+            instance_state,
+            instance_meta,
+            model_viewer,
+            rgb_image,
+            mask_image,
+            depth_image,
+            normal_image,
+            frame_slider,
+            frame_state,
+        ]
+        for filter_component in filter_inputs:
+            filter_component.change(
+                fn=filter_gallery,
+                inputs=filter_inputs,
+                outputs=filter_outputs,
+            )
+        instance_gallery.select(
+            fn=on_instance_select,
+            inputs=[instance_state],
             outputs=[
-                summary,
-                instance_gallery,
-                instance_state,
                 instance_meta,
                 model_viewer,
-                frame_gallery,
+                rgb_image,
+                mask_image,
+                depth_image,
+                normal_image,
                 frame_slider,
                 frame_state,
             ],
         )
-        instance_gallery.select(
-            fn=on_instance_select,
-            inputs=[instance_state],
-            outputs=[instance_meta, model_viewer, frame_gallery, frame_slider, frame_state],
-        )
         frame_slider.change(
             fn=on_frame_change,
             inputs=[frame_slider, frame_state],
-            outputs=[frame_gallery],
+            outputs=[rgb_image, mask_image, depth_image, normal_image],
         )
 
     return demo