Added export format validation

tod

Author: VaghinakDev

src/superannotate/lib/app/interface/sdk_interface.py

@@ -1190,7 +1190,7 @@ def prepare_export(
         :param kwargs:
             Arbitrary kwargs::
                  * integration_name: can be provided which will be used as a storage to store export file
-                 * export_type: can be CSV for the Gen AI projects
+                 * format: can be CSV for the Gen AI projects and can be COCO for Vector projects
 
         :return: metadata object of the prepared export
         :rtype: dict
@@ -1219,9 +1219,13 @@ def prepare_export(
             else:
                 raise AppException("Integration not found.")
         _export_type = None
-        export_type = kwargs.get("export_type")
-        if export_type and export_type == "csv":
-            _export_type = 3
+        export_type = kwargs.get("format")
+        if export_type:
+            export_type = export_type.lower()
+            if export_type == "csv":
+                _export_type = 3
+            elif export_type == "coco":
+                _export_type = 2
         response = self.controller.prepare_export(
             project_name=project_name,
             folder_names=folders,

src/superannotate/lib/core/usecases/models.py

@@ -77,6 +77,20 @@ def validate_fuse(self):
                 f"{ProjectType.get_name(self._project.type)} attached with URLs"
             )
 
+    def validate_export_type(self):
+        if self._export_type == 2:
+            if (
+                self._project.type != ProjectType.VECTOR.value
+                or self._project.upload_state != constances.UploadState.EXTERNAL.value
+            ):
+                raise AppValidationException(
+                    "COCO format is not supported for this project."
+                )
+        elif self._export_type == 3 and self._project.type != ProjectType.GEN_AI.value:
+            raise AppValidationException(
+                "CSV format is not supported for this project."
+            )
+
     def validate_folder_names(self):
         if self._folder_names:
             condition = Condition("project_id", self._project.id, EQ)