nedo-vision-worker 1.1.0__py3-none-any.whl → 1.1.1__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.1.0"
+__version__ = "1.1.1"
 __all__ = ["WorkerService"] 

nedo_vision_worker/cli.py

@@ -115,7 +115,7 @@ Examples:
     parser.add_argument(
         "--version",
         action="version",
-        version="nedo-vision-worker 1.1.0"
+        version="nedo-vision-worker 1.1.1"
     )
     
     parser.add_argument(

nedo_vision_worker/repositories/PPEDetectionRepository.py

@@ -94,6 +94,8 @@ class PPEDetectionRepository:
 
                 # Delete the image file associated with the detection if it exists
                 image_path = detection.image_path
+                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}")

nedo_vision_worker/repositories/RestrictedAreaRepository.py

@@ -72,6 +72,8 @@ class RestrictedAreaRepository:
 
             for violation in violations_to_delete:
                 image_path = violation.image_path
+                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}")

nedo_vision_worker/services/PPEDetectionClient.py

@@ -38,6 +38,11 @@ class PPEDetectionClient(GrpcClientBase):
         Returns:
             bytes: Binary content of the image.
         """
+        from ..database.DatabaseManager import get_storage_path
+        from pathlib import Path
+        import os
+        if not os.path.isabs(image_path):
+            image_path = str(get_storage_path("files") / Path(image_path).relative_to("data/files"))
         with open(image_path, 'rb') as image_file:
             return image_file.read()
 

nedo_vision_worker/services/RestrictedAreaClient.py

@@ -39,6 +39,11 @@ class RestrictedAreaClient(GrpcClientBase):
         Returns:
             bytes: Binary content of the image.
         """
+        from ..database.DatabaseManager import get_storage_path
+        from pathlib import Path
+        import os
+        if not os.path.isabs(image_path):
+            image_path = str(get_storage_path("restricted_violations") / Path(image_path).relative_to("data/restricted_violations"))
         with open(image_path, 'rb') as image_file:
             return image_file.read()
 

{nedo_vision_worker-1.1.0.dist-info → nedo_vision_worker-1.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nedo-vision-worker
-Version: 1.1.0
+Version: 1.1.1
 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.1.0.dist-info → nedo_vision_worker-1.1.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
-nedo_vision_worker/__init__.py,sha256=vNz9DUyK7nrpItKgWomp9-IkZXjqbztGNhAa1m97sHQ,203
-nedo_vision_worker/cli.py,sha256=ZzkRXUvewieC81OBT_rFnUYCXkDiBZKoI3yV3jaQsvE,6227
+nedo_vision_worker/__init__.py,sha256=mgPDe2FHxWQLXYxpyKnft8nCtd1Jkbwb2x1oSn8wV9o,203
+nedo_vision_worker/cli.py,sha256=9NgdpolGqMMkDtg79OxzRgd25jn5iW-nuGL2TI14GB0,6227
 nedo_vision_worker/doctor.py,sha256=dO3eOOauAFrCwj1UX1A-1jj_zsE2Y0UycGqPXaEOKjg,16152
 nedo_vision_worker/worker_service.py,sha256=XLzaN-n8f4ysPcOmJpO51I9_MWuVTnFMJF08sVHT7IA,9882
 nedo_vision_worker/config/ConfigurationManager.py,sha256=QrQaQ9Cdjpkcr2JE_miyrWJIZmMgZwJYBz-wE45Zzes,8011
@@ -40,8 +40,8 @@ nedo_vision_worker/protos/WorkerSourceService_pb2_grpc.py,sha256=fTxFjgr9gZjx2rI
 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=l3VTveoAXdHLJwruIKYlDXbU3A3MPrCWPFbGZOdYkBY,4607
-nedo_vision_worker/repositories/RestrictedAreaRepository.py,sha256=RWXRmnDMGgV6jPCqrwETfYP7PXszJNHERmCoAAlf1xU,3533
+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/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
@@ -54,9 +54,9 @@ nedo_vision_worker/services/FileToRTMPServer.py,sha256=yUJxrouoTLSq9XZ88dhDYhP-p
 nedo_vision_worker/services/GrpcClientBase.py,sha256=zEqDglS48HmcLw77LWc7l6nLYeVy_w_6y2O5pB9C88A,6344
 nedo_vision_worker/services/GrpcClientManager.py,sha256=DLXekmxlQogLo8V9-TNDXtyHT_UG-BaggqwsIups55k,5568
 nedo_vision_worker/services/ImageUploadClient.py,sha256=T353YsRfm74G7Mh-eWr5nvdQHXTfpKwHJFmNW8HyjT8,3019
-nedo_vision_worker/services/PPEDetectionClient.py,sha256=KqC391gm6dFQk7rMsswD2SlO9mKeKjfMRKFzHoc_tLY,4367
+nedo_vision_worker/services/PPEDetectionClient.py,sha256=gX2EC4NBkKX5EvqQ8r7pAsaFTiKrl9x6hXA5OVrumt4,4625
 nedo_vision_worker/services/RTSPtoRTMPStreamer.py,sha256=hkPRX6iGxUtivsOA36JWBopHEd1RlmgIP8SLqN4d2TU,3863
-nedo_vision_worker/services/RestrictedAreaClient.py,sha256=AE9SOcVQca4zn5V96boho56EgN5BCIpV-8grvFBBnGo,3853
+nedo_vision_worker/services/RestrictedAreaClient.py,sha256=-E1Ao_Pxi1DKFWDTotvyXPDgyvxbGox3wwWxKcVaXA8,4143
 nedo_vision_worker/services/SystemUsageClient.py,sha256=PbRuwDWKnMarcnkGtOKfYB5nA-3DeKv7V5_hZr8EDEo,3200
 nedo_vision_worker/services/VideoStreamClient.py,sha256=0PImMGwbPsX0Wfb4wsA42-Qbo8it0aWI-FpQSEV3f_A,6322
 nedo_vision_worker/services/WorkerSourceClient.py,sha256=pSbcPLNlnUERtHE9HyKoYtiptulcQ_DPkygKq-9Qm0M,9186
@@ -85,8 +85,8 @@ nedo_vision_worker/worker/SystemUsageManager.py,sha256=StutV4UyLUfduYfK20de4SbPd
 nedo_vision_worker/worker/VideoStreamWorker.py,sha256=TxoMYoaU6IcwJDLILC71WUkYahqVAnNX05_5qPbeS_8,6514
 nedo_vision_worker/worker/WorkerManager.py,sha256=7G8ORHpcVAb73UbTUi2WhcQJ-A1XFSq9d_9dLxABHac,6357
 nedo_vision_worker/worker/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
-nedo_vision_worker-1.1.0.dist-info/METADATA,sha256=kSwsqgaekOKQtuFLEk52zpSvSAqXygiwENX0d_X7YSI,14642
-nedo_vision_worker-1.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nedo_vision_worker-1.1.0.dist-info/entry_points.txt,sha256=LrglS-8nCi8C_PL_pa6uxdgCe879hBETHDVXAckvs-8,60
-nedo_vision_worker-1.1.0.dist-info/top_level.txt,sha256=vgilhlkyD34YsEKkaBabmhIpcKSvF3XpzD2By68L-XI,19
-nedo_vision_worker-1.1.0.dist-info/RECORD,,
+nedo_vision_worker-1.1.1.dist-info/METADATA,sha256=NsS4IUwJP_PRvuEMzT8o2EQRny_tGAEoYxOXn0iiz2k,14642
+nedo_vision_worker-1.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nedo_vision_worker-1.1.1.dist-info/entry_points.txt,sha256=LrglS-8nCi8C_PL_pa6uxdgCe879hBETHDVXAckvs-8,60
+nedo_vision_worker-1.1.1.dist-info/top_level.txt,sha256=vgilhlkyD34YsEKkaBabmhIpcKSvF3XpzD2By68L-XI,19
+nedo_vision_worker-1.1.1.dist-info/RECORD,,