label-studio-sdk 0.0.32__py3-none-any.whl → 0.0.34__py3-none-any.whl

label_studio_sdk/project.py

@@ -1,22 +1,23 @@
 """ .. include::../docs/project.md
 """
-import os
+
 import json
 import logging
+import os
 import pathlib
 import time
-
 from enum import Enum, auto
-from random import sample, shuffle
-from requests.exceptions import HTTPError, InvalidSchema, MissingSchema
-from requests import Response
 from pathlib import Path
+from random import sample, shuffle
 from typing import Optional, Union, List, Dict, Callable
-from .client import Client
-from .utils import parse_config, chunk
 
-from label_studio_tools.core.utils.io import get_local_path
 from label_studio_tools.core.label_config import parse_config
+from label_studio_tools.core.utils.io import get_local_path
+from requests import Response
+from requests.exceptions import HTTPError, InvalidSchema, MissingSchema
+
+from .client import Client
+from .utils import parse_config, chunk
 
 logger = logging.getLogger(__name__)
 
@@ -32,28 +33,28 @@ class LabelStudioAttributeError(LabelStudioException):
 class ProjectSampling(Enum):
     """Enumerate the available task sampling modes for labeling."""
 
-    RANDOM = 'Uniform sampling'
+    RANDOM = "Uniform sampling"
     """ Uniform random sampling of tasks """
-    SEQUENCE = 'Sequential sampling'
+    SEQUENCE = "Sequential sampling"
     """ Sequential sampling of tasks using task IDs """
-    UNCERTAINTY = 'Uncertainty sampling'
+    UNCERTAINTY = "Uncertainty sampling"
     """ Sample tasks based on prediction scores, such as for active learning (Enterprise only)"""
 
 
 class ProjectStorage(Enum):
     """Enumerate the available types of external source and target storage for labeling projects."""
 
-    GOOGLE = 'gcs'
+    GOOGLE = "gcs"
     """ Google Cloud Storage """
-    S3 = 's3'
+    S3 = "s3"
     """ Amazon S3 Storage """
-    AZURE = 'azure_blob'
+    AZURE = "azure_blob"
     """ Microsoft Azure Blob Storage """
-    LOCAL = 'localfiles'
+    LOCAL = "localfiles"
     """ Label Studio Local File Storage """
-    REDIS = 'redis'
+    REDIS = "redis"
     """ Redis Storage """
-    S3_SECURED = 's3s'
+    S3_SECURED = "s3s"
     """ Amazon S3 Storage secured by IAM roles (Enterprise only) """
 
 
@@ -62,13 +63,13 @@ class AssignmentSamplingMethod(Enum):
 
 
 class ExportSnapshotStatus:
-    CREATED = 'created'
+    CREATED = "created"
     """ Export snapshot is created """
-    IN_PROGRESS = 'in_progress'
+    IN_PROGRESS = "in_progress"
     """ Export snapshot is in progress  """
-    FAILED = 'failed'
+    FAILED = "failed"
     """ Export snapshot failed with errors """
-    COMPLETED = 'completed'
+    COMPLETED = "completed"
     """ Export snapshot was created and can be downloaded """
 
     def __init__(self, response):
@@ -77,30 +78,30 @@ class ExportSnapshotStatus:
     def is_created(self):
         """Export snapshot is created"""
         assert (
-            'status' in self.response
+            "status" in self.response
         ), '"status" field not found in export snapshot status response'
-        return self.response['status'] == self.CREATED
+        return self.response["status"] == self.CREATED
 
     def is_in_progress(self):
         """Export snapshot is in progress"""
         assert (
-            'status' in self.response
+            "status" in self.response
         ), '"status" field not found in export_snapshot_status response'
-        return self.response['status'] == self.IN_PROGRESS
+        return self.response["status"] == self.IN_PROGRESS
 
     def is_failed(self):
         """Export snapshot failed with errors"""
         assert (
-            'status' in self.response
+            "status" in self.response
         ), '"status" field not found in export_snapshot_status response'
-        return self.response['status'] == self.FAILED
+        return self.response["status"] == self.FAILED
 
     def is_completed(self):
         """Export snapshot was created and can be downloaded"""
         assert (
-            'status' in self.response
+            "status" in self.response
         ), '"status" field not found in export_snapshot_status response'
-        return self.response['status'] == self.COMPLETED
+        return self.response["status"] == self.COMPLETED
 
 
 class Project(Client):
@@ -161,10 +162,10 @@ class Project(Client):
             "Use get_users() instead."
         )
 
-        response = self.make_request('GET', f'/api/projects/{self.id}/members')
+        response = self.make_request("GET", f"/api/projects/{self.id}/members")
         users = []
         for user_data in response.json():
-            user_data['client'] = self
+            user_data["client"] = self
             users.append(User(**user_data))
         return users
 
@@ -181,9 +182,9 @@ class Project(Client):
             Dict with created member
 
         """
-        payload = {'user': user.id}
+        payload = {"user": user.id}
         response = self.make_request(
-            'POST', f'/api/projects/{self.id}/members', json=payload
+            "POST", f"/api/projects/{self.id}/members", json=payload
         )
         return response.json()
 
@@ -201,20 +202,20 @@ class Project(Client):
             Dict with counter of created assignments
 
         """
-        final_response = {'assignments': 0}
+        final_response = {"assignments": 0}
         users_ids = [user.id for user in users]
         # Assign tasks to users with batches
         for c in chunk(tasks_ids, 1000):
             logger.debug(f"Starting assignment for: {users_ids}")
             payload = {
-                'users': users_ids,
-                'selectedItems': {'all': False, 'included': c},
-                'type': 'AN',
+                "users": users_ids,
+                "selectedItems": {"all": False, "included": c},
+                "type": "AN",
             }
             response = self.make_request(
-                'POST', f'/api/projects/{self.id}/tasks/assignees', json=payload
+                "POST", f"/api/projects/{self.id}/tasks/assignees", json=payload
             )
-            final_response['assignments'] += response.json()['assignments']
+            final_response["assignments"] += response.json()["assignments"]
         return final_response
 
     def delete_annotators_assignment(self, tasks_ids):
@@ -230,10 +231,10 @@ class Project(Client):
             Dict with counter of deleted annotator assignments
 
         """
-        payload = {'selectedItems': {'all': False, 'included': tasks_ids}}
+        payload = {"selectedItems": {"all": False, "included": tasks_ids}}
         response = self.make_request(
-            'POST',
-            f'/api/dm/actions?id=delete_annotators&project={self.id}',
+            "POST",
+            f"/api/dm/actions?id=delete_annotators&project={self.id}",
             json=payload,
         )
         return response.json()
@@ -251,10 +252,10 @@ class Project(Client):
             Dict with counter of deleted reviewer assignments
 
         """
-        payload = {'selectedItems': {'all': False, 'included': tasks_ids}}
+        payload = {"selectedItems": {"all": False, "included": tasks_ids}}
         response = self.make_request(
-            'POST',
-            f'/api/dm/actions?id=delete_reviewers&project={self.id}',
+            "POST",
+            f"/api/dm/actions?id=delete_reviewers&project={self.id}",
             json=payload,
         )
         return response.json()
@@ -274,12 +275,12 @@ class Project(Client):
 
         """
         payload = {
-            'users': [user.id for user in users],
-            'selectedItems': {'all': False, 'included': tasks_ids},
-            'type': 'RE',
+            "users": [user.id for user in users],
+            "selectedItems": {"all": False, "included": tasks_ids},
+            "type": "RE",
         }
         response = self.make_request(
-            'POST', f'/api/projects/{self.id}/tasks/assignees', json=payload
+            "POST", f"/api/projects/{self.id}/tasks/assignees", json=payload
         )
         return response.json()
 
@@ -354,7 +355,7 @@ class Project(Client):
             Retrieve and display predictions when loading a task
 
         """
-        response = self.make_request('GET', f'/api/projects/{self.id}')
+        response = self.make_request("GET", f"/api/projects/{self.id}")
         return response.json()
 
     def get_model_versions(self):
@@ -366,7 +367,7 @@ class Project(Client):
              Model versions
 
         """
-        response = self.make_request('GET', f'/api/projects/{self.id}/model-versions')
+        response = self.make_request("GET", f"/api/projects/{self.id}/model-versions")
         return response.json()
 
     def update_params(self):
@@ -434,11 +435,11 @@ class Project(Client):
         Raises LabelStudioException in case of errors.
 
         """
-        response = self.make_request('POST', '/api/projects', json=kwargs)
+        response = self.make_request("POST", "/api/projects", json=kwargs)
         if response.status_code == 201:
             self.params = response.json()
         else:
-            raise LabelStudioException('Project not created')
+            raise LabelStudioException("Project not created")
 
     @classmethod
     def _create_from_id(cls, client, project_id, params=None):
@@ -453,7 +454,7 @@ class Project(Client):
         if params and isinstance(params, dict):
             # TODO: validate project parameters
             project.params = params
-        project.params['id'] = project_id
+        project.params["id"] = project_id
         return project
 
     @classmethod
@@ -494,13 +495,13 @@ class Project(Client):
             Imported task IDs
 
         """
-        params = {'return_task_ids': '1'}
+        params = {"return_task_ids": "1"}
         if preannotated_from_fields:
-            params['preannotated_from_fields'] = ','.join(preannotated_from_fields)
+            params["preannotated_from_fields"] = ",".join(preannotated_from_fields)
         if isinstance(tasks, (list, dict)):
             response = self.make_request(
-                method='POST',
-                url=f'/api/projects/{self.id}/import',
+                method="POST",
+                url=f"/api/projects/{self.id}/import",
                 json=tasks,
                 params=params,
                 timeout=(10, 600),
@@ -508,12 +509,12 @@ class Project(Client):
         elif isinstance(tasks, (str, Path)):
             # try import from file
             if not os.path.isfile(tasks):
-                raise LabelStudioException(f'Not found import tasks file {tasks}')
-            with open(tasks, mode='rb') as f:
+                raise LabelStudioException(f"Not found import tasks file {tasks}")
+            with open(tasks, mode="rb") as f:
                 response = self.make_request(
-                    method='POST',
-                    url=f'/api/projects/{self.id}/import',
-                    files={'file': f},
+                    method="POST",
+                    url=f"/api/projects/{self.id}/import",
+                    files={"file": f},
                     params=params,
                     timeout=(10, 600),
                 )
@@ -523,7 +524,7 @@ class Project(Client):
             )
         response = response.json()
 
-        if 'import' in response:
+        if "import" in response:
             # check import status
             timeout = 300
             fibonacci_backoff = [1, 1]
@@ -532,18 +533,18 @@ class Project(Client):
 
             while True:
                 import_status = self.make_request(
-                    method='GET',
+                    method="GET",
                     url=f'/api/projects/{self.id}/imports/{response["import"]}',
                 ).json()
 
-                if import_status['status'] == 'completed':
-                    return import_status['task_ids']
+                if import_status["status"] == "completed":
+                    return import_status["task_ids"]
 
-                if import_status['status'] == 'failed':
-                    raise LabelStudioException(import_status['error'])
+                if import_status["status"] == "failed":
+                    raise LabelStudioException(import_status["error"])
 
                 if time.time() - start_time >= timeout:
-                    raise LabelStudioException('Import timeout')
+                    raise LabelStudioException("Import timeout")
 
                 time.sleep(fibonacci_backoff[0])
                 fibonacci_backoff = [
@@ -551,11 +552,11 @@ class Project(Client):
                     fibonacci_backoff[0] + fibonacci_backoff[1],
                 ]
 
-        return response['task_ids']
+        return response["task_ids"]
 
     def export_tasks(
         self,
-        export_type: str = 'JSON',
+        export_type: str = "JSON",
         download_all_tasks: bool = False,
         download_resources: bool = False,
         ids: Optional[List[int]] = None,
@@ -592,19 +593,19 @@ class Project(Client):
 
         """
         params = {
-            'exportType': export_type,
-            'download_all_tasks': download_all_tasks,
-            'download_resources': download_resources,
+            "exportType": export_type,
+            "download_all_tasks": download_all_tasks,
+            "download_resources": download_resources,
         }
         if ids:
-            params['ids'] = ids
+            params["ids"] = ids
         response = self.make_request(
-            method='GET', url=f'/api/projects/{self.id}/export', params=params
+            method="GET", url=f"/api/projects/{self.id}/export", params=params
         )
         if export_location is None:
-            if 'JSON' not in export_type.upper():
+            if "JSON" not in export_type.upper():
                 raise ValueError(
-                    f'{export_type} export type requires an export location to be specified'
+                    f"{export_type} export type requires an export location to be specified"
                 )
             return response.json()
 
@@ -623,7 +624,7 @@ class Project(Client):
 
     def set_params(self, **kwargs):
         """Low level function to set project parameters."""
-        response = self.make_request('PATCH', f'/api/projects/{self.id}', json=kwargs)
+        response = self.make_request("PATCH", f"/api/projects/{self.id}", json=kwargs)
         assert response.status_code == 200
 
     def set_sampling(self, sampling: ProjectSampling):
@@ -705,7 +706,7 @@ class Project(Client):
         page = 1
         result = []
         data = {}
-        while not data.get('end_pagination'):
+        while not data.get("end_pagination"):
             try:
                 data = self.get_paginated_tasks(
                     filters=filters,
@@ -716,10 +717,10 @@ class Project(Client):
                     page=page,
                     page_size=100,
                 )
-                result += data['tasks']
+                result += data["tasks"]
                 page += 1
             except LabelStudioException as e:
-                logger.debug(f'Error during pagination: {e}')
+                logger.debug(f"Error during pagination: {e}")
                 break
         return result
 
@@ -730,7 +731,7 @@ class Project(Client):
         view_id=None,
         selected_ids=None,
         page: int = 1,
-        page_size: int = -1,
+        page_size: int = 100,
         only_ids: bool = False,
         resolve_uri: bool = True,
     ):
@@ -768,7 +769,7 @@ class Project(Client):
         page: int
             Page. Default is 1.
         page_size: int
-            Page size. Default is -1, to retrieve all tasks in the project.
+            Page size. Default is 100, to retrieve all tasks in the project you can use get_tasks().
         only_ids: bool
             If true, return only task IDs
         resolve_uri: bool
@@ -798,54 +799,56 @@ class Project(Client):
 
         """
         query = {
-            'filters': filters,
-            'ordering': ordering or [],
-            'selectedItems': {'all': False, 'included': selected_ids}
-            if selected_ids
-            else {'all': True, "excluded": []},
+            "filters": filters,
+            "ordering": ordering or [],
+            "selectedItems": (
+                {"all": False, "included": selected_ids}
+                if selected_ids
+                else {"all": True, "excluded": []}
+            ),
         }
         params = {
-            'project': self.id,
-            'page': page,
-            'page_size': page_size,
-            'view': view_id,
-            'query': json.dumps(query),
-            'fields': 'all',
-            'resolve_uri': resolve_uri,
+            "project": self.id,
+            "page": page,
+            "page_size": page_size,
+            "view": view_id,
+            "query": json.dumps(query),
+            "fields": "all",
+            "resolve_uri": resolve_uri,
         }
         if only_ids:
-            params['include'] = 'id'
+            params["include"] = "id"
 
         response = self.make_request(
-            'GET', '/api/tasks', params, raise_exceptions=False
+            "GET", "/api/tasks", params, raise_exceptions=False
         )
         # we'll get 404 from API on empty page
         if response.status_code == 404:
-            return {'tasks': [], 'end_pagination': True}
+            return {"tasks": [], "end_pagination": True}
         elif response.status_code != 200:
             self.log_response_error(response)
             try:
                 response.raise_for_status()
             except HTTPError as e:
-                raise LabelStudioException(f'Error loading tasks: {e}')
+                raise LabelStudioException(f"Error loading tasks: {e}")
 
         data = response.json()
-        tasks = data['tasks']
+        tasks = data["tasks"]
         if only_ids:
-            data['tasks'] = [task['id'] for task in tasks]
+            data["tasks"] = [task["id"] for task in tasks]
 
         return data
 
     def get_tasks_ids(self, *args, **kwargs):
         """Same as `label_studio_sdk.project.Project.get_tasks()` but returns only task IDs."""
-        kwargs['only_ids'] = True
+        kwargs["only_ids"] = True
         return self.get_tasks(*args, **kwargs)
 
     def get_paginated_tasks_ids(self, *args, **kwargs):
         """Same as `label_studio_sdk.project.Project.get_paginated_tasks()` but returns
         only task IDs.
         """
-        kwargs['only_ids'] = True
+        kwargs["only_ids"] = True
         return self.get_paginated_tasks(*args, **kwargs)
 
     def get_views(self):
@@ -866,10 +869,10 @@ class Project(Client):
         data: dict
             Filters, orderings and other visual settings
         """
-        response = self.make_request('GET', f'/api/dm/views?project={self.id}')
+        response = self.make_request("GET", f"/api/dm/views?project={self.id}")
         return response.json()
 
-    def create_view(self, filters, ordering=None, title='Tasks'):
+    def create_view(self, filters, ordering=None, title="Tasks"):
         """Create view
 
         Parameters
@@ -890,12 +893,29 @@ class Project(Client):
 
         """
         data = {
-            'project': self.id,
-            'data': {'title': title, 'ordering': ordering, 'filters': filters},
+            "project": self.id,
+            "data": {"title": title, "ordering": ordering, "filters": filters},
         }
-        response = self.make_request('POST', '/api/dm/views', json=data)
+        response = self.make_request("POST", "/api/dm/views", json=data)
         return response.json()
 
+    def delete_view(self, view_id):
+        """Delete view
+
+        Parameters
+        ----------
+        view_id: int
+            View ID
+
+        Returns
+        -------
+        dict:
+            dict with deleted view
+
+        """
+        response = self.make_request("DELETE", f"/api/dm/views/{view_id}")
+        return
+
     @property
     def tasks(self):
         """Retrieve all tasks from the project. This call can be very slow if the project has a lot of tasks."""
@@ -922,13 +942,13 @@ class Project(Client):
         """
         return self.get_tasks(
             filters={
-                'conjunction': 'and',
-                'items': [
+                "conjunction": "and",
+                "items": [
                     {
-                        'filter': 'filter:tasks:completed_at',
-                        'operator': 'empty',
-                        'value': False,
-                        'type': 'Datetime',
+                        "filter": "filter:tasks:completed_at",
+                        "operator": "empty",
+                        "value": False,
+                        "type": "Datetime",
                     }
                 ],
             },
@@ -963,13 +983,13 @@ class Project(Client):
         """
         return self.get_tasks(
             filters={
-                'conjunction': 'and',
-                'items': [
+                "conjunction": "and",
+                "items": [
                     {
-                        'filter': 'filter:tasks:completed_at',
-                        'operator': 'empty',
-                        'value': True,
-                        'type': 'Datetime',
+                        "filter": "filter:tasks:completed_at",
+                        "operator": "empty",
+                        "value": True,
+                        "type": "Datetime",
                     }
                 ],
             },
@@ -1029,7 +1049,7 @@ class Project(Client):
             Uploaded file used as data source for this task
         ```
         """
-        response = self.make_request('GET', f'/api/tasks/{task_id}')
+        response = self.make_request("GET", f"/api/tasks/{task_id}")
         return response.json()
 
     def update_task(self, task_id, **kwargs):
@@ -1048,7 +1068,7 @@ class Project(Client):
             Dict with updated task
 
         """
-        response = self.make_request('PATCH', f'/api/tasks/{task_id}', json=kwargs)
+        response = self.make_request("PATCH", f"/api/tasks/{task_id}", json=kwargs)
         response.raise_for_status()
         return response.json()
 
@@ -1102,11 +1122,13 @@ class Project(Client):
         model_version: str
             Any string identifying your model
         """
-        data = {'task': task_id, 'result': result, 'score': score}
+        data = {"task": task_id, "result": result, "score": score}
         if model_version is not None:
-            data['model_version'] = model_version
-        response = self.make_request('POST', '/api/predictions', json=data)
-        return response.json()
+            data["model_version"] = model_version
+        response = self.make_request("POST", "/api/predictions", json=data)
+        json = response.json()
+        logger.debug(f"Response: {json}")
+        return json
 
     def create_predictions(self, predictions):
         """Bulk create predictions for tasks. See <a href="https://labelstud.io/guide/predictions.html">more
@@ -1119,7 +1141,7 @@ class Project(Client):
             Label Studio JSON format as for annotations</a>.
         """
         response = self.make_request(
-            'POST', f'/api/projects/{self.id}/import/predictions', json=predictions
+            "POST", f"/api/projects/{self.id}/import/predictions", json=predictions
         )
         return response.json()
 
@@ -1138,20 +1160,37 @@ class Project(Client):
 
         """
         payload = {
-            'filters': {'conjunction': 'and', 'items': []},
-            'model_version': model_versions,
-            'ordering': [],
-            'project': self.id,
-            'selectedItems': {'all': True, 'excluded': []},
+            "filters": {"conjunction": "and", "items": []},
+            "model_version": model_versions,
+            "ordering": [],
+            "project": self.id,
+            "selectedItems": {"all": True, "excluded": []},
         }
         response = self.make_request(
-            'POST',
-            '/api/dm/actions',
-            params={'id': 'predictions_to_annotations', 'project': self.id},
+            "POST",
+            "/api/dm/actions",
+            params={"id": "predictions_to_annotations", "project": self.id},
             json=payload,
         )
         return response.json()
 
+    def list_annotations(self, task_id: int) -> List:
+        """List all annotations for a task.
+
+        Parameters
+        ----------
+        task_id: int
+            Task ID
+
+        Returns
+        -------
+        list of dict:
+            List of annotations objects
+        """
+        response = self.make_request("GET", f"/api/tasks/{task_id}/annotations")
+        response.raise_for_status()
+        return response.json()
+
     def create_annotation(self, task_id: int, **kwargs) -> Dict:
         """Add annotations to a task like an annotator does.
 
@@ -1170,11 +1209,28 @@ class Project(Client):
 
         """
         response = self.make_request(
-            'POST', f'/api/tasks/{task_id}/annotations/', json=kwargs
+            "POST", f"/api/tasks/{task_id}/annotations/", json=kwargs
         )
         response.raise_for_status()
         return response.json()
 
+    def get_annotation(self, annotation_id: int) -> dict:
+        """Retrieve a specific annotation for a task using the annotation ID.
+
+        Parameters
+        ----------
+        annotation_id: int
+            A unique integer value identifying this annotation.
+
+        Returns
+        ----------
+        dict
+            Retreived annotation object
+        """
+        response = self.make_request("GET", f"/api/annotations/{annotation_id}")
+        response.raise_for_status()
+        return response.json()
+
     def update_annotation(self, annotation_id, **kwargs):
         """Update specific annotation with new annotation parameters, e.g.
             ```
@@ -1195,11 +1251,29 @@ class Project(Client):
 
         """
         response = self.make_request(
-            'PATCH', f'/api/annotations/{annotation_id}', json=kwargs
+            "PATCH", f"/api/annotations/{annotation_id}", json=kwargs
         )
         response.raise_for_status()
         return response.json()
 
+    def delete_annotation(self, annotation_id: int) -> int:
+        """Delete an annotation using the annotation ID. This action can't be undone!
+
+        Parameters
+        ----------
+        annotation_id: int
+            A unique integer value identifying this annotation.
+
+        Returns
+        ----------
+        int
+            Status code for operation
+
+        """
+        response = self.make_request("DELETE", f"/api/annotations/{annotation_id}")
+        response.raise_for_status()
+        return response.status_code
+
     def get_predictions_coverage(self):
         """Prediction coverage stats for all model versions for the project.
 
@@ -1218,7 +1292,7 @@ class Project(Client):
         """
         model_versions = self.get_model_versions()
         params = self.get_params()
-        tasks_number = params['task_number']
+        tasks_number = params["task_number"]
         coverage = {
             model_version: count / tasks_number
             for model_version, count in model_versions.items()
@@ -1240,8 +1314,8 @@ class Project(Client):
         google_application_credentials: Optional[str] = None,
         presign: Optional[bool] = True,
         presign_ttl: Optional[int] = 1,
-        title: Optional[str] = '',
-        description: Optional[str] = '',
+        title: Optional[str] = "",
+        description: Optional[str] = "",
     ):
         """Connect a Google Cloud Storage (GCS) bucket to Label Studio to use as source storage and import tasks.
 
@@ -1283,23 +1357,25 @@ class Project(Client):
             Number of tasks synced in the last sync
 
         """
-        if os.path.isfile(google_application_credentials):
+        if google_application_credentials and os.path.isfile(
+            google_application_credentials
+        ):
             with open(google_application_credentials) as f:
                 google_application_credentials = f.read()
 
         payload = {
-            'bucket': bucket,
-            'project': self.id,
-            'prefix': prefix,
-            'regex_filter': regex_filter,
-            'use_blob_urls': use_blob_urls,
-            'google_application_credentials': google_application_credentials,
-            'presign': presign,
-            'presign_ttl': presign_ttl,
-            'title': title,
-            'description': description,
+            "bucket": bucket,
+            "project": self.id,
+            "prefix": prefix,
+            "regex_filter": regex_filter,
+            "use_blob_urls": use_blob_urls,
+            "google_application_credentials": google_application_credentials,
+            "presign": presign,
+            "presign_ttl": presign_ttl,
+            "title": title,
+            "description": description,
         }
-        response = self.make_request('POST', '/api/storages/gcs', json=payload)
+        response = self.make_request("POST", "/api/storages/gcs", json=payload)
         return response.json()
 
     def connect_google_export_storage(
@@ -1307,8 +1383,8 @@ class Project(Client):
         bucket: str,
         prefix: Optional[str] = None,
         google_application_credentials: Optional[str] = None,
-        title: Optional[str] = '',
-        description: Optional[str] = '',
+        title: Optional[str] = "",
+        description: Optional[str] = "",
         can_delete_objects: bool = False,
     ):
         """Connect a Google Cloud Storage (GCS) bucket to Label Studio to use as target storage and export tasks.
@@ -1350,15 +1426,15 @@ class Project(Client):
                 google_application_credentials = f.read()
 
         payload = {
-            'bucket': bucket,
-            'prefix': prefix,
-            'google_application_credentials': google_application_credentials,
-            'title': title,
-            'description': description,
-            'can_delete_objects': can_delete_objects,
-            'project': self.id,
+            "bucket": bucket,
+            "prefix": prefix,
+            "google_application_credentials": google_application_credentials,
+            "title": title,
+            "description": description,
+            "can_delete_objects": can_delete_objects,
+            "project": self.id,
         }
-        response = self.make_request('POST', '/api/storages/export/gcs', json=payload)
+        response = self.make_request("POST", "/api/storages/export/gcs", json=payload)
         return response.json()
 
     def connect_s3_import_storage(
@@ -1369,13 +1445,14 @@ class Project(Client):
         use_blob_urls: Optional[bool] = True,
         presign: Optional[bool] = True,
         presign_ttl: Optional[int] = 1,
-        title: Optional[str] = '',
-        description: Optional[str] = '',
+        title: Optional[str] = "",
+        description: Optional[str] = "",
         aws_access_key_id: Optional[str] = None,
         aws_secret_access_key: Optional[str] = None,
         aws_session_token: Optional[str] = None,
         region_name: Optional[str] = None,
         s3_endpoint: Optional[str] = None,
+        recursive_scan: Optional[bool] = False,
     ):
         """Connect an Amazon S3 bucket to Label Studio to use as source storage and import tasks.
 
@@ -1407,6 +1484,8 @@ class Project(Client):
             Optional, specify the AWS region of your S3 bucket.
         s3_endpoint: string
             Optional, specify an S3 endpoint URL to use to access your bucket instead of the standard access method.
+        recursive_scan: bool
+            Optional, specify whether to perform recursive scan over the bucket content.
 
         Returns
         -------
@@ -1425,30 +1504,113 @@ class Project(Client):
             Number of tasks synced in the last sync
         """
         payload = {
-            'bucket': bucket,
-            'prefix': prefix,
-            'regex_filter': regex_filter,
-            'use_blob_urls': use_blob_urls,
-            'aws_access_key_id': aws_access_key_id,
-            'aws_secret_access_key': aws_secret_access_key,
-            'aws_session_token': aws_session_token,
-            'region_name': region_name,
-            's3_endpoint': s3_endpoint,
-            'presign': presign,
-            'presign_ttl': presign_ttl,
-            'title': title,
-            'description': description,
-            'project': self.id,
+            "bucket": bucket,
+            "prefix": prefix,
+            "regex_filter": regex_filter,
+            "use_blob_urls": use_blob_urls,
+            "aws_access_key_id": aws_access_key_id,
+            "aws_secret_access_key": aws_secret_access_key,
+            "aws_session_token": aws_session_token,
+            "region_name": region_name,
+            "s3_endpoint": s3_endpoint,
+            "presign": presign,
+            "presign_ttl": presign_ttl,
+            "title": title,
+            "description": description,
+            "project": self.id,
+            "recursive_scan": recursive_scan,
+        }
+        response = self.make_request("POST", "/api/storages/s3", json=payload)
+        return response.json()
+
+    def connect_s3s_iam_import_storage(
+        self,
+        role_arn: str,
+        external_id: Optional[str] = None,
+        bucket: Optional[str] = None,
+        prefix: Optional[str] = None,
+        regex_filter: Optional[str] = None,
+        use_blob_urls: Optional[bool] = True,
+        presign: Optional[bool] = True,
+        presign_ttl: Optional[int] = 1,
+        title: Optional[str] = "",
+        description: Optional[str] = "",
+        region_name: Optional[str] = None,
+        s3_endpoint: Optional[str] = None,
+        recursive_scan: Optional[bool] = False,
+        aws_sse_kms_key_id: Optional[str] = None,
+    ):
+        """Create S3 secured import storage with IAM role access. Enterprise only.
+
+        Parameters
+        ----------
+        role_arn: string
+            Required, specify the AWS Role ARN to assume.
+        external_id: string or None
+            Optional, specify the external ID to use to assume the role. If None, SDK will call api/organizations/<id>
+            and use external_id from the response. You can find this ID on the organization page in the Label Studio UI.
+        bucket: string
+            Specify the name of the S3 bucket.
+        prefix: string
+            Optional, specify the prefix within the S3 bucket to import your data from.
+        regex_filter: string
+            Optional, specify a regex filter to use to match the file types of your data.
+        use_blob_urls: bool
+            Optional, true by default. Specify whether your data is raw image or video data, or JSON tasks.
+        presign: bool
+            Optional, true by default. Specify whether or not to create presigned URLs.
+        presign_ttl: int
+            Optional, 1 by default. Specify how long to keep presigned URLs active.
+        title: string
+            Optional, specify a title for your S3 import storage that appears in Label Studio.
+        description: string
+            Optional, specify a description for your S3 import storage.
+        region_name: string
+            Optional, specify the AWS region of your S3 bucket.
+        s3_endpoint: string
+            Optional, specify an S3 endpoint URL to use to access your bucket instead of the standard access method.
+        recursive_scan: bool
+            Optional, specify whether to perform recursive scan over the bucket content.
+        aws_sse_kms_key_id: string
+            Optional, specify an AWS SSE KMS Key ID for server-side encryption.
+        synchronizable, last_sync, last_sync_count, last_sync_job, status, traceback, meta:
+            Parameters for synchronization details and storage status.
+
+        Returns
+        -------
+        dict:
+            containing the response from the API including storage ID and type, among other details.
+        """
+        if external_id is None:
+            organization = self.get_organization()
+            external_id = organization["external_id"]
+
+        payload = {
+            "bucket": bucket,
+            "prefix": prefix,
+            "regex_filter": regex_filter,
+            "use_blob_urls": use_blob_urls,
+            "presign": presign,
+            "presign_ttl": presign_ttl,
+            "title": title,
+            "description": description,
+            "recursive_scan": recursive_scan,
+            "role_arn": role_arn,
+            "region_name": region_name,
+            "s3_endpoint": s3_endpoint,
+            "aws_sse_kms_key_id": aws_sse_kms_key_id,
+            "project": self.id,
+            "external_id": external_id,
         }
-        response = self.make_request('POST', '/api/storages/s3', json=payload)
+        response = self.make_request("POST", "/api/storages/s3s/", json=payload)
         return response.json()
 
     def connect_s3_export_storage(
         self,
         bucket: str,
         prefix: Optional[str] = None,
-        title: Optional[str] = '',
-        description: Optional[str] = '',
+        title: Optional[str] = "",
+        description: Optional[str] = "",
         aws_access_key_id: Optional[str] = None,
         aws_secret_access_key: Optional[str] = None,
         aws_session_token: Optional[str] = None,
@@ -1499,19 +1661,19 @@ class Project(Client):
         """
 
         payload = {
-            'bucket': bucket,
-            'prefix': prefix,
-            'aws_access_key_id': aws_access_key_id,
-            'aws_secret_access_key': aws_secret_access_key,
-            'aws_session_token': aws_session_token,
-            'region_name': region_name,
-            's3_endpoint': s3_endpoint,
-            'title': title,
-            'description': description,
-            'can_delete_objects': can_delete_objects,
-            'project': self.id,
+            "bucket": bucket,
+            "prefix": prefix,
+            "aws_access_key_id": aws_access_key_id,
+            "aws_secret_access_key": aws_secret_access_key,
+            "aws_session_token": aws_session_token,
+            "region_name": region_name,
+            "s3_endpoint": s3_endpoint,
+            "title": title,
+            "description": description,
+            "can_delete_objects": can_delete_objects,
+            "project": self.id,
         }
-        response = self.make_request('POST', '/api/storages/export/s3', json=payload)
+        response = self.make_request("POST", "/api/storages/export/s3", json=payload)
         return response.json()
 
     def connect_azure_import_storage(
@@ -1522,8 +1684,8 @@ class Project(Client):
         use_blob_urls: Optional[bool] = True,
         presign: Optional[bool] = True,
         presign_ttl: Optional[int] = 1,
-        title: Optional[str] = '',
-        description: Optional[str] = '',
+        title: Optional[str] = "",
+        description: Optional[str] = "",
         account_name: Optional[str] = None,
         account_key: Optional[str] = None,
     ):
@@ -1569,27 +1731,27 @@ class Project(Client):
             Number of tasks synced in the last sync
         """
         payload = {
-            'container': container,
-            'prefix': prefix,
-            'regex_filter': regex_filter,
-            'use_blob_urls': use_blob_urls,
-            'account_name': account_name,
-            'account_key': account_key,
-            'presign': presign,
-            'presign_ttl': presign_ttl,
-            'title': title,
-            'description': description,
-            'project': self.id,
+            "container": container,
+            "prefix": prefix,
+            "regex_filter": regex_filter,
+            "use_blob_urls": use_blob_urls,
+            "account_name": account_name,
+            "account_key": account_key,
+            "presign": presign,
+            "presign_ttl": presign_ttl,
+            "title": title,
+            "description": description,
+            "project": self.id,
         }
-        response = self.make_request('POST', '/api/storages/azure', json=payload)
+        response = self.make_request("POST", "/api/storages/azure", json=payload)
         return response.json()
 
     def connect_azure_export_storage(
         self,
         container: str,
         prefix: Optional[str] = None,
-        title: Optional[str] = '',
-        description: Optional[str] = '',
+        title: Optional[str] = "",
+        description: Optional[str] = "",
         account_name: Optional[str] = None,
         account_key: Optional[str] = None,
         can_delete_objects: bool = False,
@@ -1630,16 +1792,16 @@ class Project(Client):
             Number of tasks synced in the last sync
         """
         payload = {
-            'container': container,
-            'prefix': prefix,
-            'account_name': account_name,
-            'account_key': account_key,
-            'title': title,
-            'description': description,
-            'can_delete_objects': can_delete_objects,
-            'project': self.id,
+            "container": container,
+            "prefix": prefix,
+            "account_name": account_name,
+            "account_key": account_key,
+            "title": title,
+            "description": description,
+            "can_delete_objects": can_delete_objects,
+            "project": self.id,
         }
-        response = self.make_request('POST', '/api/storages/export/azure', json=payload)
+        response = self.make_request("POST", "/api/storages/export/azure", json=payload)
         return response.json()
 
     def connect_local_import_storage(
@@ -1647,8 +1809,8 @@ class Project(Client):
         local_store_path: [str],
         regex_filter: Optional[str] = None,
         use_blob_urls: Optional[bool] = True,
-        title: Optional[str] = '',
-        description: Optional[str] = '',
+        title: Optional[str] = "",
+        description: Optional[str] = "",
     ):
         """Connect a Local storage to Label Studio to use as source storage and import tasks.
         Parameters
@@ -1678,37 +1840,233 @@ class Project(Client):
         last_sync_count: int
             Number of tasks synced in the last sync
         """
-        if 'LABEL_STUDIO_LOCAL_FILES_DOCUMENT_ROOT' not in os.environ:
+        if "LABEL_STUDIO_LOCAL_FILES_DOCUMENT_ROOT" not in os.environ:
             raise ValueError(
-                'To use connect_local_import_storage() you should set '
-                'LABEL_STUDIO_LOCAL_FILES_DOCUMENT_ROOT environment variable, '
-                'read more: https://labelstud.io/guide/storage.html#Prerequisites-2'
+                "To use connect_local_import_storage() you should set "
+                "LABEL_STUDIO_LOCAL_FILES_DOCUMENT_ROOT environment variable, "
+                "read more: https://labelstud.io/guide/storage.html#Prerequisites-2"
             )
-        root = os.environ['LABEL_STUDIO_LOCAL_FILES_DOCUMENT_ROOT']
+        root = os.environ["LABEL_STUDIO_LOCAL_FILES_DOCUMENT_ROOT"]
 
         if not os.path.isdir(local_store_path):
-            raise ValueError(f'{local_store_path} is not a directory')
+            raise ValueError(f"{local_store_path} is not a directory")
         if (Path(root) in Path(local_store_path).parents) is False:
             raise ValueError(
-                f'{str(Path(root))} is not presented in local_store_path parents: '
-                f'{str(Path(local_store_path).parents)}'
+                f"{str(Path(root))} is not presented in local_store_path parents: "
+                f"{str(Path(local_store_path).parents)}"
             )
 
         payload = {
-            'regex_filter': regex_filter,
-            'use_blob_urls': use_blob_urls,
-            'path': local_store_path,
-            'presign': False,
-            'presign_ttl': 1,
-            'title': title,
-            'description': description,
-            'project': self.id,
+            "regex_filter": regex_filter,
+            "use_blob_urls": use_blob_urls,
+            "path": local_store_path,
+            "presign": False,
+            "presign_ttl": 1,
+            "title": title,
+            "description": description,
+            "project": self.id,
         }
         response = self.make_request(
-            'POST', f'/api/storages/localfiles?project={self.id}', json=payload
+            "POST", f"/api/storages/localfiles?project={self.id}", json=payload
+        )
+        return response.json()
+
+    def sync_import_storage(self, storage_type, storage_id):
+        """Synchronize Import (Source) Cloud Storage.
+
+        Parameters
+        ----------
+        storage_type: string
+            Specify the type of the storage container. See ProjectStorage for available types.
+        storage_id: int
+            Specify the storage ID of the storage container. See get_import_storages() to get ids.
+
+        Returns
+        -------
+        dict:
+            containing the same fields as in the original storage request and:
+
+        id: int
+            Storage ID
+        type: str
+            Type of storage
+        created_at: str
+            Creation time
+        last_sync: str
+            Time last sync finished, can be empty.
+        last_sync_count: int
+            Number of tasks synced in the last sync
+        """
+        # originally syn was implemented in Client class, keep it for compatibility
+        response = self.make_request(
+            "POST", f"/api/storages/{storage_type}/{str(storage_id)}/sync"
         )
         return response.json()
 
+    # write func for syn export storage
+    def sync_export_storage(self, storage_type, storage_id):
+        """Synchronize Export (Target) Cloud Storage.
+
+        Parameters
+        ----------
+        storage_type: string
+            Specify the type of the storage container. See ProjectStorage for available types.
+        storage_id: int
+            Specify the storage ID of the storage container. See get_export_storages() to get ids.
+
+        Returns
+        -------
+        dict:
+            containing the same fields as in the original storage request and:
+
+        id: int
+            Storage ID
+        type: str
+            Type of storage
+        created_at: str
+            Creation time
+        other fields:
+            See more https://api.labelstud.io/#tag/Storage:S3/operation/api_storages_export_s3_sync_create
+        """
+        response = self.make_request(
+            "POST", f"/api/storages/export/{storage_type}/{str(storage_id)}/sync"
+        )
+        return response.json()
+
+    # write code for get_import_storages()
+    def get_import_storages(self):
+        """Get Import (Source) Cloud Storage.
+
+        Returns
+        -------
+        list of dicts:
+            List of dicts with source storages, each dict consists of these fields:
+
+        -------
+        Each dict consists of these fields:
+
+        id : int
+            A unique integer value identifying this storage.
+        type : str
+            The type of the storage. Default is "s3".
+        synchronizable : bool
+            Indicates if the storage is synchronizable. Default is True.
+        presign : bool
+            Indicates if the storage is presign. Default is True.
+        last_sync : str or None
+            The last sync finished time. Can be None.
+        last_sync_count : int or None
+            The count of tasks synced last time. Can be None.
+        last_sync_job : str or None
+            The last sync job ID. Can be None.
+        status : str
+            The status of the storage. Can be one of "initialized", "queued", "in_progress", "failed", "completed".
+        traceback : str or None
+            The traceback report for the last failed sync. Can be None.
+        meta : dict or None
+            Meta and debug information about storage processes. Can be None.
+        title : str or None
+            The title of the cloud storage. Can be None.
+        description : str or None
+            The description of the cloud storage. Can be None.
+        created_at : str
+            The creation time of the storage.
+        bucket : str or None
+            The S3 bucket name. Can be None.
+        prefix : str or None
+            The S3 bucket prefix. Can be None.
+        regex_filter : str or None
+            The cloud storage regex for filtering objects. Can be None.
+        use_blob_urls : bool
+            Indicates if objects are interpreted as BLOBs and generate URLs.
+        aws_access_key_id : str or None
+            The AWS_ACCESS_KEY_ID. Can be None.
+        aws_secret_access_key : str or None
+            The AWS_SECRET_ACCESS_KEY. Can be None.
+        aws_session_token : str or None
+            The AWS_SESSION_TOKEN. Can be None.
+        aws_sse_kms_key_id : str or None
+            The AWS SSE KMS Key ID. Can be None.
+        region_name : str or None
+            The AWS Region. Can be None.
+        s3_endpoint : str or None
+            The S3 Endpoint. Can be None.
+        presign_ttl : int
+            The presigned URLs TTL (in minutes).
+        recursive_scan : bool
+            Indicates if a recursive scan over the bucket content is performed.
+        glob_pattern : str or None
+            The glob pattern for syncing from bucket. Can be None.
+        synced : bool
+            Flag indicating if the dataset has been previously synced or not.
+
+        """
+        response = self.make_request("GET", f"/api/storages/?project={self.id}")
+        return response.json()
+
+    def get_export_storages(self):
+        """Get Export (Target) Cloud Storage.
+
+        Returns
+        -------
+        list of dicts:
+            List of dicts with target storages
+
+        -------
+        Each dict consists of these fields:
+
+        id : int
+            A unique integer value identifying this storage.
+        type : str
+            The type of the storage. Default is "s3".
+        synchronizable : bool
+            Indicates if the storage is synchronizable. Default is True.
+        last_sync : str or None
+            The last sync finished time. Can be None.
+        last_sync_count : int or None
+            The count of tasks synced last time. Can be None.
+        last_sync_job : str or None
+            The last sync job ID. Can be None.
+        status : str
+            The status of the storage. Can be one of "initialized", "queued", "in_progress", "failed", "completed".
+        traceback : str or None
+            The traceback report for the last failed sync. Can be None.
+        meta : dict or None
+            Meta and debug information about storage processes. Can be None.
+        title : str or None
+            The title of the cloud storage. Can be None.
+        description : str or None
+            The description of the cloud storage. Can be None.
+        created_at : str
+            The creation time of the storage.
+        can_delete_objects : bool or None
+            Deletion from storage enabled. Can be None.
+        bucket : str or None
+            The S3 bucket name. Can be None.
+        prefix : str or None
+            The S3 bucket prefix. Can be None.
+        regex_filter : str or None
+            The cloud storage regex for filtering objects. Can be None.
+        use_blob_urls : bool
+            Indicates if objects are interpreted as BLOBs and generate URLs.
+        aws_access_key_id : str or None
+            The AWS_ACCESS_KEY_ID. Can be None.
+        aws_secret_access_key : str or None
+            The AWS_SECRET_ACCESS_KEY. Can be None.
+        aws_session_token : str or None
+            The AWS_SESSION_TOKEN. Can be None.
+        aws_sse_kms_key_id : str or None
+            The AWS SSE KMS Key ID. Can be None.
+        region_name : str or None
+            The AWS Region. Can be None.
+        s3_endpoint : str or None
+            The S3 Endpoint. Can be None.
+        project : int
+            A unique integer value identifying this project.
+        """
+        response = self.make_request("GET", f"/api/storages/export?project={self.id}")
+        return response.json()
+
     def _assign_by_sampling(
         self,
         users: List[int],
@@ -1739,8 +2097,8 @@ class Project(Client):
         list[dict]
             List of dicts with counter of created assignments
         """
-        assert len(users) > 0, 'Users list is empty.'
-        assert len(users) >= overlap, 'Overlap is more than number of users.'
+        assert len(users) > 0, "Users list is empty."
+        assert len(users) >= overlap, "Overlap is more than number of users."
         # check if users are int and not User objects
         if isinstance(users[0], int):
             # get users from project
@@ -1750,7 +2108,7 @@ class Project(Client):
         final_results = []
         # Get tasks to assign
         tasks = self.get_tasks(view_id=view_id, only_ids=True)
-        assert len(tasks) > 0, 'Tasks list is empty.'
+        assert len(tasks) > 0, "Tasks list is empty."
         # Choice fraction of tasks
         if fraction != 1.0:
             k = int(len(tasks) * fraction)
@@ -1839,7 +2197,7 @@ class Project(Client):
         overlap: int = 1,
     ):
         """
-        Behaves similarly like `assign_annotators()` but instead of specify tasks_ids explicitely,
+        Behaves similarly like `assign_annotators()` but instead of specify tasks_ids explicitly,
         it gets users' IDs list and optional view ID and splits all tasks across annotators.
         Fraction expresses the size of dataset to be assigned.
         Parameters
@@ -1888,7 +2246,7 @@ class Project(Client):
         finished_at: str
             Finished time
         """
-        response = self.make_request('GET', f'/api/projects/{self.id}/exports')
+        response = self.make_request("GET", f"/api/projects/{self.id}/exports")
         return response.json()
 
     def export_snapshot_create(
@@ -1965,12 +2323,107 @@ class Project(Client):
             },
         }
         response = self.make_request(
-            'POST',
-            f'/api/projects/{self.id}/exports?interpolate_key_frames={interpolate_key_frames}',
+            "POST",
+            f"/api/projects/{self.id}/exports?interpolate_key_frames={interpolate_key_frames}",
             json=payload,
         )
         return response.json()
 
+    def export(
+        self,
+        filters=None,
+        title="SDK Export",
+        export_type="JSON",
+        output_dir=".",
+        **kwargs,
+    ):
+        """
+        Export tasks from the project with optional filters,
+        and save the exported data to a specified directory.
+
+        This method:
+        (1) creates a temporary view with the specified filters if they are not None,
+        (2) creates a new export snapshot using the view ID,
+        (3) checks the status of the snapshot creation while it's in progress,
+        (4) and downloads the snapshot file in the specified export format.
+        (5) After the export, it cleans up and remove the temporary view.
+
+        Parameters
+        ----------
+        filters : data_manager.Filters, dict, optional
+            Filters to apply when exporting tasks.
+            If provided, a temporary view is created with these filters.
+            The format of the filters should match the Label Studio filter options.
+            Default is None, which means all tasks are exported.
+            Use label_studio_sdk.data_manager.Filters.create() to create filters,
+            Example of the filters JSON format:
+        ```json
+        {
+          "conjunction": "and",
+          "items": [
+            {
+              "filter": "filter:tasks:id",
+              "operator": "equal",
+              "type": "Number",
+              "value": 1
+            }
+          ]
+        }
+        ```
+        titile : str, optional
+            The title of the export snapshot. Default is 'SDK Export'.
+        export_type : str, optional
+            The format of the exported data. It should be one of the formats supported by Label Studio ('JSON', 'CSV', etc.). Default is 'JSON'.
+        output_dir : str, optional
+            The directory where the exported file will be saved. Default is the current directory.
+        kwargs : kwargs, optional
+            The same parameters as in the export_snapshot_create method.
+
+        Returns
+        -------
+        dict
+            containing the status of the export, the filename of the exported file, and the export ID.
+
+        filename : str
+            Path to the downloaded export file
+        status : int
+            200 is ok
+        export_id : int
+            Export ID, you can retrieve more details about this export using this ID
+        """
+
+        # Create a temporary view with the specified filters
+        if filters:
+            view = self.create_view(title="Temp SDK export", filters=filters)
+            task_filter_options = {"view": view["id"]}
+        else:
+            task_filter_options = None
+            view = None
+
+        # Create a new export snapshot using the view ID
+        export_result = self.export_snapshot_create(
+            title=title,
+            task_filter_options=task_filter_options,
+            **kwargs,
+        )
+
+        # Check the status of the snapshot creation
+        export_id = export_result["id"]
+        while self.export_snapshot_status(export_id).is_in_progress():
+            time.sleep(1.0)  # Wait until the snapshot is ready
+
+        os.makedirs(output_dir, exist_ok=True)
+
+        # Download the snapshot file once it's ready
+        status, filename = self.export_snapshot_download(
+            export_id, export_type=export_type, path=output_dir
+        )
+
+        # Clean up the view
+        if view:
+            self.delete_view(view["id"])
+        return {"status": status, "filename": filename, "export_id": export_id}
+
     def export_snapshot_status(self, export_id: int) -> ExportSnapshotStatus:
         """
         Get export snapshot status by Export ID
@@ -1997,12 +2450,12 @@ class Project(Client):
             Finished time
         """
         response = self.make_request(
-            'GET', f'/api/projects/{self.id}/exports/{export_id}'
+            "GET", f"/api/projects/{self.id}/exports/{export_id}"
         )
         return ExportSnapshotStatus(response.json())
 
     def export_snapshot_download(
-        self, export_id: int, export_type: str = 'JSON', path: str = "."
+        self, export_id: int, export_type: str = "JSON", path: str = "."
     ) -> (int, str):
         """
         Download file with export snapshot in provided format
@@ -2022,8 +2475,8 @@ class Project(Client):
         Status code for operation and downloaded filename
         """
         response = self.make_request(
-            'GET',
-            f'/api/projects/{self.id}/exports/{export_id}/download?exportType={export_type}',
+            "GET",
+            f"/api/projects/{self.id}/exports/{export_id}/download?exportType={export_type}",
         )
         filename = None
         if response.status_code == 200:
@@ -2032,8 +2485,8 @@ class Project(Client):
                 filename = content_disposition.split("filename=")[-1].strip("\"'")
                 filename = os.path.basename(filename)
             else:
-                raise LabelStudioException('No filename in response')
-            with open(os.path.join(path, filename), 'wb') as f:
+                raise LabelStudioException("No filename in response")
+            with open(os.path.join(path, filename), "wb") as f:
                 for chk in response:
                     f.write(chk)
         return response.status_code, filename
@@ -2051,7 +2504,7 @@ class Project(Client):
         Status code for operation
         """
         response = self.make_request(
-            'DELETE', f'/api/projects/{self.id}/exports/{export_id}'
+            "DELETE", f"/api/projects/{self.id}/exports/{export_id}"
         )
         return response.status_code
 
@@ -2075,10 +2528,10 @@ class Project(Client):
         filenames = []
         if tasks:
             for task in tasks:
-                for key in task['data']:
+                for key in task["data"]:
                     try:
                         filename = get_local_path(
-                            task['data'][key],
+                            task["data"][key],
                             access_token=self.api_key,
                             hostname=self.url,
                         )
@@ -2095,7 +2548,7 @@ class Project(Client):
         task_id: int
             Task id.
         """
-        assert isinstance(task_id, int), 'task_id should be int'
+        assert isinstance(task_id, int), "task_id should be int"
         return self.make_request("DELETE", f"/api/tasks/{task_id}")
 
     def delete_tasks(self, task_ids: list) -> Response:
@@ -2106,7 +2559,7 @@ class Project(Client):
         task_ids: list of int
             Task ids.
         """
-        assert isinstance(task_ids, list), 'task_ids should be list of int'
+        assert isinstance(task_ids, list), "task_ids should be list of int"
         if not task_ids:  # avoid deletion of all tasks when task_ids = []
             return Response()
         payload = {
@@ -2127,7 +2580,7 @@ class Project(Client):
         """
         assert (
             isinstance(excluded_ids, list) or excluded_ids is None
-        ), 'excluded_ids should be list of int or None'
+        ), "excluded_ids should be list of int or None"
         if excluded_ids is None:
             excluded_ids = []
         payload = {