Update app.py

app.py

@@ -505,7 +505,33 @@ def depth_to_mesh(depth_map, image, resolution=100, detail_level='medium'):
                 faces.append([p1, p2, p3])
                 faces.append([p2, p4, p3])
     
+    # Check if we have valid faces before creating the mesh
+    if len(faces) == 0:
+        # Create a simple square mesh as fallback
+        faces = [[0, 1, 2], [1, 3, 2]]
+    
     faces = np.array(faces)
+    
+    # Ensure we have at least one vertex and face
+    if len(vertices) == 0 or len(faces) == 0:
+        # Create a minimal mesh to avoid errors
+        vertices = np.array([[0, 0, 0], [1, 0, 0], [0, 1, 0], [1, 1, 0]])
+        faces = np.array([[0, 1, 2], [1, 3, 2]])
+    
+    # Check for out-of-bounds indexes
+    max_vertex_idx = len(vertices) - 1
+    valid_faces = []
+    for face in faces:
+        if np.all(face <= max_vertex_idx):
+            valid_faces.append(face)
+    
+    if len(valid_faces) == 0:
+        # Create a minimal mesh to avoid errors
+        vertices = np.array([[0, 0, 0], [1, 0, 0], [0, 1, 0], [1, 1, 0]])
+        faces = np.array([[0, 1, 2], [1, 3, 2]])
+    else:
+        faces = np.array(valid_faces)
+    
     mesh = trimesh.Trimesh(vertices=vertices, faces=faces)
     
     # Apply vertex colors from image
@@ -524,6 +550,10 @@ def depth_to_mesh(depth_map, image, resolution=100, detail_level='medium'):
                 
                 vertex_idx = i * resolution + j
                 
+                # Skip if vertex index is out of range
+                if vertex_idx >= len(vertices):
+                    continue
+                
                 # Handle RGBA images
                 if len(img_array.shape) == 3 and img_array.shape[2] == 4:
                     # Direct copy of RGBA values with bilinear interpolation
@@ -557,13 +587,31 @@ def depth_to_mesh(depth_map, image, resolution=100, detail_level='medium'):
         
         mesh.visual.vertex_colors = vertex_colors
     
-    # Apply smoothing for non-high detail levels
-    if detail_level != 'high':
-        mesh = mesh.smoothed(method='laplacian', iterations=1)
+    try:
+        # Apply smoothing for non-high detail levels
+        if detail_level != 'high':
+            mesh = mesh.smoothed(method='laplacian', iterations=1)
+        
+        # Try to fix normals but catch any errors
+        try:
+            mesh.fix_normals()
+        except Exception as e:
+            print(f"Warning: Could not fix normals: {str(e)}")
+            # Compute face normals manually if fix_normals fails
+            mesh.face_normals = trimesh.geometry.triangles_normals(
+                mesh.triangles
+            )
+    except Exception as e:
+        print(f"Warning: Error in mesh post-processing: {str(e)}")
     
-    # Fix normals for better rendering
-    mesh.fix_normals()
     return mesh
+        
+        
+    
+   
+    
+    
+          
 
 @app.route('/health', methods=['GET'])
 def health_check():