Update services/video_service.py

services/video_service.py

@@ -21,24 +21,33 @@ class VideoService:
         return Image.fromarray(frame_rgb)
 
     def draw_detections(self, frame, detections):
-        """Draw bounding boxes and labels on frame."""
-        frame_copy = frame.copy()
-        for detection in detections:
-            score = detection["score"]
-            label = detection["label"]
-            box = detection["box"]
-            x1, y1, x2, y2 = box["xmin"], box["ymin"], box["xmax"], box["ymax"]
-            cv2.rectangle(frame_copy, (x1, y1), (x2, y2), (0, 255, 0), 2)
-            cv2.putText(
-                frame_copy,
-                f"{label}: {score:.2f}",
-                (x1, y1 - 10),
-                cv2.FONT_HERSHEY_SIMPLEX,
-                0.5,
-                (0, 255, 0),
-                2
-            )
-        return frame_copy
+    frame_copy = frame.copy()
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    cv2.putText(
+        frame_copy,
+        timestamp,
+        (10, 30),
+        cv2.FONT_HERSHEY_SIMPLEX,
+        1,
+        (255, 255, 255),
+        2
+    )
+    for detection in detections:
+        score = detection["score"]
+        label = detection["label"]
+        box = detection["box"]
+        x1, y1, x2, y2 = box["xmin"], box["ymin"], box["xmax"], box["ymax"]
+        cv2.rectangle(frame_copy, (x1, y1), (x2, y2), (0, 255, 0), 2)
+        cv2.putText(
+            frame_copy,
+            f"{label}: {score:.2f}",
+            (x1, y1 - 10),
+            cv2.FONT_HERSHEY_SIMPLEX,
+            0.5,
+            (0, 255, 0),
+            2
+        )
+    return frame_copy
 
     def save_video(self, frames, output_path, fps=30):
         """Save frames as a video."""