Update app.py

app.py

@@ -147,6 +147,28 @@ def get_camera_matrix(depth_estimator):
     }
 
 
+def encode_base64_image(image_array):
+    """
+    Encodes a NumPy (OpenCV) image array to a base64-encoded PNG DataURL 
+    like "data:image/png;base64,<...>".
+    """
+    import base64
+    import cv2
+    
+    # If your image is BGR, that’s fine. We just need to encode it as PNG bytes.
+    # (Optionally convert to RGB first if you need consistent color channels.)
+    
+    success, encoded_buffer = cv2.imencode(".png", image_array)
+    if not success:
+        raise ValueError("Could not encode image to PNG buffer")
+
+    # Encode the buffer to base64
+    b64_str = base64.b64encode(encoded_buffer).decode("utf-8")
+    
+    # Return a data URL
+    return "data:image/png;base64," + b64_str
+
+
 @spaces.GPU
 def get_detection_data(image_data):
     """