update

app.py

@@ -39,19 +39,35 @@ from common import (
     start_session,
 )
 
+
 app_name = os.getenv("GRADIO_APP")
 if app_name == "imageto3d_sam3d":
-    enable_pre_resize = False
+    _enable_pre_resize_default = False
     sample_step = 25
     bg_rm_model_name = "rembg"  # "rembg", "rmbg14"
 elif app_name == "imageto3d":
-    enable_pre_resize = True
+    _enable_pre_resize_default = True
     sample_step = 12
     bg_rm_model_name = "rembg"  # "rembg", "rmbg14"
 
+current_rmbg_tag = bg_rm_model_name
+def set_current_rmbg_tag(rmbg: str) -> None:
+    global current_rmbg_tag
+    current_rmbg_tag = rmbg
+
+
+def preprocess_example_image(
+    img: str,
+) -> tuple[object, object, gr.Button]:
+    image, image_cache = preprocess_image_fn(
+        img, current_rmbg_tag, _enable_pre_resize_default
+    )
+    return image, image_cache, gr.Button(interactive=True)
+
+
 with gr.Blocks(delete_cache=(43200, 43200), theme=custom_theme) as demo:
     gr.HTML(image_css, visible=False)
-    gr.HTML(lighting_css, visible=False)
+    # gr.HTML(lighting_css, visible=False)
     gr.Markdown(
         """
         ## ***EmbodiedGen***: Image-to-3D Asset
@@ -77,7 +93,7 @@ with gr.Blocks(delete_cache=(43200, 43200), theme=custom_theme) as demo:
         ),
         elem_classes=["header"],
     )
-
+    enable_pre_resize = gr.State(_enable_pre_resize_default)
     with gr.Row():
         with gr.Column(scale=3):
             with gr.Tabs() as input_tabs:
@@ -262,21 +278,21 @@ with gr.Blocks(delete_cache=(43200, 43200), theme=custom_theme) as demo:
                 has quality inspection, open with an editor to view details.
             """
             )
-            enable_pre_resize = gr.State(enable_pre_resize)
             with gr.Row() as single_image_example:
                 examples = gr.Examples(
                     label="Image Gallery",
                     examples=[
                         [image_path]
                         for image_path in sorted(
-                            glob("apps/assets/example_image/*")
+                            glob("assets/example_image/*")
                         )
                     ],
-                    inputs=[image_prompt, rmbg_tag, enable_pre_resize],
-                    fn=preprocess_image_fn,
-                    outputs=[image_prompt, raw_image_cache],
+                    inputs=[image_prompt],
+                    fn=preprocess_example_image,
+                    outputs=[image_prompt, raw_image_cache, generate_btn],
                     run_on_click=True,
                     examples_per_page=10,
+                    cache_examples=False,
                 )
 
             with gr.Row(visible=False) as single_sam_image_example:
@@ -285,7 +301,7 @@ with gr.Blocks(delete_cache=(43200, 43200), theme=custom_theme) as demo:
                     examples=[
                         [image_path]
                         for image_path in sorted(
-                            glob("apps/assets/example_image/*")
+                            glob("assets/example_image/*")
                         )
                     ],
                     inputs=[image_prompt_sam],
@@ -294,7 +310,6 @@ with gr.Blocks(delete_cache=(43200, 43200), theme=custom_theme) as demo:
                     run_on_click=True,
                     examples_per_page=10,
                 )
-                
         with gr.Column(scale=2):
             gr.Markdown("<br>")
             video_output = gr.Video(
@@ -340,10 +355,22 @@ with gr.Blocks(delete_cache=(43200, 43200), theme=custom_theme) as demo:
     )
 
     image_prompt.upload(
-        preprocess_image_fn,
-        inputs=[image_prompt, rmbg_tag, enable_pre_resize],
+        lambda img, rmbg: preprocess_image_fn(img, rmbg, _enable_pre_resize_default),
+        inputs=[image_prompt, rmbg_tag],
         outputs=[image_prompt, raw_image_cache],
+        queue=False,
+    ).success(
+        active_btn_by_content,
+        inputs=image_prompt,
+        outputs=generate_btn,
     )
+
+    rmbg_tag.change(
+        set_current_rmbg_tag,
+        inputs=[rmbg_tag],
+        outputs=[],
+    )
+
     image_prompt.change(
         lambda: tuple(
             [
@@ -382,10 +409,9 @@ with gr.Blocks(delete_cache=(43200, 43200), theme=custom_theme) as demo:
             est_mu_text,
         ],
     )
-    image_prompt.change(
-        active_btn_by_content,
-        inputs=image_prompt,
-        outputs=generate_btn,
+    image_prompt.clear(
+        lambda: gr.Button(interactive=False),
+        outputs=[generate_btn],
     )
 
     image_prompt_sam.upload(
@@ -511,4 +537,5 @@ with gr.Blocks(delete_cache=(43200, 43200), theme=custom_theme) as demo:
 
 
 if __name__ == "__main__":
+    # launch_demo()
     demo.launch()

common.py

@@ -18,6 +18,8 @@ import spaces
 from embodied_gen.utils.monkey_patch.trellis import monkey_path_trellis
 
 monkey_path_trellis()
+from embodied_gen.utils.monkey_patch.gradio import _patch_gradio_schema_bool_bug
+_patch_gradio_schema_bool_bug()
 
 import gc
 import logging
@@ -73,7 +75,6 @@ current_file_path = os.path.abspath(__file__)
 current_dir = os.path.dirname(current_file_path)
 sys.path.append(os.path.join(current_dir, ".."))
 from thirdparty.TRELLIS.trellis.pipelines import TrellisImageTo3DPipeline
-from thirdparty.TRELLIS.trellis.utils import postprocessing_utils
 
 logging.basicConfig(
     format="%(asctime)s - %(levelname)s - %(message)s", level=logging.INFO
@@ -275,7 +276,7 @@ def image_to_3d(
     sam_image: Image.Image = None,
     is_sam_image: bool = False,
     req: gr.Request = None,
-) -> tuple[dict, str]:
+) -> tuple[object, str]:
     if is_sam_image:
         seg_image = filter_image_small_connected_components(sam_image)
         seg_image = Image.fromarray(seg_image, mode="RGBA")
@@ -334,7 +335,7 @@ def image_to_3d(
 
 
 def extract_3d_representations_v2(
-    state: dict,
+    state: object,
     enable_delight: bool,
     texture_size: int,
     req: gr.Request,
@@ -401,7 +402,7 @@ def extract_3d_representations_v2(
 
 
 def extract_3d_representations_v3(
-    state: dict,
+    state: object,
     enable_delight: bool,
     texture_size: int,
     req: gr.Request,

embodied_gen/utils/monkey_patch/gradio.py

@@ -0,0 +1,41 @@
+# Project EmbodiedGen
+#
+# Copyright (c) 2025 Horizon Robotics. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+# implied. See the License for the specific language governing
+# permissions and limitations under the License.
+
+
+import gradio_client.utils as gradio_client_utils
+
+
+def _patch_gradio_schema_bool_bug() -> None:
+    """Patch gradio_client schema parser for bool-style additionalProperties."""
+    original_get_type = gradio_client_utils.get_type
+    original_json_schema_to_python_type = (
+        gradio_client_utils._json_schema_to_python_type
+    )
+
+    def _safe_get_type(schema):
+        if isinstance(schema, bool):
+            return {}
+        return original_get_type(schema)
+
+    def _safe_json_schema_to_python_type(schema, defs):
+        if isinstance(schema, bool):
+            return "Any"
+        return original_json_schema_to_python_type(schema, defs)
+
+    gradio_client_utils.get_type = _safe_get_type
+    gradio_client_utils._json_schema_to_python_type = (
+        _safe_json_schema_to_python_type
+    )