Add download buttons

app.py

@@ -111,12 +111,12 @@ def generate(image, mc_resolution, formats=["obj", "glb"]):
     mesh.apply_scale([-1, 1, 1])  # Otherwise the visualized .obj will be flipped
     mesh.export(mesh_path_obj.name)
     
-    return mesh_path_obj.name, mesh_path_glb.name
+    return mesh_path_obj.name, mesh_path_glb.name, gr.update(value=mesh_path_obj.name, visible=True), gr.update(value=mesh_path_glb.name, visible=True)
 
 def run_example(image_pil):
     preprocessed = preprocess(image_pil, False, 0.9)
-    mesh_name_obj, mesh_name_glb = generate(preprocessed, 256, ["obj", "glb"])
-    return preprocessed, mesh_name_obj, mesh_name_glb
+    mesh_name_obj, mesh_name_glb, download_obj, download_glb = generate(preprocessed, 256, ["obj", "glb"])
+    return preprocessed, mesh_name_obj, mesh_name_glb, download_obj, download_glb
 
 with gr.Blocks() as demo:
     gr.Markdown(HEADER)
@@ -158,12 +158,14 @@ with gr.Blocks() as demo:
                     label="Output Model (OBJ Format)",
                     interactive=False,
                 )
+                download_obj_button = gr.DownloadButton(label="Download .obj", visible=False)
                 gr.Markdown("Note: Downloaded object will be flipped in case of .obj export. Export .glb instead or manually flip it before usage.")
             with gr.Tab("GLB"):
                 output_model_glb = gr.Model3D(
                     label="Output Model (GLB Format)",
                     interactive=False,
                 )
+                download_glb_button = gr.DownloadButton(label="Download .glb", visible=False)
                 gr.Markdown("Note: The model shown here has a darker appearance. Download to get correct results.")
     with gr.Row(variant="panel"):
         gr.Examples(
@@ -171,7 +173,7 @@ with gr.Blocks() as demo:
                 os.path.join("examples", img_name) for img_name in sorted(os.listdir("examples"))
             ],
             inputs=[input_image],
-            outputs=[processed_image, output_model_obj, output_model_glb],
+            outputs=[processed_image, output_model_obj, output_model_glb, download_obj_button, download_glb_button],
             cache_examples=True,
             fn=partial(run_example),
             label="Examples",
@@ -184,7 +186,7 @@ with gr.Blocks() as demo:
     ).success(
         fn=generate,
         inputs=[processed_image, mc_resolution],
-        outputs=[output_model_obj, output_model_glb],
+        outputs=[output_model_obj, output_model_glb, download_obj_button, download_glb_button],
     )
 
 demo.queue(max_size=10)