Update app_recon.py

app_recon.py

@@ -181,7 +181,7 @@ def reconstruction(image, files):
     _, surface, _,  _, _ = bilateral_normal_integration_function(normal_np, mask_output_temp, k=2, K=None, max_iter=100, tol=1e-4, cg_max_iter=5000, cg_tol=1e-3)
     ply_path = os.path.join(dir_name, f"{name_base}_recon.ply")
     surface.save(ply_path, binary=False)
-    return [ply_path]
+    return [ply_path], masked_image
 
 def run_demo():
 
@@ -278,14 +278,17 @@ def run_demo():
             recon_btn = gr.Button('Is there a salient foreground object? If yes, Click here to Reconstruct its 3D model.', variant='primary', interactive=True)
 
         with gr.Row():
-            # reconstructed_3d = gr.Model3D(
-            #     label = 'Bini post-processed 3D model', height=320, interactive=False,
+            gr.Column():
+                masked_image = gr.Image(interactive=False, label="Masked foreground.")
+            gr.Column():
+                reconstructed_3d = gr.Model3D(
+                    label = 'Bini post-processed 3D model', height=320, interactive=False,
+                )
+            # reconstructed_3d = gr.Files(
+            #     label = "Bini post-processed 3D model (plyfile)",
+            #     elem_id = "download",
+            #     interactive=False,
             # )
-            reconstructed_3d = gr.Files(
-                label = "Bini post-processed 3D model (plyfile)",
-                elem_id = "download",
-                interactive=False,
-            )
 
 
         run_btn.click(fn=depth_normal, 
@@ -298,7 +301,7 @@ def run_demo():
                         )
         recon_btn.click(fn=reconstruction,
                         inputs=[input_image, files],
-                        outputs=[reconstructed_3d]
+                        outputs=[reconstructed_3d, masked_image]
                         )
         demo.queue().launch(share=True, max_threads=80)