Merge pull request #1926 from kili-technology/feature/lab-3719-aau-i-can-upload-geojson-labels-directly-without-conversion

feat(LAB-3719): add `append_labels_from_geojson_files` method

Author: RuellePaul

pyproject.toml

@@ -44,7 +44,7 @@ dependencies = [
   "filelock >= 3.0.0, < 4.0.0",
   "pip-system-certs >= 4.0.0, < 5.0.0; platform_system=='Windows'",
   "pyrate-limiter >= 3, < 4",
-  "kili-formats == 0.2.3"
+  "kili-formats == 0.2.7"
 ]
 urls = { homepage = "https://github.com/kili-technology/kili-python-sdk" }
 
@@ -84,7 +84,7 @@ dev = [
   "vulture==2.11",
   "dead==1.5.2",
   # optional dependencies
-  "kili-formats[all] == 0.2.3",
+  "kili-formats[all] == 0.2.7",
   "opencv-python >= 4.0.0, < 5.0.0",
   "azure-storage-blob >= 12.0.0, < 13.0.0",
   # optional dependencies gis
@@ -101,7 +101,7 @@ dev = [
 all = [
   # aggregate all optional deps without dev
   "azure-storage-blob >= 12.0.0, < 13.0.0",
-  "kili-formats[all] == 0.2.3",
+  "kili-formats[all] == 0.2.7",
   "opencv-python >= 4.0.0, < 5.0.0",
   "Pillow >=9.0.0, <11.0.0",
   "pyproj == 3.6.1; python_version >= '3.9'",
@@ -115,7 +115,7 @@ cli = [
   "tabulate >= 0.9.0, < 0.10.0"
 ]
 coco = [
-  "kili-formats[coco] == 0.2.3"
+  "kili-formats[coco] == 0.2.7"
 ]
 gis = [
   "pyproj >= 2.6.1, < 3; python_version < '3.9'",
@@ -124,12 +124,12 @@ gis = [
 ]
 image = [
   "Pillow >=9.0.0, <11.0.0",
-  "kili-formats[image] == 0.2.3"
+  "kili-formats[image] == 0.2.7"
 ]
 image-utils = ["opencv-python >= 4.0.0, < 5.0.0"]
 
 video = [
-  "kili-formats[video] == 0.2.3"
+  "kili-formats[video] == 0.2.7"
 ]
 yolo = [
   "pyyaml >= 6.0, < 7.0"

src/kili/presentation/client/label.py

@@ -1,6 +1,7 @@
 """Client presentation methods for labels."""
 
 # pylint: disable=too-many-lines
+import json
 import warnings
 from itertools import repeat
 from typing import (
@@ -16,6 +17,7 @@
     overload,
 )
 
+from kili_formats.format.geojson import geojson_feature_collection_to_kili_json_response
 from typeguard import typechecked
 
 from kili.adapters.kili_api_gateway.helpers.queries import QueryOptions
@@ -1399,3 +1401,65 @@ def append_labels_from_shapefiles(
             json_response_array=[json_response],
             asset_external_id_array=[asset_external_id],
         )
+
+    @typechecked
+    def append_labels_from_geojson_files(
+        self,
+        project_id: str,
+        asset_external_id: str,
+        geojson_file_paths: List[str],
+    ):
+        """Import and convert GeoJSON files into annotations for a specific asset in a Kili project.
+
+        This method processes GeoJSON feature collections, converts them to the appropriate
+        Kili annotation format, and appends them as labels to the specified asset.
+        The GeoJSON features must contain job names and category information in their properties.
+
+        Args:
+            project_id: The ID of the Kili project to add the labels to.
+            asset_external_id: The external ID of the asset to label.
+            geojson_file_paths: List of file paths to the GeoJSON files to be processed.
+                Each file should contain a FeatureCollection with features that have
+                'kili' metadata in their properties, including 'job', 'type', and 'categories'.
+
+        Note:
+            The GeoJSON features must contain a 'kili' property with the following structure:
+            - 'job': The name of the job in the Kili project
+            - 'type': The annotation type ('marker', 'polyline', 'semantic', etc.)
+            - 'categories': List of category objects with 'name' field
+
+            Supported geometries: Point, LineString, Polygon, and MultiPolygon.
+            Polygon and MultiPolygon are always mapped to semantic segmentation jobs in Kili.
+
+        Examples:
+            >>> kili.append_labels_from_geojson_files(
+                    project_id="project_id",
+                    asset_external_id="asset_1",
+                    geojson_file_paths=["annotations.geojson"]
+                )
+        """
+        merged_json_response = {}
+
+        for file_path in geojson_file_paths:
+            with open(file_path, encoding="utf-8") as f:
+                feature_collection = json.load(f)
+
+            json_response = geojson_feature_collection_to_kili_json_response(feature_collection)
+
+            for job_name, job_data in json_response.items():
+                if job_name not in merged_json_response:
+                    merged_json_response[job_name] = job_data
+                else:
+                    for key, value in job_data.items():
+                        if key == "annotations":
+                            merged_json_response[job_name].setdefault("annotations", []).extend(
+                                value
+                            )
+                        else:
+                            merged_json_response[job_name][key] = value
+
+        self.append_labels(
+            project_id=project_id,
+            json_response_array=[merged_json_response],
+            asset_external_id_array=[asset_external_id],
+        )