matrice 1.0.99397__tar.gz → 1.0.99399__tar.gz

{matrice-1.0.99397/src/matrice.egg-info → matrice-1.0.99399}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrice
-Version: 1.0.99397
+Version: 1.0.99399
 Summary: SDK for connecting to matrice.ai services
 Home-page: https://github.com/matrice-ai/python-sdk
 Author: Matrice.ai

{matrice-1.0.99397 → matrice-1.0.99399}/setup.py

@@ -14,7 +14,7 @@ long_description = (here / "README.md").read_text(encoding="utf-8")
 setup(
     name="matrice",
 
-    version = "1.0.99397",
+    version = "1.0.99399",
     
     description="SDK for connecting to matrice.ai services",
     long_description=long_description,

{matrice-1.0.99397 → matrice-1.0.99399}/src/matrice/deploy/utils/post_processing/config.py

@@ -53,6 +53,7 @@ APP_NAME_TO_USECASE = {
     "gas_leak_detection": "gas_leak_detection",
     "color_detection": "color_detection",
     "License Plate Recognition" : "license_plate_monitor",
+    "cell_microscopy_segmentation": "cell_microscopy_segmentation"
     
 }
 
@@ -110,7 +111,8 @@ APP_NAME_TO_CATEGORY = {
     "abandoned_object_detection" : "security",
     "gas_leak_detection": "oil_gas",
     "color_detection": "visual_appearance",
-    "License Plate Recognition" : "license_plate_monitor"
+    "License Plate Recognition" : "license_plate_monitor",
+    "cell_microscopy_segmentation" : "healthcare"
 }
 
 def get_usecase_from_app_name(app_name: str) -> str:

{matrice-1.0.99397 → matrice-1.0.99399}/src/matrice/deploy/utils/post_processing/core/config.py

@@ -632,6 +632,7 @@ class ConfigManager:
             'plaque_segmentation_img': None,
             'cardiomegaly_classification': None,
             'histopathological_cancer_detection' : None,
+            'cell_microscopy_segmentation': None,
         }
 
     def register_config_class(self, usecase: str, config_class: type) -> None:
@@ -1110,6 +1111,14 @@ class ConfigManager:
             return HistopathologicalCancerDetectionConfig
         except ImportError:
             return None
+        
+    def cell_microscopy_segmentation_config_class(self):
+        """Register a configuration class for a use case."""
+        try:
+            from ..usecases.cell_microscopy_segmentation import CellMicroscopyConfig
+            return CellMicroscopyConfig
+        except ImportError:
+            return None
 
     def create_config(self, usecase: str, category: Optional[str] = None, **kwargs) -> BaseConfig:
         """
@@ -2213,6 +2222,21 @@ class ConfigManager:
                 alert_config=alert_config,
                 **kwargs
             )
+        elif usecase == "cell_microscopy_segmentation":
+            # Import here to avoid circular import
+            from ..usecases.cell_microscopy_segmentation import CellMicroscopyConfig
+
+            # Handle nested configurations
+            alert_config = kwargs.pop("alert_config", None)
+            if alert_config and isinstance(alert_config, dict):
+                alert_config = AlertConfig(**alert_config)
+
+            config = CellMicroscopyConfig(
+                category=category or "healthcare",
+                usecase=usecase,
+                alert_config=alert_config,
+                **kwargs
+            )
             
         else:
             raise ConfigValidationError(f"Unknown use case: {usecase}")
@@ -2640,6 +2664,11 @@ class ConfigManager:
             from ..usecases.Histopathological_Cancer_Detection_img import HistopathologicalCancerDetectionConfig
             default_config = HistopathologicalCancerDetectionConfig()
             return default_config.to_dict()
+        elif usecase == "cell_microscopy_segmentation":
+            # Import here to avoid circular import
+            from ..usecases.cell_microscopy_segmentation import CellMicroscopyConfig
+            default_config = CellMicroscopyConfig()
+            return default_config.to_dict()
 
         elif usecase not in self._config_classes:
             raise ConfigValidationError(f"Unsupported use case: {usecase}")

{matrice-1.0.99397 → matrice-1.0.99399}/src/matrice/deploy/utils/post_processing/processor.py

@@ -105,6 +105,7 @@ from .usecases import (
     PlaqueSegmentationUseCase,
     CardiomegalyUseCase,
     HistopathologicalCancerDetectionUseCase,
+    CellMicroscopyUseCase,
 
 
 )
@@ -246,6 +247,7 @@ class PostProcessor:
         registry.register_use_case("healthcare", "plaque_img_segmentation", PlaqueSegmentationUseCase)
         registry.register_use_case("healthcare", "cardiomegaly_classification", CardiomegalyUseCase)
         registry.register_use_case("healthcare", "histopathological_cancer_detection", HistopathologicalCancerDetectionUseCase)
+        registry.register_use_case("healthcare", "cell_microscopy_segmentation", CellMicroscopyUseCase)
         
 
         logger.debug("Registered use cases with registry")
@@ -432,6 +434,8 @@ class PostProcessor:
                 result = use_case.process(data, parsed_config, context, stream_info)
             elif isinstance(use_case, HistopathologicalCancerDetectionUseCase):
                 result = use_case.process(data, parsed_config, context, stream_info)
+            elif isinstance(use_case, CellMicroscopyUseCase):
+                result = use_case.process(data, parsed_config, context, stream_info)
 
             else:
                 result = use_case.process(data, parsed_config, context, stream_info)

{matrice-1.0.99397 → matrice-1.0.99399}/src/matrice/deploy/utils/post_processing/usecases/__init__.py

@@ -84,6 +84,8 @@ from .skin_cancer_classification_img import SkinCancerClassificationConfig, Skin
 from .plaque_segmentation_img import PlaqueSegmentationConfig, PlaqueSegmentationUseCase
 from .cardiomegaly_classification import CardiomegalyConfig, CardiomegalyUseCase
 from .Histopathological_Cancer_Detection_img import HistopathologicalCancerDetectionConfig,HistopathologicalCancerDetectionUseCase
+from .cell_microscopy_segmentation import CellMicroscopyConfig, CellMicroscopyUseCase
+
 
 __all__ = [
     'PeopleCountingUseCase',
@@ -156,6 +158,7 @@ __all__ = [
     'PlaqueSegmentationUseCase',
     'CardiomegalyUseCase',
     'HistopathologicalCancerDetectionUseCase',
+    'CellMicroscopyUseCase',
 
 
 
@@ -230,6 +233,7 @@ __all__ = [
     'PlaqueSegmentationConfig',
     'CardiomegalyConfig',
     'HistopathologicalCancerDetectionConfig',
+    'CellMicroscopyConfig',
 
 
 ]