nedo-vision-worker 1.2.7__py3-none-any.whl → 1.2.9__py3-none-any.whl

nedo_vision_worker/__init__.py

@@ -6,5 +6,5 @@ A library for running worker agents in the Nedo Vision platform.
 
 from .worker_service import WorkerService
 
-__version__ = "1.2.7"
+__version__ = "1.2.9"
 __all__ = ["WorkerService"] 

nedo_vision_worker/initializer/AppInitializer.py

@@ -76,3 +76,63 @@ class AppInitializer:
             logging.error(f"Grpc Error: {ge}")
         except Exception as e:
             logging.error(f"Unexpected error during initialization: {e}")
+
+    @staticmethod
+    def update_connection_info(server_host: str, server_port: int, token: str):
+        """
+        Fetch and update connection information (RabbitMQ credentials) from the server.
+        This should be called on startup to ensure credentials are up-to-date.
+        
+        Args:
+            server_host: The server hostname or IP address
+            server_port: The gRPC server port
+            token: Authentication token for the worker
+            
+        Returns:
+            bool: True if update was successful, False otherwise
+        """
+        try:
+            # Validate server host
+            AppInitializer.validate_server_host(server_host)
+
+            # Get connection info using the ConnectionInfoClient
+            connection_client = ConnectionInfoClient(server_host, server_port, token)
+            connection_result = connection_client.get_connection_info()
+
+            if not connection_result["success"]:
+                logging.error(f"Failed to fetch connection info: {connection_result['message']}")
+                return False
+
+            # Check if any RabbitMQ credentials have changed
+            current_config = ConfigurationManager.get_all_configs()
+            config_updated = False
+            
+            rabbitmq_fields = {
+                'rabbitmq_host': connection_result['rabbitmq_host'],
+                'rabbitmq_port': str(connection_result['rabbitmq_port']),
+                'rabbitmq_username': connection_result['rabbitmq_username'],
+                'rabbitmq_password': connection_result['rabbitmq_password']
+            }
+            
+            for field, new_value in rabbitmq_fields.items():
+                if current_config.get(field) != new_value:
+                    ConfigurationManager.set_config(field, new_value)
+                    config_updated = True
+                    logging.info(f"✅ [APP] Updated {field}")
+            
+            if config_updated:
+                logging.info("✅ [APP] RabbitMQ connection info updated successfully")
+            else:
+                logging.info("✅ [APP] RabbitMQ connection info is up-to-date")
+            
+            return True
+        
+        except ValueError as ve:
+            logging.error(f"Validation error: {ve}")
+            return False
+        except grpc.RpcError as ge:
+            logging.error(f"gRPC Error: {ge}")
+            return False
+        except Exception as e:
+            logging.error(f"Unexpected error updating connection info: {e}")
+            return False

nedo_vision_worker/repositories/PPEDetectionRepository.py

@@ -77,6 +77,10 @@ class PPEDetectionRepository:
         Args:
             detection_data (list): List of dictionaries containing the detection data.
         """
+        if not detection_data:
+            logging.info("No detection data provided for deletion.")
+            return
+            
         try:
             # Extract person_id from detection data to delete the corresponding records
             person_ids_to_delete = [data['person_id'] for data in detection_data]
@@ -97,10 +101,25 @@ class PPEDetectionRepository:
                 if not os.path.isabs(image_path):
                     image_path = str(get_storage_path("files") / Path(image_path).relative_to("data/files"))
                 if os.path.exists(image_path):
-                    os.remove(image_path)
-                    logging.info(f"Deleted image file: {image_path}")
+                    try:
+                        os.remove(image_path)
+                        logging.info(f"Deleted image file: {image_path}")
+                    except OSError as e:
+                        logging.warning(f"Failed to delete image file {image_path}: {e}")
                 else:
                     logging.warning(f"Image file not found for detection {detection.id}: {image_path}")
+                
+                # Delete the image tile file if it exists
+                image_tile_path = detection.image_tile_path
+                if image_tile_path:
+                    if not os.path.isabs(image_tile_path):
+                        image_tile_path = str(get_storage_path("files") / Path(image_tile_path).relative_to("data/files"))
+                    if os.path.exists(image_tile_path):
+                        try:
+                            os.remove(image_tile_path)
+                            logging.info(f"Deleted image tile file: {image_tile_path}")
+                        except OSError as e:
+                            logging.warning(f"Failed to delete image tile file {image_tile_path}: {e}")
 
                 # Delete the detection record
                 self.session.delete(detection)

nedo_vision_worker/repositories/RestrictedAreaRepository.py

@@ -61,6 +61,10 @@ class RestrictedAreaRepository:
         Args:
             violation_data (list): List of dictionaries containing the violation data.
         """
+        if not violation_data:
+            logging.info("No violation data provided for deletion.")
+            return
+            
         try:
             person_ids_to_delete = [data['person_id'] for data in violation_data]
 
@@ -75,11 +79,26 @@ class RestrictedAreaRepository:
                 if not os.path.isabs(image_path):
                     image_path = str(get_storage_path("restricted_violations") / Path(image_path).relative_to("data/restricted_violations"))
                 if os.path.exists(image_path):
-                    os.remove(image_path)
-                    logging.info(f"Deleted image file: {image_path}")
+                    try:
+                        os.remove(image_path)
+                        logging.info(f"Deleted image file: {image_path}")
+                    except OSError as e:
+                        logging.warning(f"Failed to delete image file {image_path}: {e}")
                 else:
                     logging.warning(f"Image file not found for violation {violation.id}: {image_path}")
 
+                # Delete the image tile file if it exists
+                image_tile_path = violation.image_tile_path
+                if image_tile_path:
+                    if not os.path.isabs(image_tile_path):
+                        image_tile_path = str(get_storage_path("restricted_violations") / Path(image_tile_path).relative_to("data/restricted_violations"))
+                    if os.path.exists(image_tile_path):
+                        try:
+                            os.remove(image_tile_path)
+                            logging.info(f"Deleted image tile file: {image_tile_path}")
+                        except OSError as e:
+                            logging.warning(f"Failed to delete image tile file {image_tile_path}: {e}")
+
                 self.session.delete(violation)
 
             self.session.commit()

nedo_vision_worker/services/DirectDeviceToRTMPStreamer.py

@@ -8,6 +8,7 @@ import cv2
 import os
 from .VideoSharingDaemon import VideoSharingClient
 from ..database.DatabaseManager import get_storage_path
+from ..util.EncoderSelector import EncoderSelector
 
 
 class DirectDeviceToRTMPStreamer:
@@ -68,6 +69,10 @@ class DirectDeviceToRTMPStreamer:
 
     def _start_ffmpeg_stream(self):
         """Starts an FFmpeg process to stream frames to the RTMP server silently."""
+        # Get optimal encoder for hardware
+        encoder_args, encoder_name = EncoderSelector.get_encoder_args()
+        logging.info(f"🎬 [APP] Using encoder: {encoder_name}")
+        
         ffmpeg_command = [
             "ffmpeg",
             "-y",
@@ -79,16 +84,17 @@ class DirectDeviceToRTMPStreamer:
             "-video_size", f"{self.width}x{self.height}",
             "-framerate", str(self.fps),
             "-i", "-",
-            "-c:v", "libx264",
-            "-preset", "ultrafast",
-            "-tune", "zerolatency",
+            
+            # Video encoding with optimal encoder
+            *encoder_args,
             "-b:v", self.bitrate,
-            # Disable Audio (Avoid unnecessary encoding overhead)
-            "-an",
             "-maxrate", "2500k",
             "-bufsize", "5000k",
+            
+            # Disable Audio
+            "-an",
+            
             "-f", "flv",
-            # Remove duration limit - let application control duration
             self.rtmp_url,
         ]
 

nedo_vision_worker/services/FileToRTMPServer.py

@@ -1,6 +1,7 @@
 import subprocess
 import logging
 import os
+from ..util.EncoderSelector import EncoderSelector
 
 class FileToRTMPStreamer:
     def __init__(self, video_path, rtmp_url, stream_key, fps=30, resolution="1280x720", loop=False):
@@ -31,6 +32,10 @@ class FileToRTMPStreamer:
 
         logging.info(f"📼 [APP] Starting file stream: {self.video_path} → {self.rtmp_url}")
 
+        # Get optimal encoder for hardware
+        encoder_args, encoder_name = EncoderSelector.get_encoder_args()
+        logging.info(f"🎬 [APP] Using encoder: {encoder_name}")
+
         # FFmpeg command
         ffmpeg_command = [
             "ffmpeg",
@@ -38,9 +43,8 @@ class FileToRTMPStreamer:
             "-stream_loop", "-1" if self.loop else "0",  # Loop if needed
             "-i", self.video_path,
 
-            "-c:v", "libx264",
-            "-preset", "ultrafast",
-            "-tune", "zerolatency",
+            # Video encoding with optimal encoder
+            *encoder_args,
             "-r", str(self.fps),
             "-b:v", "1500k",
             "-maxrate", "2000k",

nedo_vision_worker/services/PPEDetectionClient.py

@@ -1,4 +1,5 @@
 import logging
+import os
 from .GrpcClientBase import GrpcClientBase
 from ..protos.PPEDetectionService_pb2_grpc import PPEDetectionGRPCServiceStub
 from ..protos.PPEDetectionService_pb2 import UpsertPPEDetectionBatchRequest, UpsertPPEDetectionRequest, PPEDetectionLabelRequest
@@ -56,11 +57,23 @@ class PPEDetectionClient(GrpcClientBase):
             return {"success": False, "message": "gRPC connection is not established."}
 
         try:
-            # Prepare the list of UpsertPPEDetectionRequest messages
             ppe_detection_requests = []
+            valid_records = []
+            invalid_records = []
+            
             for data in detection_data:
-                image_binary = self.read_image_as_binary(data['image'])
-                image_tile_binary = self.read_image_as_binary(data['image_tile'])
+                if not os.path.exists(data['image']) or not os.path.exists(data['image_tile']):
+                    logger.warning(f"⚠️ Missing image files for person_id {data.get('person_id')}")
+                    invalid_records.append(data)
+                    continue
+                
+                try:
+                    image_binary = self.read_image_as_binary(data['image'])
+                    image_tile_binary = self.read_image_as_binary(data['image_tile'])
+                except Exception as e:
+                    logger.error(f"❌ Error reading images for person_id {data.get('person_id')}: {e}")
+                    invalid_records.append(data)
+                    continue
 
                 ppe_detection_labels = [
                     PPEDetectionLabelRequest(
@@ -73,11 +86,11 @@ class PPEDetectionClient(GrpcClientBase):
                     )
                     for label in data['ppe_detection_labels']
                 ]
-                source = data['worker_source_id']
+                
                 request = UpsertPPEDetectionRequest(
                     person_id=data['person_id'],
                     worker_id=worker_id,
-                    worker_source_id=source,
+                    worker_source_id=data['worker_source_id'],
                     image=image_binary,
                     image_tile=image_tile_binary,
                     worker_timestamp=data['worker_timestamp'],
@@ -85,18 +98,24 @@ class PPEDetectionClient(GrpcClientBase):
                     token=token
                 )
                 ppe_detection_requests.append(request)
+                valid_records.append(data)
+
+            if invalid_records:
+                logger.info(f"🧹 Deleting {len(invalid_records)} invalid PPE detection records")
+                self.repository.delete_records_from_db(invalid_records)
+
+            if not ppe_detection_requests:
+                return {"success": True, "message": "No valid detections to send"}
 
-            # Create the UpsertPPEDetectionBatchRequest
             batch_request = UpsertPPEDetectionBatchRequest(
                 ppe_detection_requests=ppe_detection_requests,
                 token=token
             )
 
-            # Call the UpsertBatch RPC
             response = self.handle_rpc(self.stub.UpsertBatch, batch_request)
             
             if response and response.success:
-                self.repository.delete_records_from_db(detection_data)
+                self.repository.delete_records_from_db(valid_records)
                 return {"success": True, "message": response.message}
 
             return {"success": False, "message": response.message if response else "Unknown error"}

nedo_vision_worker/services/RTSPtoRTMPStreamer.py

@@ -3,6 +3,7 @@ import logging
 import time
 import os
 from urllib.parse import urlparse
+from ..util.EncoderSelector import EncoderSelector
 
 class RTSPtoRTMPStreamer:
     def __init__(self, rtsp_url, rtmp_url, stream_key, fps=30, resolution="1280x720", duration=120):
@@ -38,6 +39,10 @@ class RTSPtoRTMPStreamer:
 
         logging.info(f"📡 [APP] Starting RTSP to RTMP stream: {self.rtsp_url} → {self.rtmp_url} for {self.duration} seconds")
 
+        # Get optimal encoder for hardware
+        encoder_args, encoder_name = EncoderSelector.get_encoder_args()
+        logging.info(f"🎬 [APP] Using encoder: {encoder_name}")
+
         # FFmpeg command
         ffmpeg_command = [
             "ffmpeg",
@@ -47,17 +52,14 @@ class RTSPtoRTMPStreamer:
             "-strict", "experimental",
             "-i", self.rtsp_url,
 
-            # ✅ Video Encoding (Fastest possible)
-            "-c:v", "libx264",
-            "-preset", "ultrafast",  # 🚀 Reduce CPU usage
-            "-tune", "zerolatency",  # 🚀 Optimize for real-time streaming
-            "-x264-params", "keyint=40:min-keyint=40",  # 🚀 Keyframe optimization
-            "-r", "25",  # ⏳ Limit FPS to 20 (prevents excessive encoding load)
-            "-b:v", "1500k",  # ✅ Lower bitrate to improve performance
+            # Video encoding with optimal encoder
+            *encoder_args,
+            "-r", "25",  # ⏳ Limit FPS to 25
+            "-b:v", "1500k",  # ✅ Bitrate
             "-maxrate", "2000k",  # ✅ Set max bitrate
             "-bufsize", "4000k",  # ✅ Reduce buffer latency
             "-g", "25",  # ✅ Reduce GOP size for faster keyframes
-            "-vf", "scale='min(1024,iw)':-2",  # ✅ Resize width to max 800px
+            "-vf", "scale='min(1024,iw)':-2",  # ✅ Resize width to max 1024px
 
             # ❌ Disable Audio (Avoid unnecessary encoding overhead)
             "-an",

nedo_vision_worker/services/RestrictedAreaClient.py

@@ -1,4 +1,5 @@
 import logging
+import os
 from ..repositories.RestrictedAreaRepository import RestrictedAreaRepository
 from .GrpcClientBase import GrpcClientBase
 from ..protos.HumanDetectionService_pb2_grpc import HumanDetectionGRPCServiceStub
@@ -60,10 +61,22 @@ class RestrictedAreaClient(GrpcClientBase):
 
         try:
             human_detection_requests = []
+            valid_records = []
+            invalid_records = []
 
             for data in violation_data:
-                image_binary = self.read_image_as_binary(data['image'])
-                image_tile_binary = self.read_image_as_binary(data['image_tile'])
+                if not os.path.exists(data['image']) or not os.path.exists(data['image_tile']):
+                    logger.warning(f"⚠️ Missing image files for person_id {data.get('person_id')}")
+                    invalid_records.append(data)
+                    continue
+                
+                try:
+                    image_binary = self.read_image_as_binary(data['image'])
+                    image_tile_binary = self.read_image_as_binary(data['image_tile'])
+                except Exception as e:
+                    logger.error(f"❌ Error reading images for person_id {data.get('person_id')}: {e}")
+                    invalid_records.append(data)
+                    continue
 
                 request = UpsertHumanDetectionRequest(
                     person_id=data['person_id'],
@@ -81,6 +94,14 @@ class RestrictedAreaClient(GrpcClientBase):
                 )
 
                 human_detection_requests.append(request)
+                valid_records.append(data)
+
+            if invalid_records:
+                logger.info(f"🧹 Deleting {len(invalid_records)} invalid violation records")
+                self.repository.delete_records_from_db(invalid_records)
+
+            if not human_detection_requests:
+                return {"success": True, "message": "No valid violations to send"}
 
             batch_request = UpsertHumanDetectionBatchRequest(
                 human_detection_requests=human_detection_requests,
@@ -90,7 +111,7 @@ class RestrictedAreaClient(GrpcClientBase):
             response = self.handle_rpc(self.stub.UpsertBatch, batch_request)
 
             if response and response.success:
-                self.repository.delete_records_from_db(violation_data)
+                self.repository.delete_records_from_db(valid_records)
                 return {"success": True, "message": response.message}
 
             return {"success": False, "message": response.message if response else "Unknown error"}

nedo_vision_worker/services/WorkerSourcePipelineClient.py

@@ -377,7 +377,15 @@ class WorkerSourcePipelineClient(GrpcClientBase):
             debug_entries = self.debug_repo.get_debug_entries_with_data()
 
             for debug_entry in debug_entries:
-                image_binary = self.read_image_as_binary(debug_entry.image_path)
+                try:
+                    image_binary = self.read_image_as_binary(debug_entry.image_path)
+                except FileNotFoundError:
+                    logging.warning(f"Image file not found: {debug_entry.image_path}, deleting entry {debug_entry.id}")
+                    self.debug_repo.delete_entry_by_id(debug_entry.id)
+                    continue
+                except Exception as e:
+                    logging.error(f"Error reading image {debug_entry.image_path}: {e}")
+                    continue
 
                 request = SendPipelineDebugRequest(
                     worker_source_pipeline_id=debug_entry.worker_source_pipeline_id,
@@ -391,12 +399,13 @@ class WorkerSourcePipelineClient(GrpcClientBase):
                 if response and response.success:
                     self.debug_repo.delete_entry_by_id(debug_entry.id)
                 else:
-                    return {"success": False, "message": response.message if response else "Unknown error"}
+                    logging.warning(f"Failed to sync debug entry {debug_entry.id}: {response.message if response else 'Unknown error'}")
 
             return {"success": True, "message": "Successfully synced debug entries"}
 
         except Exception as e:
-            logging.error(f"Error syncing pipeline debug: {e}")
+            logging.error(f"Error syncing pipeline debug: {e}", exc_info=True)
+            return {"success": False, "message": f"Exception: {str(e)}"}
 
     def sync_pipeline_detection(self, token: str):
         if not self.stub:
@@ -406,7 +415,15 @@ class WorkerSourcePipelineClient(GrpcClientBase):
             entries = self.detection_repo.get_entries()
 
             for entry in entries:
-                image_binary = self.read_image_as_binary(entry.image_path)
+                try:
+                    image_binary = self.read_image_as_binary(entry.image_path)
+                except FileNotFoundError:
+                    logging.warning(f"Image file not found: {entry.image_path}, deleting entry {entry.id}")
+                    self.detection_repo.delete_entry_by_id(entry.id)
+                    continue
+                except Exception as e:
+                    logging.error(f"Error reading image {entry.image_path}: {e}")
+                    continue
 
                 request = SendPipelineDetectionDataRequest(
                     worker_source_pipeline_id=entry.worker_source_pipeline_id,
@@ -420,9 +437,10 @@ class WorkerSourcePipelineClient(GrpcClientBase):
                 if response and response.success:
                     self.detection_repo.delete_entry_by_id(entry.id)
                 else:
-                    return {"success": False, "message": response.message if response else "Unknown error"}
+                    logging.warning(f"Failed to sync detection entry {entry.id}: {response.message if response else 'Unknown error'}")
 
-            return {"success": True, "message": "Successfully synced debug entries"}
+            return {"success": True, "message": "Successfully synced detection entries"}
 
         except Exception as e:
-            logging.error(f"Error syncing pipeline debug: {e}")
+            logging.error(f"Error syncing pipeline detection: {e}", exc_info=True)
+            return {"success": False, "message": f"Exception: {str(e)}"}

nedo_vision_worker/util/EncoderSelector.py

@@ -0,0 +1,109 @@
+"""
+Utility for selecting the best available video encoder (GPU or CPU fallback).
+"""
+import os
+import sys
+import logging
+from typing import List, Tuple
+from .PlatformDetector import PlatformDetector
+
+logger = logging.getLogger(__name__)
+
+
+class EncoderSelector:
+    """Selects optimal video encoder based on available hardware."""
+    
+    _platform = PlatformDetector()
+    
+    @classmethod
+    def get_encoder_args(cls, force_cpu: bool = False) -> Tuple[List[str], str]:
+        """
+        Get FFmpeg encoder arguments.
+        
+        Args:
+            force_cpu: Force CPU encoding even if GPU is available
+            
+        Returns:
+            Tuple of (encoder_args_list, encoder_name)
+        """
+        if force_cpu:
+            return cls._get_cpu_encoder()
+        
+        # Check environment variable override
+        force_encoder = os.environ.get("RTMP_ENCODER", "").lower()
+        
+        if force_encoder == "cpu" or force_encoder == "libx264":
+            return cls._get_cpu_encoder()
+        elif force_encoder == "nvenc":
+            return cls._get_nvenc_encoder()
+        
+        # Jetson platform
+        if cls._platform.is_jetson():
+            return cls._get_jetson_encoder()
+        
+        # macOS
+        if sys.platform == "darwin":
+            return cls._get_videotoolbox_encoder()
+        
+        # NVIDIA GPU
+        if cls._has_nvidia_gpu():
+            return cls._get_nvenc_encoder()
+        
+        # Fallback to CPU
+        return cls._get_cpu_encoder()
+    
+    @staticmethod
+    def _has_nvidia_gpu() -> bool:
+        """Check if NVIDIA GPU is available."""
+        return (
+            os.environ.get("NVIDIA_VISIBLE_DEVICES") is not None or
+            os.path.exists("/proc/driver/nvidia/version")
+        )
+    
+    @staticmethod
+    def _get_cpu_encoder() -> Tuple[List[str], str]:
+        """Get CPU encoder (libx264) with optimized settings."""
+        return [
+            "-c:v", "libx264",
+            "-preset", "ultrafast",
+            "-tune", "zerolatency",
+            "-profile:v", "main",
+        ], "libx264"
+    
+    @staticmethod
+    def _get_nvenc_encoder() -> Tuple[List[str], str]:
+        """Get NVIDIA NVENC encoder with GPU-optimized settings."""
+        return [
+            "-c:v", "h264_nvenc",
+            "-preset", "p1",  # p1 = fastest preset
+            "-tune", "ull",  # ultra-low latency
+            "-rc:v", "cbr",  # constant bitrate
+            "-rc-lookahead", "0",  # disable lookahead for lower latency
+            "-delay", "0",  # zero delay
+            "-zerolatency", "1",  # zero latency mode
+            "-profile:v", "main",
+            "-gpu", "0",  # Use first GPU
+        ], "h264_nvenc"
+    
+    @staticmethod
+    def _get_jetson_encoder() -> Tuple[List[str], str]:
+        """Get Jetson-optimized NVENC encoder."""
+        return [
+            "-c:v", "h264_nvenc",
+            "-preset", "p1",
+            "-tune", "ull",
+            "-rc:v", "cbr",
+            "-rc-lookahead", "0",
+            "-delay", "0",
+            "-zerolatency", "1",
+            "-profile:v", "main",
+        ], "h264_nvenc"
+    
+    @staticmethod
+    def _get_videotoolbox_encoder() -> Tuple[List[str], str]:
+        """Get macOS VideoToolbox encoder."""
+        return [
+            "-c:v", "h264_videotoolbox",
+            "-profile:v", "main",
+            "-realtime", "1",
+        ], "h264_videotoolbox"

nedo_vision_worker/worker/DataSyncWorker.py

@@ -138,7 +138,7 @@ class DataSyncWorker:
 
             if not response or not response.get("success"):
                 error_message = GrpcClientBase.get_error_message(response)
-                logger.error(f"❌ [APP] Failed to sync dataset sources: {error_message}")
+                logger.error(f"❌ [APP] Failed to sync worker source pipelines detection: {error_message}")
 
         except Exception as e:
-            logger.error("🚨 [APP] Error syncing worker source pipelines detection.", exc_info=True)
+            logger.error(f"🚨 [APP] Error syncing worker source pipelines detection: {e}", exc_info=True)

nedo_vision_worker/worker/DatasetFrameWorker.py

@@ -284,7 +284,7 @@ class DatasetFrameWorker:
                 self.last_sync_time = time.time()
             else:
                 error_message = response.get("message", "Unknown error") if response else "Unknown error"
-                logger.error(f"❌ [APP] Failed to sync dataset sources: {error_message}")
+                logger.error(f"❌ [APP] Failed to sync dataset sources: {error_message}", exc_info=True)
 
         except Exception as e:
             logger.error("🚨 [APP] Error syncing dataset sources.", exc_info=True)

nedo_vision_worker/worker_service.py

@@ -151,6 +151,16 @@ class WorkerService:
                 self.logger.info("✅ [APP] Configuration updated successfully")
             else:
                 self.logger.info("✅ [APP] Configuration found. No changes needed.")
+            
+            # Always fetch connection info on startup to check for updates
+            self.logger.info("🔄 [APP] Checking for connection info updates...")
+            token_to_use = self.token if self.token else config.get('token')
+            if token_to_use:
+                AppInitializer.update_connection_info(server_host, self.server_port, token_to_use)
+                # Reload config after potential updates
+                config = ConfigurationManager.get_all_configs()
+            else:
+                self.logger.warning("⚠️ [APP] No token available to fetch connection info updates")
         
         # Add runtime parameters to config
         config['rtmp_server'] = self.rtmp_server

{nedo_vision_worker-1.2.7.dist-info → nedo_vision_worker-1.2.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nedo-vision-worker
-Version: 1.2.7
+Version: 1.2.9
 Summary: Nedo Vision Worker Service 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-1.2.7.dist-info → nedo_vision_worker-1.2.9.dist-info}/RECORD

@@ -1,12 +1,12 @@
-nedo_vision_worker/__init__.py,sha256=1QtTEMaMlbF3PH-GmXuZzF1PvxRe_CUCme72WXqFci0,203
+nedo_vision_worker/__init__.py,sha256=oLzUhGDH-bezYa_Smc3pWfjWYzwQGUZ0LiTwVuu5AYA,203
 nedo_vision_worker/cli.py,sha256=ddWspJmSgVkcUYvRdkvTtMNuMTDvNCqLLuMVU9KE3Ik,7457
 nedo_vision_worker/doctor.py,sha256=wNkpe8gLVd76Y_ViyK2h1ZFdqeSl37MnzZN5frWKu30,48410
-nedo_vision_worker/worker_service.py,sha256=rXUVmyxcJPGhQEZ4UQvjQS5UqlnLBYudHQZCj0dQDxo,10421
+nedo_vision_worker/worker_service.py,sha256=9zz8hKwDwqwpfS0KPQfftGJtRci0uj_wiwcr_TGf-E0,11039
 nedo_vision_worker/config/ConfigurationManager.py,sha256=QrQaQ9Cdjpkcr2JE_miyrWJIZmMgZwJYBz-wE45Zzes,8011
 nedo_vision_worker/config/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 nedo_vision_worker/database/DatabaseManager.py,sha256=j2koXo1fnMmAyQnY4sv4txfZR8qIzrPyev-sQ4HBaOQ,9478
 nedo_vision_worker/database/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
-nedo_vision_worker/initializer/AppInitializer.py,sha256=iGw8-7Eg2aNT-nFaxiRIhWHU8iwecLuwuvjWlt1-V0Y,3163
+nedo_vision_worker/initializer/AppInitializer.py,sha256=6UVdjiuayziPYZ7JkQ436z7-9sHj7J3jtp6lfQsu-DU,5698
 nedo_vision_worker/initializer/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 nedo_vision_worker/models/__init__.py,sha256=6ZH2W1Jcy4o6xBPqFPcyxRU2UJ5Zvw_kfO38yLLGtHA,796
 nedo_vision_worker/models/ai_model.py,sha256=9muyZL9AxtX417-tYUiw8bgvFPtqdXgEAq-hm_mLxGY,2277
@@ -40,8 +40,8 @@ nedo_vision_worker/protos/WorkerSourceService_pb2_grpc.py,sha256=23Zj3WSogfhmp_Y
 nedo_vision_worker/protos/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 nedo_vision_worker/repositories/AIModelRepository.py,sha256=35cQ2EnJvMdArewumPrVIaiwueGy6cXU_g5J1Qcj6_Y,1415
 nedo_vision_worker/repositories/DatasetSourceRepository.py,sha256=MOUbXY1-7WUPhY15aMpdXl6rl0afSvu3Cq8jNtp9nzM,8595
-nedo_vision_worker/repositories/PPEDetectionRepository.py,sha256=oyQjIRyfa0pA-ErVtrKkCLMY521QIbz9ZqSXotC8nqY,4766
-nedo_vision_worker/repositories/RestrictedAreaRepository.py,sha256=y3n2ZfQbth1I_IjVYaT38qpRINQQC_dKeTf0yHPFNjw,3724
+nedo_vision_worker/repositories/PPEDetectionRepository.py,sha256=s4iCaTNYoJJgGr9ojWgoj6wtUQYFbloxSpnuntX26yw,5792
+nedo_vision_worker/repositories/RestrictedAreaRepository.py,sha256=XudFBscLCXCheUTAPC_CbwBkJMn2uIx2HCnU0NH4riE,4766
 nedo_vision_worker/repositories/WorkerSourcePipelineDebugRepository.py,sha256=kOlVEnPOoDRZdZIm8uWXlc89GMvBPI-36QyKecX7ucE,3350
 nedo_vision_worker/repositories/WorkerSourcePipelineDetectionRepository.py,sha256=cbgg_7p0eNUIgCHoPDZBaRZ1b2Y68p_dfSxpvuGMtRE,1773
 nedo_vision_worker/repositories/WorkerSourcePipelineRepository.py,sha256=xfmEvgnyt-DdfSApGyFfy0H0dXjFFkjeo4LMr0fVFXU,10053
@@ -50,15 +50,15 @@ nedo_vision_worker/repositories/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKr
 nedo_vision_worker/services/AIModelClient.py,sha256=lxRNax6FR-pV0G1NpJnlaqjbQeu3kRolIUNSw1RkoZA,15406
 nedo_vision_worker/services/ConnectionInfoClient.py,sha256=toC9zuY2Hrx1Cwq8Gycy_iFlaG1DvFT4qewlLlitpEQ,2214
 nedo_vision_worker/services/DatasetSourceClient.py,sha256=O5a7onxFl0z47zXaMXWxHAMPuuc-i_vzkd2w5fwrukc,3319
-nedo_vision_worker/services/DirectDeviceToRTMPStreamer.py,sha256=M5ei0cd3_KDhHZp6EkrOowhAY-hAHfAQh9YDVjQtbQI,22278
-nedo_vision_worker/services/FileToRTMPServer.py,sha256=yUJxrouoTLSq9XZ88dhDYhP-px10jLoHopkPoy4lQxk,2663
+nedo_vision_worker/services/DirectDeviceToRTMPStreamer.py,sha256=K0n7iyR7jdce8IWdNuqa1Im-R00QnxBcb7u2KM2Wjbc,22423
+nedo_vision_worker/services/FileToRTMPServer.py,sha256=0hY5pmeAzLw_d3uPR2Qp6gSAYb4rJHiAunuNe08OvkM,2870
 nedo_vision_worker/services/GrpcClientBase.py,sha256=hPyxOGw3aGSW1FhmY3wp3Iq8U1MArXBmvEMdmd63NZ4,6827
 nedo_vision_worker/services/GrpcClientManager.py,sha256=DLXekmxlQogLo8V9-TNDXtyHT_UG-BaggqwsIups55k,5568
 nedo_vision_worker/services/GrpcConnection.py,sha256=UNjaUC4ZcXuteHQx8AAAL5ymYkT1OpoIvyCYPUc3tCI,4915
 nedo_vision_worker/services/ImageUploadClient.py,sha256=T353YsRfm74G7Mh-eWr5nvdQHXTfpKwHJFmNW8HyjT8,3019
-nedo_vision_worker/services/PPEDetectionClient.py,sha256=CC-b0LRAgrftfIKp6TFKpeBkTYefe-C6Z1oz_X3HArQ,4345
-nedo_vision_worker/services/RTSPtoRTMPStreamer.py,sha256=hkPRX6iGxUtivsOA36JWBopHEd1RlmgIP8SLqN4d2TU,3863
-nedo_vision_worker/services/RestrictedAreaClient.py,sha256=AE9SOcVQca4zn5V96boho56EgN5BCIpV-8grvFBBnGo,3853
+nedo_vision_worker/services/PPEDetectionClient.py,sha256=3oE_Y0Avw_kcWKgTA7qmChq32woW7JerWefrCxllUYU,5172
+nedo_vision_worker/services/RTSPtoRTMPStreamer.py,sha256=LtfrWDHNcm-Ky6nZLnFCF8xgqIm7VQmsWIenK2yKNfo,3804
+nedo_vision_worker/services/RestrictedAreaClient.py,sha256=TD2Y5UJ0NjXgNFNgwS5ze4P-5jWwgrP0j0Nlyyd7-6Q,4844
 nedo_vision_worker/services/SharedDirectDeviceClient.py,sha256=dylMhqpMsfK_UKLWIVL-ApJRP4g-NCP_55xvlGYBiwo,10760
 nedo_vision_worker/services/SharedVideoStreamServer.py,sha256=WMKVxkzMoyfbgYiJ0fQOT-Ujz9btz6FLlaDP738yfoY,11601
 nedo_vision_worker/services/SystemUsageClient.py,sha256=Yf77dooQeNh6CDL5FkWVrX9543OVz1wc3exCAg6GlWw,3273
@@ -66,10 +66,11 @@ nedo_vision_worker/services/SystemWideDeviceCoordinator.py,sha256=9zBJMCbTMZS7gw
 nedo_vision_worker/services/VideoSharingDaemon.py,sha256=hYMjUIKNUVT1qSxuUuHN-7Bd85MDkxfqslxDLe2PBYQ,29721
 nedo_vision_worker/services/VideoStreamClient.py,sha256=QSgUV3LijYrNdnBG1ylABOdUaSatQamfXaqJhAiol9M,7260
 nedo_vision_worker/services/WorkerSourceClient.py,sha256=vDZeCuHL5QQ2-knZ4TOSA59jzmbbThGIwFKKLEZ72Ws,9198
-nedo_vision_worker/services/WorkerSourcePipelineClient.py,sha256=qaBx9T2gWMzpqZaeQdbIeklsXNwzWD5kqgB41rrSkBI,17135
+nedo_vision_worker/services/WorkerSourcePipelineClient.py,sha256=cjev_NRGUZrC9tuMm8s5ov92fknQno4vLEt-yMFrUCY,18241
 nedo_vision_worker/services/WorkerSourceUpdater.py,sha256=4t_CEHBLGDRvvuQS6eEPMivTI11ZuzusKKto6t9tPIk,9115
 nedo_vision_worker/services/WorkerStatusClient.py,sha256=7kC5EZjEBwWtHOE6UQ29OPCpYnv_6HSuH7Tc0alK_2Q,2531
 nedo_vision_worker/services/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
+nedo_vision_worker/util/EncoderSelector.py,sha256=-9lZwVmiKzJr1cELeuCXi-jRonty2bpociZq4KDScmA,3399
 nedo_vision_worker/util/FFmpegUtil.py,sha256=QnQrzurmllzGb7SlAAYCrzKBUblweoFU-0h-X-32IYg,1829
 nedo_vision_worker/util/HardwareID.py,sha256=rSW8-6stm7rjXEdkYGqXMUn56gyw62YiWnSwZQVCCLM,4315
 nedo_vision_worker/util/ImageUploader.py,sha256=2xipN3fwpKgFmbvoGIdElpGn5ARJyrgR4dXtbRf73hw,3764
@@ -80,9 +81,9 @@ nedo_vision_worker/util/VideoProbeUtil.py,sha256=cF-vJ7hIDlXfEJby2a0s9tqwkPGVz_6
 nedo_vision_worker/util/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 nedo_vision_worker/worker/CoreActionWorker.py,sha256=lb7zPY3yui6I3F4rX4Ii7JwpWZahLEO72rh3iWOgFmg,5441
 nedo_vision_worker/worker/DataSenderWorker.py,sha256=9FudRRItiMOcQx5UfVyu4p0Enb9BbgwZZ5EgX6Ho2U4,7160
-nedo_vision_worker/worker/DataSyncWorker.py,sha256=WvYfi3bG4mOKHU09J_MavfjFPrVgmxrrZYtrlQ-bnio,6265
+nedo_vision_worker/worker/DataSyncWorker.py,sha256=LmDPt2J1frmXwuR46L6b0MjlFOHfgG-4_0MGQa78zF4,6288
 nedo_vision_worker/worker/DatasetFrameSender.py,sha256=1SFYj8LJFNi-anBTapsbq8U_NGMM7mnoMKg9NeFAHys,8087
-nedo_vision_worker/worker/DatasetFrameWorker.py,sha256=Ni5gPeDPk9Rz4_cbg63u7Y6LVw_-Bz24OvfeY-6Yp44,19320
+nedo_vision_worker/worker/DatasetFrameWorker.py,sha256=Hh_wZuMjwovxsEKFqXSuTRin9eYRBZCbcFKm3CKLMbE,19335
 nedo_vision_worker/worker/PPEDetectionManager.py,sha256=fAolWlrsY5SQAWygvjNBNU56IlC0HLlhPgpz7shL-gk,3588
 nedo_vision_worker/worker/PipelineActionWorker.py,sha256=xgvryjKtEsMj4BKqWzDIaK_lFny-DfMCj5Y2DxHnWww,5651
 nedo_vision_worker/worker/PipelineImageWorker.py,sha256=J8VBUG0cwcH3qOJp2zTl30B-XhmPFyvJLjxitKJYq0E,5642
@@ -93,8 +94,8 @@ nedo_vision_worker/worker/SystemUsageManager.py,sha256=mkh4sT-HkIEY1CJHMEG6LP9AT
 nedo_vision_worker/worker/VideoStreamWorker.py,sha256=5n6v1PNO7IB-jj_McALLkUP-cBjJoIEw4UiSAs3vTb0,7606
 nedo_vision_worker/worker/WorkerManager.py,sha256=2bxXi19fp3p1qjYBStYRdVVgko8dnevXx1_M_sqH5og,5521
 nedo_vision_worker/worker/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
-nedo_vision_worker-1.2.7.dist-info/METADATA,sha256=xcM1OD1cAy83Rz4OaFYnHsHO8-DgNBrP0zEX9-_uX3I,14728
-nedo_vision_worker-1.2.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nedo_vision_worker-1.2.7.dist-info/entry_points.txt,sha256=LrglS-8nCi8C_PL_pa6uxdgCe879hBETHDVXAckvs-8,60
-nedo_vision_worker-1.2.7.dist-info/top_level.txt,sha256=vgilhlkyD34YsEKkaBabmhIpcKSvF3XpzD2By68L-XI,19
-nedo_vision_worker-1.2.7.dist-info/RECORD,,
+nedo_vision_worker-1.2.9.dist-info/METADATA,sha256=ySyI3I8lzUmhonotTgOrhH2Tt-AxL_l9oYifd_qaHcU,14728
+nedo_vision_worker-1.2.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nedo_vision_worker-1.2.9.dist-info/entry_points.txt,sha256=LrglS-8nCi8C_PL_pa6uxdgCe879hBETHDVXAckvs-8,60
+nedo_vision_worker-1.2.9.dist-info/top_level.txt,sha256=vgilhlkyD34YsEKkaBabmhIpcKSvF3XpzD2By68L-XI,19
+nedo_vision_worker-1.2.9.dist-info/RECORD,,