feat: DIA-1685: [sdk] Create example predictions and annotations from a LabelConfig

pyproject.toml

@@ -49,6 +49,7 @@ typing_extensions = ">= 4.0.0"
 ujson = ">=5.8.0"
 xmljson = "0.2.1"
 
+jsf = "^0.11.2"
 [tool.poetry.dev-dependencies]
 mypy = "1.0.1"
 pytest = "^7.4.0"

src/label_studio_sdk/label_interface/control_tags.py

@@ -505,6 +505,13 @@ class SpanSelectionOffsets(SpanSelection):
 class ChoicesValue(BaseModel):
     choices: List[str]
 
+    # I don't know how Choices predictions with choice != 'multiple' was working without this...
+    @validator("choices", pre=True, always=True)
+    def coerce_to_list(cls, value: Union[str, List[str]]):
+        if isinstance(value, str):
+            return [value]
+        return value
+
 
 class ChoicesTag(ControlTag):
     """ """
@@ -540,6 +547,15 @@ def to_json_schema(self):
             "description": f"Choices for {self.to_name[0]}"
         }
 
+    def _validate_labels(self, labels):
+        if super()._validate_labels(labels):
+            return True
+
+        # HACK to continue to support single-item output in json schema
+        if not self.is_multiple_choice and isinstance(labels, str):
+            return super()._validate_labels([labels])
+
+
 
 class LabelsValue(SpanSelection):
     labels: List[str]

src/label_studio_sdk/label_interface/interface.py

@@ -15,6 +15,7 @@
 from collections import defaultdict, OrderedDict
 from lxml import etree
 import xmljson
+from jsf import JSF
 
 from label_studio_sdk._legacy.exceptions import (
     LSConfigParseException,
@@ -770,7 +771,7 @@ def validate_region(self, region) -> bool:
             return False
 
         # type of the region should match the tag name        
-        if control.tag.lower() != region["type"]:
+        if control.tag.lower() != region["type"].lower():
             return False
 
         # make sure that in config it connects to the same tag as
@@ -839,9 +840,23 @@ def generate_sample_task(self, mode="upload", secure_mode=False):
 
         return task
 
-    def generate_sample_annotation(self):
+    def generate_sample_prediction(self):
         """ """
-        raise NotImplemented()
+        prediction = PredictionValue(
+            model_version='sample model version',
+            result=[
+                {
+                    'from_name': control.name,
+                    'to_name': control.to_name[0],
+                    'type': control.tag,
+                    # TODO: put special case for choices in generation instead of validation
+                    'value': {control._label_attr_name: JSF(control.to_json_schema()).generate()}
+                } for control in self.controls
+            ]
+        )
+        prediction_dct = prediction.model_dump()
+        assert self.validate_prediction(prediction_dct), 'could not generate a sample prediction'
+        return prediction_dct
 
     #####
     ##### COMPATIBILITY LAYER