fix: Update geometry_export

app.py

@@ -40,18 +40,75 @@ class Examples(GradioExamples):
             self.cached_file = Path(self.cached_folder) / "log.csv"
             self.create()
 
-def export_geometry(geometry):
-    output_path = tempfile.mktemp(suffix='.obj')
-
+def export_geometry(geometry, file_format='obj'):
+    """
+    Export Open3D geometry (triangle mesh or point cloud) to a file.
+    
+    Args:
+        geometry: Open3D geometry object (TriangleMesh or PointCloud)
+        file_format: str, output format ('obj', 'ply', 'pcd')
+        
+    Returns:
+        str: Path to the exported file
+        
+    Raises:
+        ValueError: If geometry type is not supported or file format is invalid
+    """
+    # Validate geometry type
+    if not isinstance(geometry, (o3d.geometry.TriangleMesh, o3d.geometry.PointCloud)):
+        raise ValueError("Geometry must be either TriangleMesh or PointCloud")
+    
+    # Validate and set file format
+    supported_formats = {
+        'obj': '.obj',
+        'ply': '.ply',
+        'pcd': '.pcd'
+    }
+    
+    if file_format.lower() not in supported_formats:
+        raise ValueError(f"Unsupported file format. Supported formats: {list(supported_formats.keys())}")
+    
+    # Create temporary file with appropriate extension
+    output_path = tempfile.mktemp(suffix=supported_formats[file_format.lower()])
+    
+    # Create a copy of the geometry to avoid modifying the original
+    geometry_copy = geometry.clone()
+    
     # Apply rotation
     rot = np.eye(4)
     rot[:3, :3] = Rotation.from_euler('y', np.deg2rad(180)).as_matrix()
     transform = np.linalg.inv(OPENGL @ rot)
-    geometry.transform(transform)
-
-    o3d.io.write_triangle_mesh(output_path, geometry, write_ascii=False, compressed=True)
     
-    return output_path
+    # Transform geometry
+    geometry_copy.transform(transform)
+    
+    # Export based on geometry type and format
+    try:
+        if isinstance(geometry_copy, o3d.geometry.TriangleMesh):
+            if file_format.lower() == 'obj':
+                o3d.io.write_triangle_mesh(output_path, geometry_copy, 
+                                         write_ascii=False, compressed=True)
+            elif file_format.lower() == 'ply':
+                o3d.io.write_triangle_mesh(output_path, geometry_copy,
+                                         write_ascii=False, compressed=True)
+        
+        elif isinstance(geometry_copy, o3d.geometry.PointCloud):
+            if file_format.lower() == 'pcd':
+                o3d.io.write_point_cloud(output_path, geometry_copy,
+                                       write_ascii=False, compressed=True)
+            elif file_format.lower() == 'ply':
+                o3d.io.write_point_cloud(output_path, geometry_copy,
+                                       write_ascii=False, compressed=True)
+            else:
+                raise ValueError(f"Format {file_format} not supported for point clouds. Use 'ply' or 'pcd'")
+        
+        return output_path
+    
+    except Exception as e:
+        # Clean up temporary file if export fails
+        if os.path.exists(output_path):
+            os.remove(output_path)
+        raise RuntimeError(f"Failed to export geometry: {str(e)}")
 
 
 def extract_frames(video_path: str, duration: float = 20.0, fps: float = 3.0) -> str:
@@ -316,7 +373,7 @@ def reconstruct(video_path, conf_thresh, kf_every,
         # Create 3D model result using gaussian splatting
         return coarse_output_path, gs_output_path
     else:
-        pcd_output_path = export_geometry(pcd_combined)
+        pcd_output_path = export_geometry(pcd_combined, file_format='ply')
         return coarse_output_path, pcd_output_path
 
     # Clean up temporary directory