1.1.2

app.py

@@ -53,10 +53,10 @@ class DepthEstimator:
     def estimate_depth(self, image_path):
         try:
             # Load image
-            image = Image.open(image_path)
+
             
             # Resize image for processing
-            resized_image, new_size = self.resize_image(image)
+            resized_image, new_size = self.resize_image(image_path)
 
             rgb_image = load_rgb(resized_image.name)
             f_px = rgb_image[-1]
@@ -231,14 +231,17 @@ async def estimate_depth_endpoint(file: UploadFile = File(...), mask: UploadFile
                 content={"error": "Depth estimation failed"}
             )
         
-        # Resize image to match depth map size
+        # Resize image and mask to match depth map size
         resized_image = cv2.resize(image, new_size)
+        resized_mask = cv2.resize(mask, new_size)
+        
+
         
-        # Find key pixels
-        topmost_pixel = find_topmost_pixel(resized_image)
+        # Find key pixels from the mask
+        topmost_pixel = find_topmost_pixel(resized_mask)
         
         # Calculate distance
-        distance_meters = estimate_real_world_distance(depth_map, topmost_pixel, mask)
+        distance_meters = estimate_real_world_distance(depth_map, topmost_pixel, resized_mask)
         
         # Clean up
         os.unlink(temp_file_path)
@@ -265,6 +268,11 @@ async def estimate_depth_endpoint(file: UploadFile = File(...), mask: UploadFile
                 os.unlink(temp_file_path)
             except:
                 pass
+        if 'temp_file_path_mask' in locals():
+            try:
+                os.unlink(temp_file_path_mask)
+            except:
+                pass
         return JSONResponse(
             status_code=500,
             content={"error": str(e)}