Add a Download button to help users to understand they can retrieve the 3D model

gradio_app/gradio_3dgen.py

@@ -39,7 +39,7 @@ def generate3dv2(preview_img, input_processing, seed, render_video=True, do_refi
     new_meshes = Meshes(verts=[vertices], faces=new_meshes.faces_list(), textures=new_meshes.textures)
     
     ret_mesh, video = save_glb_and_video("/tmp/gradio/generated", new_meshes, with_timestamp=True, dist=3.5, fov_in_degrees=2 / 1.35, cam_type="ortho", export_video=render_video)
-    return ret_mesh, video
+    return ret_mesh, video, gr.update(value=ret_mesh, visible=True)
 
 #######################################
 def create_ui(concurrency_id="wkl"):
@@ -47,6 +47,18 @@ def create_ui(concurrency_id="wkl"):
         with gr.Column(scale=1):
             input_image = gr.Image(type='pil', image_mode='RGBA', label='Frontview')
             
+            input_processing = gr.Checkbox(
+                value=True,
+                label='Remove Background',
+                visible=True,
+            )
+            do_refine = gr.Checkbox(value=True, label="Refine Multiview Details", visible=False)
+            expansion_weight = gr.Slider(minimum=-1., maximum=1.0, value=0.1, step=0.1, label="Expansion Weight", visible=False)
+            init_type = gr.Dropdown(choices=["std", "thin"], label="Mesh Initialization", value="std", visible=False)
+            setable_seed = gr.Slider(-1, 1000000000, -1, step=1, visible=True, label="Seed")
+            render_video = gr.Checkbox(value=False, visible=False, label="generate video")
+            fullrunv2_btn = gr.Button('Generate 3D', variant = "primary", interactive=True)
+            
             example_folder = os.path.join(os.path.dirname(__file__), "./examples")
             example_fns = sorted([os.path.join(example_folder, example) for example in os.listdir(example_folder)])
             gr.Examples(
@@ -61,24 +73,13 @@ def create_ui(concurrency_id="wkl"):
         with gr.Column(scale=1):
             # export mesh display
             output_mesh = gr.Model3D(value=None, label="Mesh Model", show_label=True, height=320, camera_position=(90, 90, 2))
+            download_button = gr.DownloadButton(label="Download 3D", visible=False)
             output_video = gr.Video(label="Preview", show_label=True, show_share_button=True, height=320, visible=False)
             
-            input_processing = gr.Checkbox(
-                value=True,
-                label='Remove Background',
-                visible=True,
-            )
-            do_refine = gr.Checkbox(value=True, label="Refine Multiview Details", visible=False)
-            expansion_weight = gr.Slider(minimum=-1., maximum=1.0, value=0.1, step=0.1, label="Expansion Weight", visible=False)
-            init_type = gr.Dropdown(choices=["std", "thin"], label="Mesh Initialization", value="std", visible=False)
-            setable_seed = gr.Slider(-1, 1000000000, -1, step=1, visible=True, label="Seed")
-            render_video = gr.Checkbox(value=False, visible=False, label="generate video")
-            fullrunv2_btn = gr.Button('Generate 3D', variant = "primary", interactive=True)
-            
     fullrunv2_btn.click(
         fn = generate3dv2,
         inputs=[input_image, input_processing, setable_seed, render_video, do_refine, expansion_weight, init_type],
-        outputs=[output_mesh, output_video],
+        outputs=[output_mesh, output_video, download_button],
         concurrency_id=concurrency_id,
         api_name="generate3dv2",
     ).success(clean_up, api_name=False)