Update code for .ply file downloading

app.py

@@ -1,8 +1,3 @@
-# Copyright (c) Meta Platforms, Inc. and affiliates.
-# All rights reserved.
-#
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
 
 import os
 import cv2
@@ -467,6 +462,10 @@ def gradio_demo(
     )
     glbscene.export(file_obj=glbfile)
 
+    # we save a .ply file too
+    plyfile = glbfile.replace('.glb', '.ply')
+    glbscene.export(file_obj=plyfile)
+
     # Cleanup
     del predictions
     gc.collect()
@@ -476,7 +475,7 @@ def gradio_demo(
     print(f"Total time: {end_time - start_time:.2f} seconds (including IO)")
     log_msg = f"Reconstruction Success ({len(all_files)} frames). Waiting for visualization."
 
-    return glbfile, log_msg, gr.Dropdown(choices=frame_filter_choices, value=frame_filter, interactive=True)
+    return glbfile, plyfile, log_msg, gr.Dropdown(choices=frame_filter_choices, value=frame_filter, interactive=True)
 
 
 # -------------------------------------------------------------------------
@@ -539,6 +538,8 @@ def update_visualization(
         )
         glbscene.export(file_obj=glbfile)
 
+        glbscene.export(file_obj=glbfile.replace('.glb', '.ply'))
+
     return glbfile, "Updating Visualization"
 
 
@@ -791,6 +792,9 @@ if __name__ == '__main__':
                 gr.Markdown("### 2. View Reconstruction")
                 log_output = gr.Markdown("Please upload media and click Reconstruct.", elem_classes=["custom-log"])
                 reconstruction_output = gr.Model3D(height=480, zoom_speed=0.5, pan_speed=0.5, label="3D Output")
+
+                # optional to download .ply file
+                download_ply_output = gr.File(label="Download .ply Result", interactive=False)
                 
                 with gr.Row():
                     submit_btn = gr.Button("Reconstruct", scale=3, variant="primary")
@@ -836,10 +840,10 @@ if __name__ == '__main__':
             target_dir, image_paths = handle_uploads(input_video, input_images, interval)
             # Always use "All" for frame_filter in examples
             frame_filter = "All"
-            glbfile, log_msg, dropdown = gradio_demo(
+            glbfile, ply_file, log_msg, dropdown = gradio_demo(
                 target_dir, conf_thres, frame_filter, show_cam
             )
-            return glbfile, log_msg, target_dir, dropdown, image_paths
+            return glbfile, ply_file, log_msg, target_dir, dropdown, image_paths
 
         gr.Markdown("Click any row to load an example.", elem_classes=["example-log"])
 
@@ -852,7 +856,7 @@ if __name__ == '__main__':
                 conf_thres,
                 show_cam,
             ],
-            outputs=[reconstruction_output, log_output, target_dir_output, frame_filter, image_gallery],
+            outputs=[reconstruction_output, download_ply_output, log_output, target_dir_output, frame_filter, image_gallery],
             fn=example_pipeline,
             cache_examples=False,
             examples_per_page=50,
@@ -876,7 +880,7 @@ if __name__ == '__main__':
                 frame_filter,
                 show_cam,
             ],
-            outputs=[reconstruction_output, log_output, frame_filter],
+            outputs=[reconstruction_output, download_ply_output, log_output, frame_filter],
         ).then(
             fn=lambda: "False", inputs=[], outputs=[is_example]  # set is_example to "False"
         )