Fix Gradio 4.44.1 output updates for video/model visibility

AGENTS.md

@@ -5,12 +5,12 @@
 - **Goal**: High-fidelity 3D asset generation from 2D images.
 - **Priority**: Robust execution, VRAM efficiency, and seamless Hugging Face Space deployment.
 
-## 2. Tech Stack
-- **Core**: Python, PyTorch, CUDA.
-- **Model Architecture**: Diffusion-based 3D generation (TRELLIS).
-- **UI**: Gradio (`app.py`).
-- **Deployment**: Dockerized on Hugging Face Spaces.
-- **Models**: Loaded via Hugging Face Hub (DinoV2, RMBG2, etc.) or local cache.
+## 2. Tech Stack
+- **Core**: Python, PyTorch, CUDA.
+- **Model Architecture**: Diffusion-based 3D generation (TRELLIS).
+- **UI**: Gradio (`app.py`) pinned to **4.44.1**. Keep event/callback usage compatible with Gradio 4.44.1 APIs.
+- **Deployment**: Dockerized on Hugging Face Spaces.
+- **Models**: Loaded via Hugging Face Hub (DinoV2, RMBG2, etc.) or local cache.
 
 ## 3. Repository Structure
 - `app.py`: Main Gradio web application. Handles UI, video rendering, and high-level pipeline orchestration.

app.py

@@ -147,7 +147,12 @@ def process_image_yielding(
         for name, exr_data in envmap.items():
             loaded_envmap[name] = EnvMap(torch.tensor(exr_data, dtype=torch.float32, device='cuda'))
 
-        yield None, None, gr.update(value=None, interactive=False), None
+        yield (
+            gr.update(value=None, visible=True, label="Preview"),
+            gr.update(value=None, visible=True, label="Generating 3D Model..."),
+            gr.update(value=None, visible=True, interactive=False),
+            None,
+        )
 
         if do_preprocess:
             progress(0, desc="Removing background...")
@@ -217,9 +222,15 @@ def process_image_yielding(
 
             if video_path:
                 logger.info(f"Video rendered: {video_path}")
-                yield video_path, None, gr.update(value=None, interactive=False), None
+                yield (
+                    gr.update(value=video_path, visible=True, label="Double Tap To Play"),
+                    gr.update(value=None, visible=True, label="Generating 3D Model..."),
+                    gr.update(value=None, visible=True, interactive=False),
+                    None,
+                )
         except Exception as e:
             logger.error(f"Video rendering error: {e}", exc_info=True)
+            logger.warning("Preview video generation failed; continuing with GLB/STL outputs.")
 
         progress(0.7, desc="Extracting GLB model...")
         glb_start = time.time()
@@ -276,8 +287,14 @@ def process_image_yielding(
         except Exception as stl_e:
             logger.error(f"STL export error: {stl_e}", exc_info=True)
 
-        stl_update = gr.update(value=stl_path, interactive=True) if stl_path else gr.update(value=None, interactive=False)
-        yield video_path, glb_path, stl_update, None
+        video_update = gr.update(
+            value=video_path,
+            visible=True,
+            label="Double Tap To Play" if video_path else "Preview unavailable",
+        )
+        model_update = gr.update(value=glb_path, visible=True, label="Interactive 3D Model")
+        stl_update = gr.update(value=stl_path, visible=True, interactive=bool(stl_path))
+        yield video_update, model_update, stl_update, None
 
     except gr.Error:
         raise
@@ -334,7 +351,7 @@ with gr.Blocks(theme='Taithrah/Minimal', css=css, title="Pocket 3D AI") as demo:
                     label=" ",
                     height=240,
                     elem_classes="video-container",
-                    visible=False,
+                    visible=True,
                     autoplay=True,
                     loop=True,
                     show_download_button=True,
@@ -360,9 +377,9 @@ with gr.Blocks(theme='Taithrah/Minimal', css=css, title="Pocket 3D AI") as demo:
                     exposure=10.0,
                     height=360,
                     elem_classes="model-container",
-                    visible=False,
+                    visible=True,
                 )
-                stl_download_button = gr.DownloadButton(label="Download STL", visible=False, interactive=False, size="lg", variant="primary")
+                stl_download_button = gr.DownloadButton(label="Download STL", visible=True, interactive=False, size="lg", variant="primary")
 
         with gr.Accordion(label="Generation Settings", open=False, visible=not prod):
             seed_slider = gr.Slider(0, MAX_SEED, label="Seed", value=0, step=1)
@@ -432,26 +449,6 @@ with gr.Blocks(theme='Taithrah/Minimal', css=css, title="Pocket 3D AI") as demo:
         scroll_to_output=True,
     )
 
-    @gr.on(triggers=[image_prompt.change], inputs=None, outputs=[video_output, model_output, stl_download_button], show_progress="minimal")
-    def toggle_outputs_on_new_image():
-        return (
-            gr.update(visible=True, value=None),
-            gr.update(visible=False, value=None),
-            gr.update(visible=False, value=None, interactive=False)
-        )
-
-    @gr.on(triggers=[video_output.change], inputs=None, outputs=[model_output, stl_download_button])
-    def toggle_model_and_stl_visibility():
-        return (
-            gr.update(label="Interactive 3D Model", visible=True),
-            gr.update(visible=True)
-        )
-
-    @gr.on(triggers=[video_output.change], inputs=None, outputs=video_output, show_progress="hidden")
-    def toggle_label():
-        return gr.update(label="Double Tap To Play", visible=True)
-
-
 if __name__ == "__main__":
     if pipeline is None:
         logger.critical("Pipeline failed to initialize. Exiting.")