Update code for .ply file downloading

app.py

@@ -116,11 +116,13 @@ def predictions_to_glb(
 
     # Initialize a 3D scene
     scene_3d = trimesh.Scene()
+    scene_3d_no_cam = trimesh.Scene()
 
     # Add point cloud data to the scene
     point_cloud_data = trimesh.PointCloud(vertices=vertices_3d, colors=colors_rgb)
 
     scene_3d.add_geometry(point_cloud_data)
+    scene_3d_no_cam.add_geometry(point_cloud_data)
 
     # Prepare 4x4 matrices for camera extrinsics
     num_cameras = len(camera_poses)
@@ -140,9 +142,10 @@ def predictions_to_glb(
     align_rotation[:3, :3] = Rotation.from_euler("y", 100, degrees=True).as_matrix()            # plane rotate
     align_rotation[:3, :3] = align_rotation[:3, :3] @ Rotation.from_euler("x", 155, degrees=True).as_matrix()           # roll
     scene_3d.apply_transform(align_rotation)
+    scene_3d_no_cam.apply_transform(align_rotation)
 
     print("GLB Scene built")
-    return scene_3d
+    return scene_3d, scene_3d_no_cam
 
 def integrate_camera_into_scene(scene: trimesh.Scene, transform: np.ndarray, face_colors: tuple, scene_scale: float):
     """
@@ -454,7 +457,7 @@ def gradio_demo(
     )
 
     # Convert predictions to GLB
-    glbscene = predictions_to_glb(
+    glbscene, glbscene_no_cam = predictions_to_glb(
         predictions,
         conf_thres=conf_thres,
         filter_by_frames=frame_filter,
@@ -464,7 +467,7 @@ def gradio_demo(
 
     # we save a .ply file too
     plyfile = glbfile.replace('.glb', '.ply')
-    glbscene.export(file_obj=plyfile)
+    glbscene_no_cam.export(file_obj=plyfile)
 
     # Cleanup
     del predictions
@@ -530,7 +533,7 @@ def update_visualization(
     )
 
     if not os.path.exists(glbfile):
-        glbscene = predictions_to_glb(
+        glbscene, glbscene_no_cam = predictions_to_glb(
             predictions,
             conf_thres=conf_thres,
             filter_by_frames=frame_filter,
@@ -538,7 +541,7 @@ def update_visualization(
         )
         glbscene.export(file_obj=glbfile)
 
-        glbscene.export(file_obj=glbfile.replace('.glb', '.ply'))
+        glbscene_no_cam.export(file_obj=glbfile.replace('.glb', '.ply'))
 
     return glbfile, "Updating Visualization"