nedo-vision-worker 1.1.0__tar.gz → 1.1.2__tar.gz

{nedo_vision_worker-1.1.0 → nedo_vision_worker-1.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nedo-vision-worker
-Version: 1.1.0
+Version: 1.1.2
 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 → nedo_vision_worker-1.1.2}/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.2"
 __all__ = ["WorkerService"] 

{nedo_vision_worker-1.1.0 → nedo_vision_worker-1.1.2}/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.2"
     )
     
     parser.add_argument(

{nedo_vision_worker-1.1.0 → nedo_vision_worker-1.1.2}/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-1.1.0 → nedo_vision_worker-1.1.2}/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-1.1.0 → nedo_vision_worker-1.1.2}/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-1.1.0 → nedo_vision_worker-1.1.2}/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 → nedo_vision_worker-1.1.2}/nedo_vision_worker.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nedo-vision-worker
-Version: 1.1.0
+Version: 1.1.2
 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>