bringing back prettyfication, annotations are pretty again!

Author: VahagnTKA

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

@@ -774,6 +774,18 @@ def validate_item_names(self):
 
             self._item_names = [item.name for item in self._images.get_all(condition)]
 
+    def _prettify_annotations(self, annotations: List[dict]):
+        restruct = {}
+
+        if self._item_names_provided:
+            for annotation in annotations:
+                restruct[annotation["metadata"]["name"]] = annotation
+            try:
+                return [restruct[x] for x in self._item_names]
+            except KeyError:
+                raise AppException("Broken data.")
+
+        return annotations
 
     async def get_big_annotation(
         self,
@@ -858,7 +870,7 @@ def execute(self):
                 self.reporter.log_warning(
                     f"Could not find annotations for {items_count - received_items_count}/{items_count} items."
                 )
-            self._response.data = annotations
+            self._response.data = self._prettify_annotations(annotations)
         return self._response