Update handler.py

updated handler to deliver pixel

handler.py

@@ -33,11 +33,13 @@ class EndpointHandler:
                 - 'file': Base64 encoded image
                 - 'image': PIL Image object
                 - 'visualization': Boolean flag to return visualization-friendly format (default: False)
+                - 'x': Int pixel position on axis x
+                - 'y': Int pixel position on axis y
         Returns:
             Dictionary containing the depth map and metadata
         """
         # Process input data
-        print(data)
+        
         if "url" in data:
             # Download image from URL
             response = requests.get(data["url"], stream=True)
@@ -84,6 +86,13 @@ class EndpointHandler:
         # Check if visualization is requested
         visualization = data.get("visualization", False)
 
+        # Check the pixels to return  if no pixel provided will return the [0,0] position
+        x= data.get('x',0)
+        y= data.get('y',0)
+
+        map = np.array(depth_map)
+        print(map.shape)
+
         if visualization:
             # Convert depth map to a visualization-friendly format (grayscale image)
             # Create a figure and plot the depth map
@@ -107,15 +116,14 @@ class EndpointHandler:
                 "format": "base64_png"
             }
         else:
-
-
             result = {
+                "deph": depth_map[x][y]
                 # "depth": normalized_depth.tolist(),
-                "depth": compressed_depth_base64,
-                "depth_map": depth_map,
-                "min_depth": float(depth_min),
-                "max_depth": float(depth_max),
-                "shape": list(normalized_depth.shape)
+                # "depth": compressed_depth_base64,
+                # "depth_map": depth_map,
+                # "min_depth": float(depth_min),
+                # "max_depth": float(depth_max),
+                # "shape": list(normalized_depth.shape)
             }
 
         return result