nedo-vision-worker-core 0.4.3__tar.gz → 0.4.4__tar.gz

{nedo_vision_worker_core-0.4.3 → nedo_vision_worker_core-0.4.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nedo-vision-worker-core
-Version: 0.4.3
+Version: 0.4.4
 Summary: Nedo Vision Worker Core Library for AI Vision Processing
 Author-email: Willy Achmat Fauzi <willy.achmat@gmail.com>
 Maintainer-email: Willy Achmat Fauzi <willy.achmat@gmail.com>

{nedo_vision_worker_core-0.4.3 → nedo_vision_worker_core-0.4.4}/nedo_vision_worker_core/__init__.py

@@ -7,7 +7,7 @@ A library for running AI vision processing and detection in the Nedo Vision plat
 from .core_service import CoreService
 from .callbacks import DetectionType, CallbackTrigger, DetectionData, IntervalMetadata
 
-__version__ = "0.4.3"
+__version__ = "0.4.4"
 __all__ = [
     "CoreService", 
     "DetectionType",

{nedo_vision_worker_core-0.4.3 → nedo_vision_worker_core-0.4.4}/nedo_vision_worker_core/pipeline/PipelineProcessor.py

@@ -202,9 +202,6 @@ class PipelineProcessor:
                 if loop_start - last_preview_check >= self.preview_check_interval:
                     self._check_and_update_rtmp_streaming()
                     last_preview_check = loop_start
-
-                # Only draw frames when actually needed (RTMP active or debug)
-                should_draw = self.rtmp_streaming_active or self.debug_flag
                 
                 # For RTMP, also check if it's time to send a frame (rate limiting)
                 should_draw_for_rtmp = (

{nedo_vision_worker_core-0.4.3 → nedo_vision_worker_core-0.4.4}/nedo_vision_worker_core/pipeline/PipelineSyncThread.py

@@ -151,14 +151,15 @@ class PipelineSyncThread(threading.Thread):
         ])
 
         if requires_restart:
-            logging.info(f"🔄 Restarting pipeline due to significant changes: {pid}")
-            self.pipeline_manager.stop_pipeline(pid)
-            
             # Check if source is connected before restarting
             if not self.source_repo.is_source_connected(db_pipeline.worker_source_id):
                 logging.warning(f"⚠️ Cannot restart pipeline {pid}: Source {db_pipeline.worker_source_id} is disconnected")
                 return
             
+            logging.info(f"🔄 Restarting pipeline due to significant changes: {pid}")
+            self.pipeline_manager.stop_pipeline(pid)
+            
+            
             self.pipeline_manager.start_pipeline(db_pipeline, db_detector)
         else:
             # Update config for minor changes that don't require restart

{nedo_vision_worker_core-0.4.3 → nedo_vision_worker_core-0.4.4}/nedo_vision_worker_core/repositories/WorkerSourcePipelineDetectionRepository.py

@@ -18,12 +18,14 @@ class WorkerSourcePipelineDetectionRepository(BaseRepository):
     def save_detection(self, pipeline_id: int, frame, tracked_objects, frame_drawer: FrameDrawer):
         """
         Save detection data that need to be sent to database.
+        Only saves if there are violations detected.
         """
         now = datetime.now(timezone.utc)
         current_datetime = now.strftime("%Y%m%d_%H%M%S")
 
         frame_drawer.draw_polygons(frame)
         filtered_objects = []
+        has_violations = False
 
         for tracked_obj in tracked_objects:
             attributes = tracked_obj["attributes"]
@@ -34,9 +36,17 @@ class WorkerSourcePipelineDetectionRepository(BaseRepository):
             obj = tracked_obj.copy()
             obj["attributes"] = [attr for attr in attributes if attr.get("count", 0) >= 5]
 
+            # Check if any attribute is a violation
+            for attr in obj["attributes"]:
+                attr_label = attr.get("label", "")
+                if attr_label in frame_drawer.violation_labels:
+                    has_violations = True
+                    break
+
             filtered_objects.append(obj)
 
-        if not filtered_objects:
+        # Only save and trigger webhook/MQTT if there are actual violations
+        if not filtered_objects or not has_violations:
             return
 
         drawn_frame = frame_drawer.draw_frame(frame.copy(), filtered_objects)
@@ -64,7 +74,5 @@ class WorkerSourcePipelineDetectionRepository(BaseRepository):
                 )
                 session.add(new_detection)
                 session.flush()
-                # Commit happens automatically via context manager
-                print(f"✅ Inserted detection data for pipeline {pipeline_id}")
         except Exception as e:
             print(f"❌ Database error while saving detection: {e}")

{nedo_vision_worker_core-0.4.3 → nedo_vision_worker_core-0.4.4}/nedo_vision_worker_core/streams/VideoStream.py

@@ -237,6 +237,8 @@ class VideoStream(threading.Thread):
     def run(self) -> None:
         """Main capture loop."""
         failures = 0
+        last_frame_time = time.perf_counter()
+        
         while self.running:
             try:
                 if not self.capture or not self.capture.isOpened():
@@ -247,6 +249,18 @@ class VideoStream(threading.Thread):
                     failures = 0
                     self._reconnect_attempts = 0
                     self._current_interval = self.reconnect_interval
+                    last_frame_time = time.perf_counter()
+
+                # For video files, respect the FPS to avoid playing too fast
+                if self.is_file:
+                    target_fps = self.target_fps if self.target_fps else self.fps
+                    if target_fps > 0:
+                        frame_interval = 1.0 / target_fps
+                        elapsed = time.perf_counter() - last_frame_time
+                        if elapsed < frame_interval:
+                            sleep_time = frame_interval - elapsed
+                            time.sleep(sleep_time)
+                    last_frame_time = time.perf_counter()
 
                 ret, frame = self.capture.read()
 

{nedo_vision_worker_core-0.4.3 → nedo_vision_worker_core-0.4.4}/nedo_vision_worker_core.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nedo-vision-worker-core
-Version: 0.4.3
+Version: 0.4.4
 Summary: Nedo Vision Worker Core Library for AI Vision Processing
 Author-email: Willy Achmat Fauzi <willy.achmat@gmail.com>
 Maintainer-email: Willy Achmat Fauzi <willy.achmat@gmail.com>