nedo-vision-worker-core 0.4.3__py3-none-any.whl → 0.4.4__py3-none-any.whl

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/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/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/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/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.dist-info → nedo_vision_worker_core-0.4.4.dist-info}/METADATA

@@ -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.dist-info → nedo_vision_worker_core-0.4.4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-nedo_vision_worker_core/__init__.py,sha256=65HOyr4VnCAkUlmbf0yGE_zIL4hLNL1IOY8zh_rnl8E,1923
+nedo_vision_worker_core/__init__.py,sha256=ZlpHZZ54eVsepVdpBxhuJHhyEpKF8OqZAP9UvfFjMa4,1923
 nedo_vision_worker_core/cli.py,sha256=8YuKWsIgICUYXE_QtwyU3WzGhVjTWiAo5uzpFOmjNc8,5766
 nedo_vision_worker_core/core_service.py,sha256=q8-GuGW_l5l6wTWQDqc7BDdhM7zKC-mMLZ5wIHu9xV0,11628
 nedo_vision_worker_core/doctor.py,sha256=K_-hVV2-mdEefZ4Cfu5hMCiOxBiI1aXY8VtkkpK80Lc,10651
@@ -61,8 +61,8 @@ nedo_vision_worker_core/pipeline/ModelManager.py,sha256=2DoQiIdF-PAqU7nT_u6bj-DY
 nedo_vision_worker_core/pipeline/PipelineConfigManager.py,sha256=X55i9GyXcW9ylO6cj2UMAZFSxxPViacL4H4DZl60CAY,1157
 nedo_vision_worker_core/pipeline/PipelineManager.py,sha256=AlDwBYYRPeAeh2ilmC8n-A_2gYPqAzeSSVpR1Tc0ipE,8366
 nedo_vision_worker_core/pipeline/PipelinePrepocessor.py,sha256=cCiVSHHqsKCtKYURdYoEjHJX2GnT6zd8kQ6ZukjQ3V0,1271
-nedo_vision_worker_core/pipeline/PipelineProcessor.py,sha256=XITn3d8la37sCYEI_ZQzBoYYK6nrY9hAg6CaLKbxBF0,35305
-nedo_vision_worker_core/pipeline/PipelineSyncThread.py,sha256=UmjME7b1UYABVXKzYvlYBhVIyDcrsL5JDMxNt9G-0yk,9504
+nedo_vision_worker_core/pipeline/PipelineProcessor.py,sha256=5j0FvGwBw5hgJ-FzI493cLpZomfYiG-bmUZZyklmahI,35149
+nedo_vision_worker_core/pipeline/PipelineSyncThread.py,sha256=8HOVdeWO0dAXY9rMAk4VlX-bIhWrkhSzyJwlJobUnvw,9517
 nedo_vision_worker_core/pipeline/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 nedo_vision_worker_core/preprocessing/ImageResizer.py,sha256=RvOazxe6dJQuiy0ZH4lIGbdFfiu0FLUVCHoMvxkDNT4,1324
 nedo_vision_worker_core/preprocessing/ImageRoi.py,sha256=iO7oQ-SdUSA_kTIVBuq_mdycXsiJNfiFD3J7-VTxiQ4,2141
@@ -73,7 +73,7 @@ nedo_vision_worker_core/repositories/BaseRepository.py,sha256=FvdcD8I8_4_6TMZa8X
 nedo_vision_worker_core/repositories/PPEDetectionRepository.py,sha256=dIOPUU7xOCJ8A5v6AnBVKwtKPVlc5M5cUhLCWumTXyo,6882
 nedo_vision_worker_core/repositories/RestrictedAreaRepository.py,sha256=a5Vc8WLTtAa6Tn-ZSKkBEw4-cM29VW8WUwhbH7YfM9E,4416
 nedo_vision_worker_core/repositories/WorkerSourcePipelineDebugRepository.py,sha256=lN_yip6woya9YUA5sYKbTyDQz2qSfgqkr3YP2hSd9ws,3211
-nedo_vision_worker_core/repositories/WorkerSourcePipelineDetectionRepository.py,sha256=5m4lvmIETJSGDH1T1EHuUDWC-13t5I860UbN_uzEj9A,2641
+nedo_vision_worker_core/repositories/WorkerSourcePipelineDetectionRepository.py,sha256=AVmrpeu6YziX2255_ZhK-5X5snxVMlr8xzb4ymAe5SA,2954
 nedo_vision_worker_core/repositories/WorkerSourcePipelineRepository.py,sha256=vwwRA1INuK66siOHNZxSBX8CE9uEW8VVcCIA7dmshKo,4714
 nedo_vision_worker_core/repositories/WorkerSourceRepository.py,sha256=YFevfYhAsYd7Eho1iagzjk67tKAQfqmoIExyxvR2Bzc,1760
 nedo_vision_worker_core/repositories/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
@@ -83,7 +83,7 @@ nedo_vision_worker_core/services/VideoSharingDaemonManager.py,sha256=sc8VZo5iwoO
 nedo_vision_worker_core/streams/RTMPStreamer.py,sha256=7vg2as_TtmZavLnzhEUOv6HhjoGH5X3JO9HUI74meNs,18789
 nedo_vision_worker_core/streams/SharedVideoDeviceManager.py,sha256=vSslwxbhKH6FPndR1HcSFIVWtF-iiOQMlSa4VvFa6M4,16265
 nedo_vision_worker_core/streams/StreamSyncThread.py,sha256=ETT0N_P90ksn6Q5pb7NvMadqCuoicz_g52lcDkHIp88,5382
-nedo_vision_worker_core/streams/VideoStream.py,sha256=nGtJ4FAZ1Ek-8hVRopEt0bLWLpa10OtyUwdDEuXLObQ,13343
+nedo_vision_worker_core/streams/VideoStream.py,sha256=Xg3gxa9JZfCkwqD5OnvNmvf_iKmlEpjTlk4_YS2f2xU,14046
 nedo_vision_worker_core/streams/VideoStreamManager.py,sha256=g5cz-YXPewSubBXxCg4mfzsuGKoOHXu-SrMxaGjYPHw,14956
 nedo_vision_worker_core/streams/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 nedo_vision_worker_core/tracker/SFSORT.py,sha256=0kggw0l4yPZ55AKHdqVX6mu9ehHmJed7jcJ3JQoC4sk,14061
@@ -98,8 +98,8 @@ nedo_vision_worker_core/util/PipelinePreviewChecker.py,sha256=XxlSMlrDlRrzfV8_Y-
 nedo_vision_worker_core/util/PlatformDetector.py,sha256=GGL8UfeMQITR22EMYIRWnuOEnSqo7Dr5mb0PaFrl8AM,3006
 nedo_vision_worker_core/util/TablePrinter.py,sha256=wzLGgb1GFMeIbAP6HmKcZD33j4D-IlyqlyeR7C5yD7w,1137
 nedo_vision_worker_core/util/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
-nedo_vision_worker_core-0.4.3.dist-info/METADATA,sha256=niAjMl89ByK-mEOVchjKAZ2s52xyctOOIS-lLK7_QkU,14426
-nedo_vision_worker_core-0.4.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nedo_vision_worker_core-0.4.3.dist-info/entry_points.txt,sha256=pIPafsvPnBw-fpBKBmc1NQCQ6PQY3ad8mZ6mn8_p5FI,70
-nedo_vision_worker_core-0.4.3.dist-info/top_level.txt,sha256=y8kusXjVYqtG8MSHYWTrk8bRrvjOrphKXYyzu943TTQ,24
-nedo_vision_worker_core-0.4.3.dist-info/RECORD,,
+nedo_vision_worker_core-0.4.4.dist-info/METADATA,sha256=NeX_eRQ8lbk8z9HYN0ERFouzwjnbdc_xLXytxzvrQtw,14426
+nedo_vision_worker_core-0.4.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nedo_vision_worker_core-0.4.4.dist-info/entry_points.txt,sha256=pIPafsvPnBw-fpBKBmc1NQCQ6PQY3ad8mZ6mn8_p5FI,70
+nedo_vision_worker_core-0.4.4.dist-info/top_level.txt,sha256=y8kusXjVYqtG8MSHYWTrk8bRrvjOrphKXYyzu943TTQ,24
+nedo_vision_worker_core-0.4.4.dist-info/RECORD,,