label-studio-sdk 1.0.5__py3-none-any.whl → 1.0.7__py3-none-any.whl

label_studio_sdk/comments/client.py

@@ -0,0 +1,512 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from json.decoder import JSONDecodeError
+
+from ..core.api_error import ApiError
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.jsonable_encoder import jsonable_encoder
+from ..core.pydantic_utilities import pydantic_v1
+from ..core.request_options import RequestOptions
+from ..types.comment import Comment
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class CommentsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def list(
+        self,
+        *,
+        project: typing.Optional[int] = None,
+        expand_created_by: typing.Optional[bool] = None,
+        annotation: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.List[Comment]:
+        """
+        Get a list of comments for a specific project.
+
+        Parameters
+        ----------
+        project : typing.Optional[int]
+            Project ID
+
+        expand_created_by : typing.Optional[bool]
+            Expand the created_by field with object instead of ID
+
+        annotation : typing.Optional[int]
+            Annotation ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[Comment]
+
+
+        Examples
+        --------
+        from label_studio_sdk.client import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.comments.list()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/comments/",
+            method="GET",
+            params={"project": project, "expand_created_by": expand_created_by, "annotation": annotation},
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(typing.List[Comment], _response.json())  # type: ignore
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def create(
+        self,
+        *,
+        annotation: typing.Optional[int] = OMIT,
+        project: typing.Optional[int] = OMIT,
+        text: typing.Optional[str] = OMIT,
+        is_resolved: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> Comment:
+        """
+        Create a new comment.
+
+        Parameters
+        ----------
+        annotation : typing.Optional[int]
+
+        project : typing.Optional[int]
+
+        text : typing.Optional[str]
+
+        is_resolved : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Comment
+
+
+        Examples
+        --------
+        from label_studio_sdk.client import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.comments.create()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/comments/",
+            method="POST",
+            json={"annotation": annotation, "project": project, "text": text, "is_resolved": is_resolved},
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(Comment, _response.json())  # type: ignore
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def get(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> Comment:
+        """
+        Get a specific comment.
+
+        Parameters
+        ----------
+        id : int
+            Comment ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Comment
+
+
+        Examples
+        --------
+        from label_studio_sdk.client import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.comments.get(
+            id=1,
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"api/comments/{jsonable_encoder(id)}", method="GET", request_options=request_options
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(Comment, _response.json())  # type: ignore
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def delete(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> None:
+        """
+        Delete a specific comment.
+
+        Parameters
+        ----------
+        id : int
+            Comment ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from label_studio_sdk.client import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.comments.delete(
+            id=1,
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"api/comments/{jsonable_encoder(id)}", method="DELETE", request_options=request_options
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def update(
+        self,
+        id: int,
+        *,
+        annotation: typing.Optional[int] = OMIT,
+        project: typing.Optional[int] = OMIT,
+        text: typing.Optional[str] = OMIT,
+        is_resolved: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> Comment:
+        """
+        Update a specific comment.
+
+        Parameters
+        ----------
+        id : int
+            Comment ID
+
+        annotation : typing.Optional[int]
+
+        project : typing.Optional[int]
+
+        text : typing.Optional[str]
+
+        is_resolved : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Comment
+
+
+        Examples
+        --------
+        from label_studio_sdk.client import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.comments.update(
+            id=1,
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"api/comments/{jsonable_encoder(id)}",
+            method="PATCH",
+            json={"annotation": annotation, "project": project, "text": text, "is_resolved": is_resolved},
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(Comment, _response.json())  # type: ignore
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+
+class AsyncCommentsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def list(
+        self,
+        *,
+        project: typing.Optional[int] = None,
+        expand_created_by: typing.Optional[bool] = None,
+        annotation: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.List[Comment]:
+        """
+        Get a list of comments for a specific project.
+
+        Parameters
+        ----------
+        project : typing.Optional[int]
+            Project ID
+
+        expand_created_by : typing.Optional[bool]
+            Expand the created_by field with object instead of ID
+
+        annotation : typing.Optional[int]
+            Annotation ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[Comment]
+
+
+        Examples
+        --------
+        from label_studio_sdk.client import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        await client.comments.list()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/comments/",
+            method="GET",
+            params={"project": project, "expand_created_by": expand_created_by, "annotation": annotation},
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(typing.List[Comment], _response.json())  # type: ignore
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def create(
+        self,
+        *,
+        annotation: typing.Optional[int] = OMIT,
+        project: typing.Optional[int] = OMIT,
+        text: typing.Optional[str] = OMIT,
+        is_resolved: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> Comment:
+        """
+        Create a new comment.
+
+        Parameters
+        ----------
+        annotation : typing.Optional[int]
+
+        project : typing.Optional[int]
+
+        text : typing.Optional[str]
+
+        is_resolved : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Comment
+
+
+        Examples
+        --------
+        from label_studio_sdk.client import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        await client.comments.create()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/comments/",
+            method="POST",
+            json={"annotation": annotation, "project": project, "text": text, "is_resolved": is_resolved},
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(Comment, _response.json())  # type: ignore
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def get(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> Comment:
+        """
+        Get a specific comment.
+
+        Parameters
+        ----------
+        id : int
+            Comment ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Comment
+
+
+        Examples
+        --------
+        from label_studio_sdk.client import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        await client.comments.get(
+            id=1,
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"api/comments/{jsonable_encoder(id)}", method="GET", request_options=request_options
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(Comment, _response.json())  # type: ignore
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def delete(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> None:
+        """
+        Delete a specific comment.
+
+        Parameters
+        ----------
+        id : int
+            Comment ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from label_studio_sdk.client import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        await client.comments.delete(
+            id=1,
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"api/comments/{jsonable_encoder(id)}", method="DELETE", request_options=request_options
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def update(
+        self,
+        id: int,
+        *,
+        annotation: typing.Optional[int] = OMIT,
+        project: typing.Optional[int] = OMIT,
+        text: typing.Optional[str] = OMIT,
+        is_resolved: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> Comment:
+        """
+        Update a specific comment.
+
+        Parameters
+        ----------
+        id : int
+            Comment ID
+
+        annotation : typing.Optional[int]
+
+        project : typing.Optional[int]
+
+        text : typing.Optional[str]
+
+        is_resolved : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Comment
+
+
+        Examples
+        --------
+        from label_studio_sdk.client import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        await client.comments.update(
+            id=1,
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"api/comments/{jsonable_encoder(id)}",
+            method="PATCH",
+            json={"annotation": annotation, "project": project, "text": text, "is_resolved": is_resolved},
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(Comment, _response.json())  # type: ignore
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)

label_studio_sdk/converter/converter.py

@@ -346,6 +346,8 @@ class Converter(object):
             and "TextArea" in output_tag_types
         ):
             all_formats.remove(Format.ASR_MANIFEST.name)
+        if 'Video' in input_tag_types and 'TimelineLabels' in output_tag_types:
+            all_formats.remove(Format.YOLO_OBB.name)
 
         return all_formats
 

label_studio_sdk/converter/imports/coco.py

@@ -48,10 +48,9 @@ def create_bbox(annotation, categories, from_name, image_height, image_width, to
 
 
 def create_segmentation(
-    annotation, categories, from_name, image_height, image_width, to_name
+    category_id, segmentation, categories, from_name, image_height, image_width, to_name
 ):
-    label = categories[int(annotation["category_id"])]
-    segmentation = annotation["segmentation"][0]
+    label = categories[int(category_id)]
     points = [list(x) for x in zip(*[iter(segmentation)] * 2)]
 
     for i in range(len(points)):
@@ -216,15 +215,17 @@ def convert_coco_to_ls(
             task[out_type][0]["result"].append(item)
 
         if "segmentation" in annotation and len(annotation["segmentation"]):
-            item = create_segmentation(
-                annotation,
-                categories,
-                segmentation_from_name,
-                image_height,
-                image_width,
-                to_name,
-            )
-            task[out_type][0]["result"].append(item)
+            for single_segmentation in annotation["segmentation"]:
+                item = create_segmentation(
+                    annotation["category_id"],
+                    single_segmentation,
+                    categories,
+                    segmentation_from_name,
+                    image_height,
+                    image_width,
+                    to_name,
+                )
+                task[out_type][0]["result"].append(item)
 
         if "keypoints" in annotation:
             items = create_keypoints(
@@ -270,7 +271,7 @@ def add_parser(subparsers):
         "--input",
         dest="input",
         required=True,
-        help="directory with COCO where images, labels, notes.json are located",
+        help="input COCO json file",
         action=ExpandFullPath,
     )
     coco.add_argument(

label_studio_sdk/converter/utils.py

@@ -17,10 +17,11 @@ from urllib.parse import urlparse
 import numpy as np
 import requests
 from PIL import Image
-from label_studio_sdk._extensions.label_studio_tools.core.utils.params import get_env
 from lxml import etree
 from nltk.tokenize.treebank import TreebankWordTokenizer
 
+from label_studio_sdk._extensions.label_studio_tools.core.utils.params import get_env
+
 logger = logging.getLogger(__name__)
 
 _LABEL_TAGS = {"Label", "Choice"}
@@ -422,7 +423,7 @@ def convert_annotation_to_yolo(label):
     return x, y, w, h
 
 
-def convert_annotation_to_yolo_obb(label):
+def convert_annotation_to_yolo_obb(label, normalize=True):
     """
     Convert LS annotation to Yolo OBB format.
 
@@ -435,6 +436,7 @@ def convert_annotation_to_yolo_obb(label):
             - width (float): Width of the object in percentage of the original width.
             - height (float): Height of the object in percentage of the original height.
             - rotation (float, optional): Rotation angle of the object in degrees (default is 0).
+        normalize (bool, optional): Whether to normalize the coordinates to the range [0, 1] (default is True).
 
     Returns:
         list of tuple or None: List of tuples containing the coordinates of the object in Yolo OBB format.
@@ -470,4 +472,71 @@ def convert_annotation_to_yolo_obb(label):
     ]
 
     # Normalize coordinates
-    return [(coord[0] / org_width, coord[1] / org_height) for coord in coords]
+    if normalize:
+        return [(coord[0] / org_width, coord[1] / org_height) for coord in coords]
+    else:
+        return coords
+
+
+def convert_yolo_obb_to_annotation(xyxyxyxy, original_width, original_height):
+    """
+    Convert YOLO Oriented Bounding Box (OBB) format to Label Studio format.
+
+    Args:
+        xyxyxyxy (list): List of 8 float values representing the absolute pixel coordinates
+                         of the OBB in the format [x1, y1, x2, y2, x3, y3, x4, y4].
+        original_width (int): Original width of the image.
+        original_height (int): Original height of the image.
+
+    Returns:
+        dict: Dictionary containing the converted bounding box with the following keys:
+              - x: X-coordinate of the top-left corner of the bounding box in percentage.
+              - y: Y-coordinate of the top-left corner of the bounding box in percentage.
+              - width: Width of the bounding box in percentage.
+              - height: Height of the bounding box in percentage.
+              - rotation: Rotation angle of the bounding box in degrees.
+    """
+    # Reshape the coordinates into a 4x2 matrix
+    coords = np.array(xyxyxyxy, dtype=np.float64).reshape((4, 2))
+
+    # Calculate the center of the bounding box
+    center_x = np.mean(coords[:, 0])
+    center_y = np.mean(coords[:, 1])
+
+    # Calculate the width and height of the bounding box
+    width = np.linalg.norm(coords[0] - coords[1])
+    height = np.linalg.norm(coords[0] - coords[3])
+
+    # Calculate the rotation angle
+    dx = coords[1, 0] - coords[0, 0]
+    dy = coords[1, 1] - coords[0, 1]
+    r = np.degrees(np.arctan2(dy, dx))
+
+    # Find the top-left corner (x, y)
+    top_left_x = (
+        center_x
+        - (width / 2) * np.cos(np.radians(r))
+        + (height / 2) * np.sin(np.radians(r))
+    )
+    top_left_y = (
+        center_y
+        - (width / 2) * np.sin(np.radians(r))
+        - (height / 2) * np.cos(np.radians(r))
+    )
+
+    # Normalize the values
+    x = (top_left_x / original_width) * 100
+    y = (top_left_y / original_height) * 100
+    width = (width / original_width) * 100
+    height = (height / original_height) * 100
+
+    # Create the dictionary for Label Studio
+    return {
+        "x": x,
+        "y": y,
+        "width": width,
+        "height": height,
+        "rotation": r,
+        "original_width": original_width,
+        "original_height": original_height,
+    }

label_studio_sdk/core/client_wrapper.py

@@ -17,7 +17,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "label-studio-sdk",
-            "X-Fern-SDK-Version": "1.0.4",
+            "X-Fern-SDK-Version": "1.0.6",
         }
         headers["Authorization"] = f"Token  {self.api_key}"
         return headers