Update app.py

app.py

@@ -7,9 +7,9 @@ from PIL import Image as PILImage
 import io
 import os
 import base64
-
 import torch
 
+
 class SafetyMonitor:
     def __init__(self):
         """Initialize Safety Monitor with configuration."""
@@ -44,6 +44,14 @@ class SafetyMonitor:
             return cv2.resize(image, (new_width, new_height), interpolation=cv2.INTER_AREA)
         return image
 
+    def encode_image(self, frame):
+        """Convert image to base64 encoding."""
+        frame_pil = PILImage.fromarray(frame)
+        buffered = io.BytesIO()
+        frame_pil.save(buffered, format="JPEG", quality=95)
+        img_base64 = base64.b64encode(buffered.getvalue()).decode('utf-8')
+        return f"data:image/jpeg;base64,{img_base64}"
+
     def detect_objects(self, frame):
         """Detect objects using YOLOv5."""
         results = self.yolo_model(frame)