supervisely 6.73.376__py3-none-any.whl → 6.73.378__py3-none-any.whl

supervisely/__init__.py

@@ -133,6 +133,7 @@ from supervisely.cli import _handle_creds_error_to_console
 from supervisely._utils import (
     rand_str,
     batched,
+    batched_iter,
     get_bytes_hash,
     generate_names,
     ENTERPRISE,

supervisely/_utils.py

@@ -98,6 +98,17 @@ def batched(seq, batch_size=50):
         yield seq[i : i + batch_size]
 
 
+def batched_iter(iterable, batch_size=50):
+    batch = []
+    for item in iterable:
+        batch.append(item)
+        if len(batch) == batch_size:
+            yield batch
+            batch = []
+    if batch:
+        yield batch
+
+
 def get_bytes_hash(bytes):
     return base64.b64encode(hashlib.sha256(bytes).digest()).decode("utf-8")
 

supervisely/api/entity_annotation/tag_api.py

@@ -1,6 +1,6 @@
 # coding: utf-8
 
-from typing import List, Optional
+from typing import Any, Dict, List, Optional, Union
 
 from supervisely._utils import batched
 from supervisely.api.module_api import ApiField, ModuleApi
@@ -210,11 +210,13 @@ class TagApi(ModuleApi):
             raise RuntimeError("SDK error: len(tags_keys) != len(tags_to_add)")
         if len(tags_keys) == 0:
             return
-        ids = self.append_to_objects_json(entity_id, tags_to_add)
+        ids = self.append_to_objects_json(entity_id, tags_to_add, project_id)
         KeyIdMap.add_tags_to(key_id_map, tags_keys, ids)
         return ids
 
-    def append_to_objects_json(self, entity_id: int, tags_json: dict) -> list:
+    def append_to_objects_json(
+        self, entity_id: int, tags_json: List[Dict], project_id: Optional[int] = None
+    ) -> List[int]:
         """
         Add Tags to Annotation Objects for specific entity (image etc.).
 
@@ -224,14 +226,50 @@ class TagApi(ModuleApi):
         :type tags_json: dict
         :return: List of tags IDs
         :rtype: list
+
+        :Usage example:
+
+             .. code-block:: python
+
+                import supervisely as sly
+
+                api = sly.Api(server_address, token)
+
+                tags_list = [
+                    {
+                        "tagId": 25926,
+                        "objectId": 652959,
+                        "value": None
+                    },
+                    {
+                        "tagId": 25927,
+                        "objectId": 652959,
+                        "value": "v1"
+                    },
+                    {
+                        "tagId": 25927,
+                        "objectId": 652958,
+                        "value": "v2"
+                    }
+                ]
+                response = api.video.tag.append_to_objects_json(12345, tags_list)
+
+                print(response)
+                # Output:
+                #    [
+                #        80421101,
+                #        80421102,
+                #        80421103
+                #    ]
         """
 
         if len(tags_json) == 0:
             return []
-        response = self._api.post(
-            "annotation-objects.tags.bulk.add",
-            {ApiField.ENTITY_ID: entity_id, ApiField.TAGS: tags_json},
-        )
+        if project_id is not None:
+            json_data = {ApiField.PROJECT_ID: project_id, ApiField.TAGS: tags_json}
+        else:
+            json_data = {ApiField.ENTITY_ID: entity_id, ApiField.TAGS: tags_json}
+        response = self._api.post("annotation-objects.tags.bulk.add", json_data)
         ids = [obj[ApiField.ID] for obj in response.json()]
         return ids
 
@@ -242,11 +280,21 @@ class TagApi(ModuleApi):
         batch_size: int = 100,
         log_progress: bool = False,
         progress: Optional[tqdm_sly] = None,
-    ) -> List[dict]:
+    ) -> List[Dict[str, Union[str, int, None]]]:
         """
-        Add Tags to existing Annotation Figures.
-        All figures must belong to entities of the same project.
-        Applies only to images project.
+        For images project:
+            Add Tags to existing Annotation Figures (labels).
+            The `tags_list` example:
+                [{"tagId": 12345, "figureId": 54321, "value": "tag_value"}, ...].
+        For video, pointcloud, volume and pointcloud episodes projects:
+            Add Tags to existing Annotation Objects.
+            The `frameRange` field is optional and is supported only for video and pointcloud episodes projects.
+            The `tags_list`` example:
+                [{"tagId": 12345, "objectId": 54321, "value": "tag_value"}, ...].
+            or with frameRange:
+                [{"tagId": 12345, "objectId": 54321, "value": "tag_value", "frameRange": [1, 10]}, ...].
+
+        All objects must belong to entities of the same project.
 
         :param project_id: Project ID in Supervisely.
         :type project_id: int
@@ -259,7 +307,7 @@ class TagApi(ModuleApi):
         :param progress: Progress bar object to display progress.
         :type progress: Optional[tqdm_sly]
         :return: List of tags infos as dictionaries.
-        :rtype: List[dict]
+        :rtype: List[Dict[str, Union[str, int, None]]]
 
         Usage example:
         .. code-block:: python
@@ -272,7 +320,8 @@ class TagApi(ModuleApi):
                 {
                     "tagId": 25926,
                     "figureId": 652959,
-                    "value": None # value is optional for tag with type 'None'
+                    "value": None # optional for tag with type 'None'
+                    "frameRange": [1, 10] # optional (supported only for video and pointcloud episodes projects)
                 },
                 {
                     "tagId": 25927,
@@ -282,7 +331,7 @@ class TagApi(ModuleApi):
                 {
                     "tagId": 25927,
                     "figureId": 652958,
-                    "value": "v2"
+                    "value": "v2",
                 }
             ]
             response = api.image.tag.add_to_figures(12345, tag_list)
@@ -310,16 +359,14 @@ class TagApi(ModuleApi):
             #        }
             #    ]
         """
-        if type(self) is not TagApi:
-            raise NotImplementedError("This method is not available for classes except TagApi")
-
-        if len(tags_list) == 0:
-            return []
 
         if progress is not None:
             log_progress = False
 
         result = []
+
+        if len(tags_list) == 0:
+            return result
         if log_progress:
             progress = tqdm_sly(
                 desc="Adding tags to figures",
@@ -327,8 +374,164 @@ class TagApi(ModuleApi):
             )
         for batch in batched(tags_list, batch_size):
             data = {ApiField.PROJECT_ID: project_id, ApiField.TAGS: batch}
-            response = self._api.post("figures.tags.bulk.add", data)
+            if type(self) is TagApi:
+                response = self._api.post("figures.tags.bulk.add", data)
+            else:
+                response = self._api.post("annotation-objects.tags.bulk.add", data)
             result.extend(response.json())
             if progress is not None:
                 progress.update(len(batch))
         return result
+
+    def add_to_entities_json(
+        self,
+        project_id: int,
+        tags_list: List[Dict[str, Union[str, int, None]]],
+        batch_size: int = 100,
+        log_progress: bool = False,
+    ) -> List[int]:
+        """
+        Bulk add tags to entities (images, videos, pointclouds, volumes) in a project.
+        Not supported for pointcloud episodes projects.
+        All entities must belong to the same project.
+        The `frameRange` field in a tag object within the tags list is optional and is supported only for video projects.
+
+        The `tags_list` example:
+            [{"tagId": 12345, "entityId": 54321, "value": "tag_value"}, ...].
+        or with frameRange:
+            [{"tagId": 12345, "entityId": 54321, "value": "tag_value", "frameRange": [1, 10]}, ...].
+
+        :param project_id: Project ID in Supervisely.
+        :type project_id: int
+        :param tags_list: List of tag object infos as dictionaries
+                          (e.g. {"tagId": 12345, "entityId": 54321, "value": "tag_value"}).
+        :param batch_size: Number of tags to add in one request.
+        :type batch_size: int
+        :param log_progress: If True, will display a progress bar.
+        :type log_progress: bool
+        :return: List of tags IDs.
+        :rtype: List[int]
+
+        Usage example:
+        .. code-block:: python
+
+            import supervisely as sly
+
+            api = sly.Api(server_address, token)
+
+            tag_list = [
+                {
+                    "tagId": 25926,
+                    "entityId": 652959,
+                    "value": None # optional for tag with type 'None'
+                    "frameRange": [1, 10] # optional (supported only for video projects)
+                },
+                {
+                    "tagId": 25927,
+                    "entityId": 652959,
+                    "value": "v1"
+                },
+                {
+                    "tagId": 25927,
+                    "entityId": 652958,
+                    "value": "v2"
+                }
+            ]
+            api.image.tag.add_to_entities_json(project_id=12345, tag_list=tag_list)
+        """
+
+        result = []
+
+        if len(tags_list) == 0:
+            return result
+
+        if log_progress:
+            ds_progress = tqdm_sly(desc="Adding tags to entities", total=len(tags_list))
+
+        for batch in batched(tags_list, batch_size):
+            data = {ApiField.PROJECT_ID: project_id, ApiField.TAGS: batch}
+            response = self._api.post("tags.entities.bulk.add", data)
+            result.extend([obj[ApiField.ID] for obj in response.json()])
+            if log_progress:
+                ds_progress.update(len(batch))
+
+        return result
+
+    def add_tags_collection_to_objects(
+        self,
+        project_id: int,
+        tags_map: Dict[int, Any],
+        batch_size: int = 100,
+        log_progress: bool = False,
+    ) -> List[Dict[str, Union[str, int, None]]]:
+        """
+        For images project:
+            Add Tags to existing Annotation Figures (labels).
+            The `tags_map` example: {figure_id_1: TagCollection, ...}.
+        For video, pointcloud, volume and pointcloud episodes projects:
+            Add Tags to existing Annotation Objects.
+            The `frameRange` field is optional and is supported only for video and pointcloud episodes projects.
+            The `tags_map` example: {object_id_1: TagCollection, ...}.
+
+        All objects must belong to entities of the same project.
+
+        :param project_id: Project ID in Supervisely.
+        :type project_id: int
+        :param tags_map: Dictionary with mapping figure/object ID to tags collection.
+        :type tags_map: Dict[int, Any]
+        :param batch_size: Number of tags to add in one request.
+        :type batch_size: int
+        :param log_progress: If True, will display a progress bar.
+        :type log_progress: bool
+        :return: List of tags infos as dictionaries.
+        :rtype: List[Dit[str, Union[str, int, None]]]
+
+        Usage example:
+        .. code-block:: python
+
+            import supervisely as sly
+
+            api = sly.Api(server_address, token)
+
+            project_id = 12345
+
+            tag_meta = sly.TagMeta("tag_name", sly.TagValueType.ANY_STRING)
+            meta = sly.ProjectMeta(tag_metas=[tag_meta])
+            meta = sly.ProjectMeta.from_json(api.project.update_meta(project_id, meta))
+            tag_meta = meta.get_tag_meta("tag_name")
+
+            # for images project:
+            tag_map = {
+                652959: sly.TagCollection([sly.Tag(tag_meta, value="v1"), sly.Tag(tag_meta, value="v2"), ...]),
+                652958: sly.TagCollection([sly.Tag(tag_meta, value="v3"), sly.Tag(tag_meta, value="v4"), ...]),
+                ...
+            }
+            api.image.tag.add_tags_to_objects(project_id, tag_map)
+
+            # for videos projects (frameRange is optional):
+            tag_map = {
+                652959: sly.VideoTagCollection([sly.VideoTag(tag_meta, value="v1", frameRange=[1, 10]), ...]),
+                652958: sly.VideoTagCollection([sly.VideoTag(tag_meta, value="v2", frameRange=[4, 12]), ...]),
+                ...
+            }
+            api.video.tag.add_to_objects_json_batch(project_id, tag_map)
+        """
+
+        OBJ_ID_FIELD = ApiField.FIGURE_ID if type(self) is TagApi else ApiField.OBJECT_ID
+
+        data = []
+        for obj_id, tags in tags_map.items():
+            for tag in tags:
+
+                if tag.meta.sly_id is None:
+                    raise ValueError(f"Tag {tag.name} meta has no sly_id")
+
+                data.append(
+                    {
+                        ApiField.TAG_ID: tag.meta.sly_id,
+                        OBJ_ID_FIELD: obj_id,
+                        **tag.to_json()
+                    }
+                )
+
+        return self.add_to_objects(project_id, data, batch_size, log_progress)

supervisely/api/image_api.py

@@ -3601,7 +3601,87 @@ class ImageApi(RemoveableBulkModuleApi):
             if progress_cb is not None:
                 progress_cb(len(batch_ids))
 
-    def update_tag_value(self, tag_id: int, value: Union[str, float]) -> Dict:
+    def add_tags_batch(
+        self,
+        image_ids: List[int],
+        tag_ids: Union[int, List[int]],
+        values: Optional[Union[str, int, List[Union[str, int, None]]]] = None,
+        log_progress: bool = False,
+        batch_size: Optional[int] = 100,
+        tag_metas: Optional[Union[TagMeta, List[TagMeta]]] = None,
+    ) -> List[int]:
+        """
+        Add tag with given ID to Images by IDs with different values.
+
+        :param image_ids: List of Images IDs in Supervisely.
+        :type image_ids: List[int]
+        :param tag_ids: Tag IDs in Supervisely.
+        :type tag_ids: int or List[int]
+        :param values: List of tag values for each image or single value for all images.
+        :type values: List[str] or List[int] or str or int, optional
+        :param log_progress: If True, will log progress.
+        :type log_progress: bool, optional
+        :param batch_size: Batch size
+        :type batch_size: int, optional
+        :param tag_metas: Tag Metas. Needed for values validation, omit to skip validation
+        :type tag_metas: TagMeta or List[TagMeta], optional
+        :return: List of tags IDs.
+        :rtype: List[int]
+        :Usage example:
+
+         .. code-block:: python
+
+            import supervisely as sly
+
+            os.environ['SERVER_ADDRESS'] = 'https://app.supervisely.com'
+            os.environ['API_TOKEN'] = 'Your Supervisely API Token'
+            api = sly.Api.from_env()
+            image_ids = [2389126, 2389127]
+            tag_ids = 277083
+            values = ['value1', 'value2']
+            api.image.add_tags_batch(image_ids, tag_ids, values)
+        """
+        if len(image_ids) == 0:
+            return []
+
+        if isinstance(tag_ids, int):
+            tag_ids = [tag_ids] * len(image_ids)
+
+        if isinstance(tag_metas, TagMeta):
+            tag_metas = [tag_metas] * len(image_ids)
+
+        if values is None:
+            values = [None] * len(image_ids)
+        elif isinstance(values, (str, int)):
+            values = [values] * len(image_ids)
+
+        if len(values) != len(image_ids):
+            raise ValueError("Length of image_ids and values should be the same")
+
+        if len(tag_ids) != len(image_ids):
+            raise ValueError("Length of image_ids and tag_ids should be the same")
+
+        if tag_metas and len(tag_metas) != len(image_ids):
+            raise ValueError("Length of image_ids and tag_metas should be the same")
+
+        if tag_metas:
+            for tag_meta, tag_id, value in zip(tag_metas, tag_ids, values):
+                if not (tag_meta.sly_id == tag_id):
+                    raise ValueError(f"{tag_meta.name = } and {tag_id = } should be same")
+                if not tag_meta.is_valid_value(value):
+                    raise ValueError(f"{tag_meta.name = } can not have value {value = }")
+
+        project_id = self.get_project_id(image_ids[0])
+        data = [
+            {ApiField.ENTITY_ID: image_id, ApiField.TAG_ID: tag_id, ApiField.VALUE: value}
+            for image_id, tag_id, value in zip(image_ids, tag_ids, values)
+        ]
+
+        return self.tag.add_to_entities_json(project_id, data, batch_size, log_progress)
+
+    def update_tag_value(
+        self, tag_id: int, value: Union[str, float]
+    ) -> Dict:
         """
         Update tag value with given ID.
 

supervisely/app/fastapi/index.html

@@ -57,6 +57,26 @@
       border-bottom: 1px solid #dfe2e8;
       position: relative;
     }
+
+    .el-input-number--mini .el-input-number__decrease,
+    .el-input-number--mini .el-input-number__increase {
+      height: 20px !important;
+      width: 20px !important;
+      line-height: 18px !important;
+      font-size: 12px !important;
+      min-width: 20px !important;
+      padding: 0 !important;
+    }
+
+    .el-input-number--mini .el-input-number__decrease {
+      right: 20px !important;
+    }
+
+    .el-input-number--mini .el-input-number__decrease i,
+    .el-input-number--mini .el-input-number__increase i {
+      font-size: 10px !important;
+      line-height: 18px !important;
+    }
   </style>
   <title>{{{app_name}}}</title>
 </head>

supervisely/app/widgets/__init__.py

@@ -151,3 +151,4 @@ from supervisely.app.widgets.experiment_selector.experiment_selector import Expe
 from supervisely.app.widgets.bokeh.bokeh import Bokeh
 from supervisely.app.widgets.run_app_button.run_app_button import RunAppButton
 from supervisely.app.widgets.select_collection.select_collection import SelectCollection
+from supervisely.app.widgets.sampling.sampling import Sampling

supervisely/app/widgets/flexbox/flexbox.py

@@ -1,6 +1,6 @@
-from supervisely.app.widgets import Widget
-from typing import Dict, List
+from typing import Dict, List, Literal
 
+from supervisely.app.widgets import Widget
 
 """
 <div 
@@ -21,10 +21,12 @@ class Flexbox(Widget):
         gap: int = 10,
         center_content: bool = False,
         widget_id: str = None,
+        vertical_alignment: Literal["start", "end", "center", "stretch", "baseline"] = None,
     ):
         self._widgets = widgets
         self._gap = gap
         self._center_content = center_content
+        self._vertical_alignment = vertical_alignment
         super().__init__(widget_id=widget_id, file_path=__file__)
 
     def get_json_data(self) -> Dict:

supervisely/app/widgets/flexbox/template.html

@@ -1,11 +1,7 @@
-<div
-  {% if widget._center_content == true %}
-  style="display: flex; gap: {{{widget._gap}}}px; justify-content: center;"
-  {% else %}
-  style="display: flex; gap: {{{widget._gap}}}px;"
-  {% endif %}
->
+<div {% if widget._center_content==true %}
+  style="display: flex; gap: {{{widget._gap}}}px; align-items: {{{widget._vertical_alignment}}}; justify-content: center;"
+  {% else %} style="display: flex; gap: {{{widget._gap}}}px; align-items: {{{widget._vertical_alignment}}};" {% endif%}>
   {% for w in widget._widgets %}
-    {{{w}}}
+  {{{w}}}
   {% endfor %}
-</div>
+</div>

supervisely/app/widgets/input_number/input_number.py

@@ -20,6 +20,7 @@ class InputNumber(Widget):
         debounce: int = 300,
         precision: int = 0,
         widget_id: str = None,
+        width: int = None,
     ):
         self._value = value
         self._min = min
@@ -29,6 +30,7 @@ class InputNumber(Widget):
         self._controls = controls
         self._debounce = debounce
         self._precision = precision
+        self._width = width
         self._changes_handled = False
 
         super().__init__(widget_id=widget_id, file_path=__file__)

supervisely/app/widgets/input_number/template.html

@@ -16,5 +16,8 @@
   :controls="data.{{{widget.widget_id}}}.controls"
   :debounce="data.{{{widget.widget_id}}}.debounce"
   :precision="data.{{{widget.widget_id}}}.precision"
+  {% if widget._width %}
+  :style="{ width: '{{{widget._width}}}px' }"
+  {% endif %}
 >
 </el-input-number>