Update app.py

app.py

@@ -167,18 +167,14 @@ def encode_base64_image(image_array):
     # Return a data URL
     return "data:image/png;base64," + b64_str
 
-def generate_image_url(image):
-    """Generate a shareable URL for an OpenCV image."""
-    success, encoded_buffer = cv2.imencode(".png", image)
-    if not success:
-        raise ValueError("Could not encode image to PNG buffer")
-    
-    b64_str = base64.b64encode(encoded_buffer).decode("utf-8")
-    return "data:image/png;base64," + b64_str
-
-
-def generate_plot_url(objs_pcd):
-    """Generate a shareable URL for a Plotly 3D scatter plot of point clouds."""
+def save_image_to_url(image):
+    """Save an OpenCV image to a temporary file and return its URL."""
+    with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as temp_file:
+        cv2.imwrite(temp_file.name, image)
+        return "/".join(temp_file.name.split("/")[-2:])  # Return relative path for URL
+
+def save_plot_to_url(objs_pcd):
+    """Save a Plotly 3D scatter plot to a temporary file and return its URL."""
     fig = go.Figure()
 
     for data, clr in objs_pcd:
@@ -205,9 +201,11 @@ def generate_plot_url(objs_pcd):
         )
     )
     
-    plot_html = fig.to_html(include_plotlyjs='cdn', full_html=False)
-    b64_str = base64.b64encode(plot_html.encode("utf-8")).decode("utf-8")
-    return "data:text/html;base64," + b64_str
+    with tempfile.NamedTemporaryFile(suffix=".html", delete=False) as temp_file:
+        fig.write_html(temp_file.name)
+        return "/".join(temp_file.name.split("/")[-2:])  # Return relative path for URL
+
+ 
 
 def get_3d_position(center, depth, camera_matrix):
     """Project 2D center into 3D space using depth and camera matrix."""
@@ -272,10 +270,10 @@ def get_detection_data(image_data):
 
         response = {
             "detections": detections,
-            "segmentation_url": generate_image_url(image_segmentation),
-            "depth_url": generate_image_url(depth_colormap),
-            "distance_url": generate_image_url(utils.draw_depth_info(image, depthmap, objects_data)),
-            "point_cloud_url": generate_plot_url(utils.generate_obj_pcd(depthmap, objects_data)),
+            "segmentation_url": save_image_to_url(image_segmentation),
+            "depth_url": save_image_to_url(depth_colormap),
+            "distance_url": save_image_to_url(utils.draw_depth_info(image, depthmap, objects_data)),
+            "point_cloud_url": save_plot_to_url(utils.generate_obj_pcd(depthmap, objects_data)),
             "camera_matrix": get_camera_matrix(depth_estimator),
             "camera_position": [0, 0, 0]  # Assumed at origin based on camera intrinsics
         }