Update app.py

app.py

@@ -99,44 +99,38 @@ class RobustDepthTo3DConverter:
             print(f"Depth estimation error: {e}")
             return None
     
-    def create_point_cloud(self, image, depth_map):
-        """
-        Create point cloud with enhanced error handling
-        """
-        if depth_map is None:
-            return None
-        
-        try:
-            img_array = np.array(image)
-            height, width = img_array.shape[:2]
-            
-            # More adaptive sampling
-            step = max(1, min(height, width) // 300)
-            
-            points = []
-            colors = []
-            
-            for y in range(0, height, step):
-                for x in range(0, width, step):
-                    z = depth_map[y, x] / 255.0 * 10  # Extended depth scaling
-                    points.append([x, y, z])
-                    
-                    # Safer color extraction
-                    try:
-                        color = img_array[y, x][:3] / 255.0
-                    except IndexError:
-                        color = [0.5, 0.5, 0.5]
-                    colors.append(color)
-            
-            pcd = o3d.geometry.PointCloud()
-            pcd.points = o3d.utility.Vector3dVector(points)
-            pcd.colors = o3d.utility.Vector3dVector(colors)
-            
-            return pcd
+def create_point_cloud(self, image, depth_map):
+    """
+    Create point cloud with error handling and dimension checks
+    """
+    if depth_map is None:
+        return None
+
+    try:
+        img_array = np.array(image)
+        depth_map_resized = cv2.resize(depth_map, (img_array.shape[1], img_array.shape[0]), interpolation=cv2.INTER_LINEAR)
+        height, width = img_array.shape[:2]
+
+        step = max(1, min(height, width) // 200)
+        points, colors = [], []
+
+        for y in range(0, height, step):
+            for x in range(0, width, step):
+                z = depth_map_resized[y, x] / 255.0 * 5  # Scaled depth
+                points.append([x, y, z])
+                color = img_array[y, x][:3] / 255.0 if len(img_array[y, x]) >= 3 else [0.5, 0.5, 0.5]
+                colors.append(color)
+
+        pcd = o3d.geometry.PointCloud()
+        pcd.points = o3d.utility.Vector3dVector(points)
+        pcd.colors = o3d.utility.Vector3dVector(colors)
+
+        return pcd
+
+    except Exception as e:
+        print(f"Point cloud creation error: {e}")
+        return None
         
-        except Exception as e:
-            print(f"Point cloud creation error: {e}")
-            return None
     
     def convert_to_mesh(self, point_cloud):
         """