@konfuzio/document-validation-ui 0.1.5-styles-refactor → 0.1.6-multi-ann-set

package/src/locales/de.json

@@ -49,8 +49,7 @@
   "data_being_extracted": "Neue Daten werden extrahiert",
   "analysing_document": "Wir analysieren Ihr Dokument.",
   "few_minutes": "Dies könnte ein paar Minuten dauern.",
-  "reject_label": "Ablehnen",
-  "rejected": "Abgelehnt",
+  "missing_annotation": "Als fehlend markieren",
   "ann_exists": "Eine Annotation für dieses Label und Label-Set existiert bereits.",
   "lite_mode": "Schreibgeschützt",
   "limited_functionalities": "Sie verwenden die schreibgeschützte Version mit begrenzten Funktionalitäten.",
@@ -88,14 +87,13 @@
   "categorize_document_no_category_description": "Bitten Sie den Projektmanager, eine Beschreibung dieser Kategorie hinzuzufügen, um mehr über diese Kategorie zu erfahren.",
   "annotations_pending": "Ausstehende Annotationen",
   "annotations_accepted": "Akzeptierte Annotationen",
-  "reject_all_empty": "Alle fehlenden Annotationen ablehnen",
+  "mark_all_missing": "Alle leeren als fehlend markieren",
   "no_labels_to_choose": "Keine Labels",
   "accept_group": "Alle akzeptieren",
   "use_your_keyboard": "Benutze deine Tastatur",
   "arrow_keys": "Navigieren",
   "esc_key": "Exit",
   "enter_key": "Akzeptieren oder Speichern",
-  "delete_key": "Ablehnen",
   "decline": "Decline",
   "server_error": "Wir haben derzeit ein Serverproblem. Bitte versuchen Sie es später noch einmal oder",
   "get_support": "Hole dir Unterstützung",
@@ -110,8 +108,27 @@
   "new_multi_ann_description": "Wählen Sie ein Datenmodell aus den vorhandenen aus und deaktivieren Sie dann die Labels, die in diesem Dokument nicht vorhanden sind.",
   "drag_drop_columns_multi_ann": "Drag and Drop, um die Reihenfolge der Labels zu ändern",
   "error_creating_multi_ann": "Nicht alle Annotationen wurden erfolgreich erstellt.",
-  "disabled_finish_review": "Der Abschluss der Dokumentenprüfung ist erst möglich, nachdem alle Annotationen überarbeitet wurden, unabhängig davon, ob sie akzeptiert oder abgelehnt wurden.",
   "no_multi_ann_labelset_model": "Es gibt keine verfügbaren Label Sets, die mehrfach in diesem Dokument zu finden sind. Weitere Informationen finden Sie unter <a href='https://help.konfuzio.com/tutorials/advanced-document-extraction/index.html#multiple-annotation-sets' target='_blank'>diesem Link</a>.",
   "single_category_in_project": "Das aktuelle Projekt hat nur eine Kategorie. Um sie ändern zu können, sollte die neue zuerst dem Projekt hinzugefügt werden. Einzelheiten dazu finden Sie unter <a href='https://help.konfuzio.com/modules/categories/index.html#add-a-category' target='_blank'>diesem Link</a>.",
-  "approved_annotations": "Die Kategorie kann nicht geändert werden, da bereits akzeptierte Annotationen oder manuell erstellte Annotationen zu diesem Dokument vorhanden sind."
+  "approved_annotations": "Die Kategorie kann nicht geändert werden, da bereits akzeptierte Annotationen oder manuell erstellte Annotationen zu diesem Dokument vorhanden sind.",
+  "restore": "Restore",
+  "disabled_finish_review": "Das Beenden der Dokumentenprüfung ist erst möglich, nachdem alle Annotationen überarbeitet wurden, unabhängig davon, ob sie akzeptiert oder als fehlend markiert wurden.",
+  "split_modal_title": "Dokument aufteilen",
+  "split_modal_body": "Wir haben Ihre Datei gescannt und <strong>{number_of_split_documents}</strong> verschiedene Dokumente gefunde. <strong>Wir empfehlen dringend,</strong> sie zu diesem Zeitpunkt aufzuteilen, da sonst alle Fortschritte, die Sie gemacht haben, aufgrund der erneuten Extraktion der Dokumentdaten verloren gehen.",
+  "do_it_later": "Ich werde es später machen",
+  "review_now": "Jetzt reviewen",
+  "recommended": "Empfohlen",
+  "smart_split": "Smart Split",
+  "smart_split_suggestions": "Unsere Smart Split-Vorschläge sind grün hervorgehoben",
+  "no_splitting_suggestions": "Dieses Dokument enthält keine Split-Vorschläge",
+  "confirm_splitting": "Sind Sie sicher, dass Sie die Änderungen bestätigen möchten?",
+  "splitting_warning": "Beachten Sie bitte, dass alle Annotationen, die Sie in diesem Dokument gemacht haben, nicht gespeichert werden, wenn Sie fortfahren.",
+  "no": "Nein",
+  "yes": "Ja",
+  "split_modal_no_suggestions": "Do you need to split the document? <strong>We strongly recommend</strong> splitting it at this stage, otherwise all the progress you’ve made will be lost in the future.",
+  "missing_from_document": "Fehlt im Dokument",
+  "edit_label": "Edit label",
+  "delete_label": "Delete label",
+  "table": "Table",
+  "delete_table": "Delete table"
 }

package/src/locales/en.json

@@ -51,8 +51,7 @@
   "data_being_extracted": "New data is being extracted",
   "analysing_document": "We are analysing your document.",
   "few_minutes": "This might take a few minutes.",
-  "reject_label": "Reject",
-  "rejected": "Rejected",
+  "missing_annotation": "Mark as Missing",
   "ann_exists": "An annotation for this label and label set already exists.",
   "lite_mode": "Read Only",
   "limited_functionalities": "You are using the Read Only version with limited functionalities.",
@@ -89,7 +88,7 @@
   "categorize_document_no_category_description": "Ask the Project Manager to add a description of this Category to know more about this category.",
   "annotations_pending": "pending",
   "annotations_accepted": "accepted",
-  "reject_all_empty": "Reject all empty",
+  "mark_all_missing": "Mark all empty as Missing",
   "no_labels_to_choose": "No Labels",
   "accept_group": "Accept all",
   "new_ann_set_title": "New annotation set",
@@ -102,7 +101,6 @@
   "arrow_keys": "to Navigate",
   "esc_key": "Exit",
   "enter_key": "Accept or Save",
-  "delete_key": "Reject",
   "decline": "Decline",
   "server_error": "We are currently experiencing a server issue. Please try again later or",
   "get_support": "Get Support",
@@ -111,8 +109,28 @@
   "new_multi_ann_description": "Select a data model from the existing ones, then deselect the labels that don't exist in this document.",
   "drag_drop_columns_multi_ann": "Drag and drop to change the order of the labels",
   "error_creating_multi_ann": "Not all annotation sets were created successfully.",
-  "disabled_finish_review": "Finishing the document review is only possible after all annotations have been revised, whether they are accepted or rejected.",
+  "disabled_finish_review": "Finishing the document review is only possible after all annotations have been revised, whether they are accepted or marked as missing.",
   "no_multi_ann_labelset_model": "There are no available label sets that can be found multiple times in this document. For more details, you can visit <a href='https://help.konfuzio.com/tutorials/advanced-document-extraction/index.html#multiple-annotation-sets' target='_blank'>this link</a>.",
   "single_category_in_project": "The current project has only one category. To be able to change it, the new one should be first added to the project. For details on how to do this, visit <a href='https://help.konfuzio.com/modules/categories/index.html#add-a-category' target='_blank'>this link</a>.",
-  "approved_annotations": "It is not possible to change the current category since the document has approved or manually created annotations."
+  "approved_annotations": "It is not possible to change the current category since the document has approved or manually created annotations.",
+  "restore": "Restore",
+  "split_modal_title": "Split the document",
+  "split_modal_body": "We've scanned your file and found <strong>{number_of_split_documents}</strong> different documents. <strong>We strongly recommend</strong> splitting it at this stage, otherwise all the progress you’ve made will be lost due to document data re-extraction.",
+  "do_it_later": "I'll do it later",
+  "review_now": "Review now",
+  "recommended": "Recommended",
+  "smart_split": "Smart Split",
+  "smart_split_suggestions": "Our Smart Split suggestions are highlighted in green",
+  "no_splitting_suggestions": "This document does not have splitting suggestions",
+  "confirm_splitting": "Are you sure you want to confirm the changes?",
+  "splitting_warning": "Be aware that any annotations you have made in this document will not be saved if you proceed.",
+  "no": "No",
+  "yes": "Yes",
+  "prepare_document": "Prepare the document",
+  "split_modal_no_suggestions": "Do you need to split the document? <strong>We strongly recommend</strong> splitting it at this stage, otherwise all the progress you’ve made will be lost in the future.",
+  "missing_from_document": "Missing from the document",
+  "edit_label": "Edit label",
+  "delete_label": "Delete label",
+  "table": "Table",
+  "delete_table": "Delete table"
 }

package/src/locales/es.json

@@ -51,8 +51,7 @@
   "data_being_extracted": "Se están extrayendo nuevos datos",
   "analysing_document": "Estamos analizando el documento.",
   "few_minutes": "Esto puede tardar unos minutos.",
-  "reject_label": "Rechazar",
-  "rejected": "Rechazadas",
+  "missing_annotation": "Marcar como Faltante",
   "ann_exists": "Ya existe una anotación para esta etiqueta y grupo de etiquetas.",
   "lite_mode": "Sólo lectura",
   "limited_functionalities": "Está utilizando la versión de sólo lectura con funcionalidad limitada.",
@@ -89,14 +88,13 @@
   "categorize_document_no_category_description": "Solicite a su Gestor de Proyecto que agregue una descripción a esta Categoría para saber más sobre ella.",
   "annotations_pending": "pendientes",
   "annotations_accepted": "aceptadas",
-  "reject_all_empty": "Rechazar todas las anotaciones vacías",
+  "mark_all_missing": "Marcas todas las anotaciones vacías como Faltantes",
   "no_labels_to_choose": "Sin etiquetas",
   "accept_group": "Aceptar todas",
   "use_your_keyboard": "Use su teclado",
   "arrow_keys": "para Navegar",
   "esc_key": "Salir del modo de edición",
   "enter_key": "Aceptar o Guardar",
-  "delete_key": "Rechazar",
   "decline": "Declinar",
   "server_error": "Estamos experimentando problemas con el servidor. Por favor, repita la operación o ",
   "get_support": "Contacte al Soporte",
@@ -109,8 +107,27 @@
   "continue": "Continuar",
   "drag_drop_columns_multi_ann": "Arrastre y suelte las columnas para ordenar las etiquetas",
   "error_creating_multi_ann": "No todos los grupos de anotaciones fueron creados de manera exitosa.",
-  "disabled_finish_review": "Solo es posible finalizar la revisión del documento si todas las anotaciones han sido revisadas, ya sea aceptadas o rechazadas.",
+  "disabled_finish_review": "Solo es posible finalizar la revisión del documento si todas las anotaciones han sido revisadas, ya sea aceptadas o marcadas como faltantes.",
   "no_multi_ann_labelset_model": "En este documento no hay grupos de etiquetas múltiples disponibles. Para más información, haz clic en <a href='https://help.konfuzio.com/tutorials/advanced-document-extraction/index.html#multiple-annotation-sets' target='_blank'>este enlace</a>.",
   "single_category_in_project": "Este proyecto solo tiene una categoría. Para poder modificarla, es necesario agregar esta nueva categoría al proyecto. Para más información, visita <a href='https://help.konfuzio.com/modules/categories/index.html#add-a-category' target='_blank'>este enlace</a>.",
-  "approved_annotations": "No es posible cambiar la categoría, ya que existen anotaciones aceptadas o creadas manualmente en este documento."
+  "approved_annotations": "No es posible cambiar la categoría, ya que existen anotaciones aceptadas o creadas manualmente en este documento.",
+  "restore": "Restore",
+  "split_modal_title": "Dividir el documento",
+  "split_modal_body": "Hemos escaneado tu documento y hemos encontrado <strong>{number_of_split_documents}</strong> documentos diferentes. <strong>Te recomendamos</strong> dividir el documento en esta etapa, ya que de lo contrario se perderá el progreso logrado debido a una nueva extracción de los datos del documento.",
+  "do_it_later": "Lo haré luego",
+  "review_now": "Revisar ahora",
+  "recommended": "Recomendado",
+  "smart_split": "División Inteligente",
+  "smart_split_suggestions": "Nuestras sugerencias de División Inteligente están marcadas en color verde",
+  "no_splitting_suggestions": "Este documento no tiene sugerencias de división",
+  "confirm_splitting": "¿Seguro que quieres confirmar los cambios?",
+  "splitting_warning": "Ten en cuenta que se perderán las anotaciones que hayas confirmado en el documento.",
+  "no": "No",
+  "yes": "Sí",
+  "split_modal_no_suggestions": "Do you need to split the document? <strong>We strongly recommend</strong> splitting it at this stage, otherwise all the progress you’ve made will be lost in the future.",
+  "missing_from_document": "Falta en el documento",
+  "edit_label": "Edit label",
+  "delete_label": "Delete label",
+  "table": "Table",
+  "delete_table": "Delete table"
 }

package/src/store/display.js

@@ -30,6 +30,7 @@ const state = {
   documentActionBar: null, // document action bar properties
   categorizeModalIsActive: false,
   pageChangedFromThumbnail: false,
+  showAnnSetTable: null,
 };
 
 const getters = {
@@ -60,6 +61,31 @@ const getters = {
   imageScale: (state) => (page) => {
     return new BigNumber(page.size[0]).div(page.original_size[0]).toNumber();
   },
+  bboxToPoint:
+    (state, getters) =>
+    (page, point, hasOffset = false) => {
+      const imageScale = getters.imageScale(page);
+      const { x, y } = point;
+      const pageHeight = new BigNumber(page.original_size[1]);
+      const newPoint = {
+        // left
+        x: new BigNumber(x)
+          .minus(hasOffset ? 1 : 0)
+          .times(state.scale)
+          .times(imageScale)
+          .div(PIXEL_RATIO)
+          .toNumber(),
+        // top
+        y: pageHeight
+          .minus(new BigNumber(y))
+          .minus(hasOffset ? 17.1 : 0)
+          .times(state.scale)
+          .times(imageScale)
+          .div(PIXEL_RATIO)
+          .toNumber(),
+      };
+      return newPoint;
+    },
   bboxToRect:
     (state, getters) =>
     (page, bbox, hasOffset = false) => {
@@ -207,6 +233,12 @@ const actions = {
       show ? { icon, text, action, loading } : null
     );
   },
+  showAnnSetTable({ commit }, tableSet) {
+    commit("SET_ANN_SET_TABLE", tableSet);
+  },
+  toggleAnnSetTable({ commit }, tableSet) {
+    commit("TOGGLE_ANN_SET_TABLE", tableSet);
+  },
   setCategorizeModalIsActive: ({ commit }, value) => {
     commit("SET_CATEGORIZE_MODAL_IS_ACTIVE", value);
   },
@@ -236,6 +268,18 @@ const mutations = {
     state.documentActionBar = actionBar;
   },
 
+  SET_ANN_SET_TABLE: (state, tableSet) => {
+    state.showAnnSetTable = tableSet;
+  },
+
+  TOGGLE_ANN_SET_TABLE: (state, tableSet) => {
+    if (state.showAnnSetTable) {
+      state.showAnnSetTable = null;
+    } else {
+      state.showAnnSetTable = tableSet;
+    }
+  },
+
   SET_CATEGORIZE_MODAL_IS_ACTIVE: (state, value) => {
     state.categorizeModalIsActive = value;
   },

package/src/store/document.js

@@ -3,6 +3,7 @@ import { sleep } from "../utils/utils";
 
 const HTTP = myImports.HTTP;
 const documentPollDuration = 1000;
+export const table_reference_api = "api.v3.dvui.table";
 
 const state = {
   loading: true,
@@ -21,13 +22,14 @@ const state = {
   showActionError: false,
   errorMessage: null,
   showDocumentError: false,
-  rejectedMissingAnnotations: null,
+  annotationsMarkedAsMissing: null,
   errorMessageWidth: null,
   hoveredAnnotationSet: null,
   finishedReview: false,
   newAcceptedAnnotations: null,
   selectedEntities: null,
   serverError: false,
+  splittingSuggestions: null,
 };
 
 const getters = {
@@ -157,6 +159,78 @@ const getters = {
     return state.annotationSets.find((annSet) => annSet.id === annotationSetId);
   },
 
+  /* Get annotation sets created in table */
+  annotationSetsInTable: (state) => () => {
+    const annotationSetsList = {};
+    state.annotationSets.forEach((annotationSet) => {
+      let addAnnotationSet = false;
+      if (annotationSet.labels) {
+        annotationSet.labels.forEach((label) => {
+          if (
+            label.annotations &&
+            label.annotations.find(
+              (annotation) =>
+                annotation.origin && annotation.origin === table_reference_api
+            )
+          ) {
+            addAnnotationSet = true;
+            return;
+          }
+        });
+      }
+      if (addAnnotationSet) {
+        // group by label set
+        if (annotationSetsList[`${annotationSet.label_set.id}`]) {
+          annotationSetsList[`${annotationSet.label_set.id}`].push(
+            annotationSet
+          );
+        } else {
+          annotationSetsList[`${annotationSet.label_set.id}`] = [annotationSet];
+        }
+      }
+    });
+    return annotationSetsList;
+  },
+
+  /* Get annotation sets without tables */
+  annotationSetsToShowInList: (state) => () => {
+    const annotationSetsList = [];
+    state.annotationSets.forEach((annotationSet) => {
+      let addAnnotationSet = true;
+      if (annotationSet.labels) {
+        annotationSet.labels.forEach((label) => {
+          if (
+            label.annotations &&
+            label.annotations.find(
+              (annotation) =>
+                annotation.origin && annotation.origin === table_reference_api
+            )
+          ) {
+            addAnnotationSet = false;
+            return;
+          }
+        });
+      }
+      if (addAnnotationSet) {
+        annotationSetsList.push(annotationSet);
+      }
+    });
+    return annotationSetsList;
+  },
+
+  /* Get annotations inside a list of annotation sets */
+  annotationsInAnnotationsSets: (state) => (annotationsSets) => {
+    const annotations = [];
+    annotationsSets.forEach((annotationSet) => {
+      annotationSet.labels.forEach((label) => {
+        label.annotations.forEach((annotation) => {
+          annotations.push(annotation);
+        });
+      });
+    });
+    return annotations;
+  },
+
   /* Process annotations and extract labels and sets */
   processAnnotationSets: (state, getters) => (annotationSets) => {
     // group annotations for sidebar
@@ -228,6 +302,7 @@ const getters = {
       });
       return found ? `${value + 1}` : "";
     }
+    return "";
   },
 
   /**
@@ -308,6 +383,27 @@ const getters = {
     return pendingEmpty.length;
   },
 
+  annotationIsNotFound: (state) => (annotationSet, label) => {
+    // Check if the combined label and label set have been marked as missing
+    // or if the document is in public mode
+    if (state.missingAnnotations.length === 0) {
+      return false;
+    } else {
+      const found = state.missingAnnotations.filter(
+        (el) =>
+          el.label === label.id &&
+          el.annotation_set === annotationSet.id &&
+          el.label_set === annotationSet.label_set.id
+      );
+
+      if (found.length !== 0) {
+        return true;
+      } else {
+        return false;
+      }
+    }
+  },
+
   // Check if document is ready to be finished
   isDocumentReviewFinished: (state) => () => {
     // check if all annotations have been revised
@@ -335,7 +431,7 @@ const getters = {
 
     // if all annotations have been revised
     // and if there are no empty annotations or
-    // all empty annotations were rejected,
+    // all empty annotations were marked as missing,
     // we can finish the review
     if (
       !emptyAnnotations ||
@@ -419,6 +515,20 @@ const getters = {
     }
   },
 
+  /**
+   * If automatic splitting is enabled for the project
+   */
+  waitingForSplittingConfirmation: () => (document) => {
+    return document && document.status_data === 41;
+  },
+
+  /**
+   * Show the Smart Split switch or not
+   */
+  documentHasProposedSplit: () => (document) => {
+    return document.proposed_split && document.proposed_split.length > 0;
+  },
+
   /**
    * Joins all strings in a multi-entity Annotation array
    * to look like a single string
@@ -445,7 +555,7 @@ const getters = {
   /**
    * Check status of annotation
    */
-  notFound: () => (annotation) => {
+  notExtracted: () => (annotation) => {
     if (annotation) {
       return !annotation.span;
     } else {
@@ -574,8 +684,8 @@ const actions = {
   setDocumentError: ({ commit }, value) => {
     commit("SET_DOCUMENT_ERROR", value);
   },
-  setRejectedMissingAnnotations: ({ commit }, annotations) => {
-    commit("SET_REJECTED_MISSING_ANNOTATIONS", annotations);
+  setAnnotationsMarkedAsMissing: ({ commit }, annotations) => {
+    commit("SET_ANNOTATIONS_MARKED_AS_MISSING", annotations);
   },
   setErrorMessageWidth: ({ commit }, width) => {
     commit("SET_ERROR_MESSAGE_WIDTH", width);
@@ -589,6 +699,9 @@ const actions = {
   setSelectedEntities: ({ commit }, entities) => {
     commit("SET_SELECTED_ENTITIES", entities);
   },
+  setSplittingSuggestions: ({ commit }, value) => {
+    commit("SET_SPLITTING_SUGGESTIONS", value);
+  },
 
   /**
    * Actions that use HTTP requests always return the axios promise,
@@ -635,6 +748,10 @@ const actions = {
             });
           }
 
+          if (getters.documentHasProposedSplit(response.data)) {
+            commit("SET_SPLITTING_SUGGESTIONS", response.data.proposed_split);
+          }
+
           categoryId = response.data.category;
           // TODO: add this validation to a method
           isRecalculatingAnnotations = response.data.labeling_available !== 1;
@@ -698,7 +815,7 @@ const actions = {
       id: annotation.id,
       span,
       page: span.page_index + 1,
-      labelName: label.name,
+      labelName: label ? label.name : "",
     };
     commit("SET_DOCUMENT_ANNOTATION_SELECTED", value);
   },
@@ -716,7 +833,7 @@ const actions = {
       HTTP.post(`/annotations/`, annotation)
         .then(async (response) => {
           if (response.status === 201) {
-            dispatch("fetchMissingAnnotations");
+            await dispatch("fetchMissingAnnotations");
             commit("SET_FINISHED_REVIEW", getters.isDocumentReviewFinished());
 
             if (!getters.annotationSetExists(response.data.annotation_set)) {
@@ -827,13 +944,14 @@ const actions = {
     });
   },
 
-  addMissingAnnotations: ({ commit, dispatch }, missingAnnotations) => {
+  addMissingAnnotations: ({ commit, getters }, missingAnnotations) => {
     return new Promise((resolve, reject) => {
-      return HTTP.post(`/missing-annotations/`, missingAnnotations)
-        .then((response) => {
+      HTTP.post(`/missing-annotations/`, missingAnnotations)
+        .then(async (response) => {
           if (response.status === 201) {
-            commit("SET_REJECTED_MISSING_ANNOTATIONS", null);
-            dispatch("fetchMissingAnnotations");
+            commit("SET_ANNOTATIONS_MARKED_AS_MISSING", null);
+            commit("ADD_MISSING_ANNOTATIONS", response.data);
+            commit("SET_FINISHED_REVIEW", getters.isDocumentReviewFinished());
           }
 
           resolve(response);
@@ -845,12 +963,13 @@ const actions = {
     });
   },
 
-  deleteMissingAnnotation: ({ commit, getters, dispatch }, id) => {
+  deleteMissingAnnotation: ({ commit, getters }, id) => {
     return new Promise((resolve, reject) => {
       return HTTP.delete(`/missing-annotations/${id}/`)
         .then((response) => {
           if (response.status === 204) {
-            dispatch("fetchMissingAnnotations");
+            commit("DELETE_MISSING_ANNOTATION", id);
+            commit("SET_FINISHED_REVIEW", getters.isDocumentReviewFinished());
             resolve(true);
           }
         })
@@ -891,8 +1010,11 @@ const actions = {
         `documents/${state.documentId}/?fields=status_data,labeling_available`
       )
         .then((response) => {
-          if (getters.isDocumentReadyToBeReviewed(response.data)) {
-            // ready
+          if (
+            getters.isDocumentReadyToBeReviewed(response.data) ||
+            getters.waitingForSplittingConfirmation(response.data)
+          ) {
+            // ready or has splitting suggestions
             return resolve(true);
           } else if (getters.documentHadErrorDuringExtraction(response.data)) {
             // error
@@ -1029,11 +1151,21 @@ const mutations = {
           (existingAnnotation) => existingAnnotation.id === annotation.id
         );
         if (indexOfAnnotationAnnotationSets > -1) {
-          label.annotations.splice(
-            indexOfAnnotationAnnotationSets,
-            1,
-            annotation
-          );
+          // checks if an annotation label was changed and add it to the new label
+          if (annotation.label && annotation.label.id !== label.id) {
+            label.annotations.splice(indexOfAnnotationAnnotationSets, 1);
+
+            const labelToAdd = annotationSet.labels.find(
+              (labelToAdd) => labelToAdd.id === annotation.label.id
+            );
+            labelToAdd.annotations.push(annotation);
+          } else {
+            label.annotations.splice(
+              indexOfAnnotationAnnotationSets,
+              1,
+              annotation
+            );
+          }
           updatedAnnotation = true;
           return;
         }
@@ -1123,7 +1255,20 @@ const mutations = {
   SET_MISSING_ANNOTATIONS: (state, missingAnnotations) => {
     state.missingAnnotations = missingAnnotations;
   },
+  ADD_MISSING_ANNOTATIONS: (state, missingAnnotations) => {
+    missingAnnotations.map((annotation) => {
+      state.missingAnnotations.push(annotation);
+    });
+  },
+  DELETE_MISSING_ANNOTATION: (state, id) => {
+    const indexOfAnnotationToDelete = state.missingAnnotations.findIndex(
+      (annotation) => annotation.id === id
+    );
 
+    if (indexOfAnnotationToDelete > -1) {
+      state.missingAnnotations.splice(indexOfAnnotationToDelete, 1);
+    }
+  },
   SET_SHOW_ACTION_ERROR: (state, value) => {
     state.showActionError = value;
   },
@@ -1133,8 +1278,8 @@ const mutations = {
   SET_DOCUMENT_ERROR: (state, value) => {
     state.showDocumentError = value;
   },
-  SET_REJECTED_MISSING_ANNOTATIONS: (state, annotations) => {
-    state.rejectedMissingAnnotations = annotations;
+  SET_ANNOTATIONS_MARKED_AS_MISSING: (state, annotations) => {
+    state.annotationsMarkedAsMissing = annotations;
   },
   SET_ERROR_MESSAGE_WIDTH: (state, width) => {
     state.errorMessageWidth = width;
@@ -1164,6 +1309,9 @@ const mutations = {
   UPDATE_FILE_NAME: (state, value) => {
     state.selectedDocument.data_file_name = value;
   },
+  SET_SPLITTING_SUGGESTIONS: (state, array) => {
+    state.splittingSuggestions = array;
+  },
 };
 
 export default {