Update code for .ply file downloading

app.py

@@ -14,7 +14,7 @@ import spaces
 
 from pi3.utils.geometry import se3_inverse, homogenize_points, depth_edge
 from pi3.models.pi3 import Pi3
-from pi3.utils.basic import load_images_as_tensor
+from pi3.utils.basic import load_images_as_tensor, write_ply
 
 import trimesh
 import matplotlib
@@ -143,7 +143,7 @@ def predictions_to_glb(
     scene_3d.apply_transform(align_rotation)
 
     print("GLB Scene built")
-    return scene_3d, point_cloud_data
+    return scene_3d, [vertices_3d, colors_rgb]
 
 def integrate_camera_into_scene(scene: trimesh.Scene, transform: np.ndarray, face_colors: tuple, scene_scale: float):
     """
@@ -455,7 +455,7 @@ def gradio_demo(
     )
 
     # Convert predictions to GLB
-    glbscene, glbscene_no_cam = predictions_to_glb(
+    glbscene, pcd = predictions_to_glb(
         predictions,
         conf_thres=conf_thres,
         filter_by_frames=frame_filter,
@@ -465,7 +465,7 @@ def gradio_demo(
 
     # we save a .ply file too
     plyfile = glbfile.replace('.glb', '.ply')
-    glbscene_no_cam.export(file_obj=plyfile)
+    write_ply(pcd[0], pcd[1]/255, path=plyfile)
 
     # Cleanup
     del predictions
@@ -795,7 +795,7 @@ if __name__ == '__main__':
                 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)
+                download_ply_output = gr.File(height=100, label="Download .ply Result", interactive=False)
                 
                 with gr.Row():
                     submit_btn = gr.Button("Reconstruct", scale=3, variant="primary")