feat: DEV-1621: Add ability to create annotation with pre-created regions (#496)

* feat: DEV-1621: Add ability to create new annotations already with some results from predictions

* fix: DEV-1621: Fix default coords in not interactive predictions for images

* Use proper FF notation

Co-authored-by: Nick Skriabin <[email protected]>
Co-authored-by: nicholasrq <[email protected]>

Author: Gondragos

src/components/AnnotationTabs/AnnotationTabs.js

@@ -77,7 +77,7 @@ export const AnnotationTabs = observer(({
   }, [as]);
 
   const onCreateAnnotation = useCallback(() => {
-    const c = as.addAnnotation({ userGenerate: true });
+    const c = as.createAnnotation();
 
     as.selectAnnotation(c.id);
   }, [as]);

src/components/Annotations/Annotations.js

@@ -223,7 +223,7 @@ class Annotations extends Component {
                 size="small"
                 onClick={ev => {
                   ev.preventDefault();
-                  const c = store.annotationStore.addAnnotation({ userGenerate: true });
+                  const c = store.annotationStore.createAnnotation();
 
                   store.annotationStore.selectAnnotation(c.id);
                   // c.list.selectAnnotation(c);

src/components/TopBar/Annotations.js

@@ -105,7 +105,7 @@ export const Annotations = observer(({ store, annotationStore }) => {
 
 const CreateAnnotation = observer(({ annotationStore, onClick }) => {
   const onCreateAnnotation = useCallback(() => {
-    const c = annotationStore.addAnnotation({ userGenerate: true });
+    const c = annotationStore.createAnnotation();
 
     annotationStore.selectAnnotation(c.id);
     onClick();

src/stores/AnnotationStore.js

@@ -17,7 +17,7 @@ import Area from "../regions/Area";
 import throttle from "lodash.throttle";
 import { ViewModel } from "../tags/visual";
 import { UserExtended } from "./UserStore";
-import { FF_DEV_1555, isFF } from "../utils/feature-flags";
+import { FF_DEV_1555, FF_DEV_1621, isFF } from "../utils/feature-flags";
 
 const hotkeys = Hotkey("Annotations", "Annotations");
 
@@ -1224,6 +1224,15 @@ export default types
       return self.root;
     }
 
+    function findNonInteractivePredictionResults() {
+      return self.predictions.reduce((results, prediction) => {
+        return [
+          ...results,
+          ...prediction._initialAnnotationObj.filter(result => result.interactive_mode === false).map(r => ({ ...r })),
+        ];
+      }, []);
+    }
+
     function createItem(options) {
       const { user, config } = self.store;
 
@@ -1285,6 +1294,39 @@ export default types
       return record;
     }
 
+    function createAnnotation(options = { userGenerate: true }) {
+      const result = isFF(FF_DEV_1621) ? findNonInteractivePredictionResults() : [];
+      const c = self.addAnnotation({ ...options, result });
+
+      if (result && result.length) {
+        const ids = {};
+
+        // Area id is <uniq-id>#<annotation-id> to be uniq across all tree
+        result.forEach(r => {
+          if ("id" in r) {
+            const id = r.id.replace(/#.*$/, `#${c.id}`);
+
+            ids[r.id] = id;
+            r.id = id;
+          }
+        });
+
+        result.forEach(r => {
+          if (r.parent_id) {
+            if (ids[r.parent_id]) r.parent_id = ids[r.parent_id];
+            // impossible case but to not break the app better to reset it
+            else r.parent_id = null;
+          }
+        });
+
+        selectAnnotation(c.id);
+        c.deserializeAnnotation(result);
+        // reinit will trigger `updateObjects()` so we omit it here
+        c.reinitHistory();
+      }
+      return c;
+    }
+
 
     function addHistory(options = {}) {
       options.type = "history";
@@ -1404,6 +1446,7 @@ export default types
 
       addPrediction,
       addAnnotation,
+      createAnnotation,
       addAnnotationFromPrediction,
       addHistory,
       clearHistory,

src/utils/feature-flags.js

@@ -20,6 +20,9 @@ export const FF_DEV_1536 = "ff_front_dev_1536_taxonomy_user_labels_150222_long";
 // Show or not dialog for rejection
 export const FF_DEV_1593 = "ff_front_1593_rejection_comment_040222_short";
 
+// Add an interactivity flag to the results to make some predictions' results be able to be automatically added to newly created annotations.
+export const FF_DEV_1621 = "ff_front_dev_1621_interactive_mode_150222_short";
+
 function getFeatureFlags() {
   return window.APP_SETTINGS?.feature_flags || {};
 }
@@ -28,7 +31,7 @@ export function isFF(id) {
   const featureFlags = getFeatureFlags();
 
   if (id in featureFlags) {
-    return featureFlags[id] === true; 
+    return featureFlags[id] === true;
   } else {
     return window.APP_SETTINGS?.feature_flags_default_value === true;
   }