Update app.py

app.py

@@ -8,6 +8,7 @@ import io
 import base64
 import torch
 
+
 class RobustSafetyMonitor:
     def __init__(self):
         """Initialize the robust safety detection tool with configuration."""
@@ -43,12 +44,12 @@ class RobustSafetyMonitor:
         return image
 
     def encode_image(self, frame):
-        """Convert image to base64 encoding."""
+        """Convert image to base64 encoding without extra formatting."""
         frame_pil = PILImage.fromarray(frame)
         buffered = io.BytesIO()
-        frame_pil.save(buffered, format="JPEG", quality=95)
+        frame_pil.save(buffered, format="JPEG", quality=95)  # Ensure JPEG format
         img_base64 = base64.b64encode(buffered.getvalue()).decode('utf-8')
-        return f"data:image/jpeg;base64,{img_base64}"
+        return img_base64  # Return only the base64 string, no "data:image/jpeg;base64,"
 
     def detect_objects(self, frame):
         """Detect objects using YOLOv5."""
@@ -83,7 +84,7 @@ class RobustSafetyMonitor:
                             {
                                 "type": "image_url",
                                 "image_url": {
-                                    "url": f"data:image/jpeg;base64,{image_base64}"  # Use base64 for image
+                                    "url": image_base64  # Corrected: Send only the base64 string
                                 }
                             }
                         ]