matrice-streaming 0.1.60__py3-none-any.whl → 0.1.61__py3-none-any.whl

matrice_streaming/streaming_gateway/camera_streamer/__init__.py

@@ -26,14 +26,33 @@ except (ImportError, ValueError):
     GStreamerPipeline = None
     GStreamerAsyncWorker = None
     GStreamerWorkerManager = None
-    
+
     def is_gstreamer_available():
         return False
 
+# NVDEC components (optional - graceful import)
+try:
+    from .nvdec_worker_manager import (
+        NVDECWorkerManager,
+        is_nvdec_available,
+        get_available_gpu_count,
+    )
+    NVDEC_AVAILABLE = is_nvdec_available()
+except ImportError:
+    # NVDEC not available (requires CuPy, PyNvVideoCodec)
+    NVDEC_AVAILABLE = False
+    NVDECWorkerManager = None
+
+    def is_nvdec_available():
+        return False
+
+    def get_available_gpu_count():
+        return 1
+
 __all__ = [
     # Original components
     "CameraStreamer",
-    "WorkerManager", 
+    "WorkerManager",
     "AsyncCameraWorker",
     # GStreamer components
     "GStreamerCameraStreamer",
@@ -43,4 +62,9 @@ __all__ = [
     "GStreamerWorkerManager",
     "is_gstreamer_available",
     "GSTREAMER_AVAILABLE",
+    # NVDEC components
+    "NVDECWorkerManager",
+    "is_nvdec_available",
+    "get_available_gpu_count",
+    "NVDEC_AVAILABLE",
 ]