@searpent/react-image-annotate 2.0.1 → 2.0.4

package/src/utils/photosToImages.js

@@ -0,0 +1,40 @@
+function labelAndTextFromResultText(resultText) {
+  if (!resultText) { return {} }
+  const parsedResultText = JSON.parse(resultText);
+  const label = parsedResultText[0].label;
+  const text = parsedResultText[0].text;
+  return { label, text }
+}
+
+function modelResultsToRegions(modelResults) {
+  return modelResults.map(r => {
+    const { label, text } = labelAndTextFromResultText(r.text);
+    return {
+      id: r.id,
+      type: "box",
+      visible: true,
+      cls: label,
+      highlighted: false,
+      groupHighlighted: false,
+      x: r.box.X1,
+      y: r.box.Y1,
+      w: r.box.X2 - r.box.X1,
+      h: r.box.Y2 - r.box.Y1,
+      groupId: r.groupId,
+      text: text,
+    }
+  })
+}
+
+function photosToImages(photos) {
+  return photos.map(photo => ({
+    id: photo.id,
+    src: photo.fullsize.key,
+    thumbnail: photo.thumbnail.key,
+    name: photo.fullsize.key,
+    regions: modelResultsToRegions(photo.modelResults.v1.filter(mr => mr.name === 'extraction-engine')[0].results),
+    metadata: photo.metadata
+  }))
+}
+
+export default photosToImages;

package/src/utils/regions-to-blocks.js

@@ -0,0 +1,14 @@
+function regionsToBlocks(regions) {
+  return regions.map(r => ({
+    id: r.id,
+    type: "annotation",
+    data: {
+      text: r.text || '',
+      labelName: r.cls,
+      groupColor: r.groupColor,
+      groupId: r.groupId
+    }
+  }))
+}
+
+export default regionsToBlocks;