Merge pull request #1153 from AI4Bharat/ocr-new

added a default label for each bbox
AI4Bharat · Feb 11, 2025 · b769def · b769def
2 parents ee71501 + f25b267
commit b769def
Show file tree

Hide file tree

Showing 5 changed files with 60 additions and 15 deletions.
diff --git a/src/ui/pages/container/Label-Studio/AllTaskLSF.jsx b/src/ui/pages/container/Label-Studio/AllTaskLSF.jsx
@@ -73,7 +73,7 @@ const LabelStudioWrapper = ({annotationNotesRef, loader, showLoader, hideLoader,
 useEffect(() => {
   getProjectsandTasks(projectId, taskId).then(
     ([labelConfig, taskData, annotations, predictions]) => {
-    let sidePanel = labelConfig?.project_type?.includes("OCRSegmentCategorization");
+    let sidePanel = labelConfig?.project_type?.includes("OCRSegmentCategorization") || labelConfig?.project_type?.includes("OCRTranscriptionEditing");
     let showLabelsOnly = labelConfig?.project_type?.includes("OCRSegmentCategorization");
     let selectAfterCreateOnly = labelConfig?.project_type?.includes("OCRSegmentCategorization");
     let continousLabelingOnly = labelConfig?.project_type?.includes("OCRSegmentCategorization");    

diff --git a/src/ui/pages/container/Label-Studio/LSF.jsx b/src/ui/pages/container/Label-Studio/LSF.jsx
@@ -51,7 +51,7 @@ import InfoIcon from "@mui/icons-material/Info";
 import InfoOutlinedIcon from "@mui/icons-material/InfoOutlined";
 import getTaskAssignedUsers from "../../../../utils/getTaskAssignedUsers";
 import LightTooltip from "../../component/common/Tooltip";
-import { labelConfigJS } from "./labelConfigJSX";
+import { addLabelsToBboxes, labelConfigJS } from "./labelConfigJSX";
 
 const filterAnnotations = (
   annotations,
@@ -222,7 +222,7 @@ const LabelStudioWrapper = ({
   useEffect(() => {
   getProjectsandTasks(projectId, taskId).then(
     ([labelConfig, taskData, annotations, predictions]) => {
-    let sidePanel = labelConfig?.project_type?.includes("OCRSegmentCategorization");
+    let sidePanel = labelConfig?.project_type?.includes("OCRSegmentCategorization") || labelConfig?.project_type?.includes("OCRTranscriptionEditing");
     let showLabelsOnly = labelConfig?.project_type?.includes("OCRSegmentCategorization");
     let selectAfterCreateOnly = labelConfig?.project_type?.includes("OCRSegmentCategorization");
     let continousLabelingOnly = labelConfig?.project_type?.includes("OCRSegmentCategorization");    
@@ -397,6 +397,9 @@ const LabelStudioWrapper = ({
           let temp = annotation.serializeAnnotation();
           let ids = new Set();
           let countLables = 0;
+          if (projectType.includes("OCRTranscriptionEditing")){
+            addLabelsToBboxes(temp);
+          }
           temp.map((curr) => {
             if(curr.type !== "relation"){
               ids.add(curr.id);
@@ -440,7 +443,7 @@ const LabelStudioWrapper = ({
             showLoader();
             if (taskData.annotation_status !== "freezed") {
               postAnnotation(
-                annotation.serializeAnnotation(),
+                temp,
                 taskData.id,
                 userData.id,
                 load_time.current,
@@ -498,8 +501,10 @@ const LabelStudioWrapper = ({
           let temp = annotation.serializeAnnotation();
           let ids = new Set();
           let countLables = 0;
+          if (projectType.includes("OCRTranscriptionEditing")){
+            addLabelsToBboxes(temp);
+          }
           temp.map((curr) => {
-            // console.log(curr);
             if(curr.type !== "relation"){
               ids.add(curr.id);
             }
@@ -542,13 +547,12 @@ const LabelStudioWrapper = ({
               for (let i = 0; i < annotations.length; i++) {
                 if (
                   !annotations[i].result?.length ||
-                  !annotation.serializeAnnotation().length ||
-                  annotation.serializeAnnotation()[0].id ===
+                  !temp.length ||
+                  temp[0].id ===
                     annotations[i].result[0].id
                 ) {
                   setAutoSave(false);
                   showLoader();
-                  let temp = annotation.serializeAnnotation();
                   for (let i = 0; i < temp.length; i++) {
                     if(temp[i].type === "relation"){
                       continue;

diff --git a/src/ui/pages/container/Label-Studio/ReviewLSF.jsx b/src/ui/pages/container/Label-Studio/ReviewLSF.jsx
@@ -53,7 +53,7 @@ import styles from "./lsf.module.css";
 import "./lsf.css";
 import { useSelector, useDispatch } from "react-redux";
 import { translate } from "../../../../config/localisation";
-import { labelConfigJS } from "./labelConfigJSX";
+import { addLabelsToBboxes, labelConfigJS } from "./labelConfigJSX";
 
 
 const StyledMenu = styled((props) => (
@@ -281,7 +281,7 @@ const LabelStudioWrapper = ({
 useEffect(() => {
   getProjectsandTasks(projectId, taskId).then(
     ([labelConfig, taskData, annotations, predictions]) => {
-    let sidePanel = labelConfig?.project_type?.includes("OCRSegmentCategorization");
+    let sidePanel = labelConfig?.project_type?.includes("OCRSegmentCategorization") || labelConfig?.project_type?.includes("OCRTranscriptionEditing");
     let showLabelsOnly = labelConfig?.project_type?.includes("OCRSegmentCategorization");
     let selectAfterCreateOnly = labelConfig?.project_type?.includes("OCRSegmentCategorization");
     let continousLabelingOnly = labelConfig?.project_type?.includes("OCRSegmentCategorization");    
@@ -578,7 +578,9 @@ useEffect(() => {
           let temp = annotation.serializeAnnotation();
           let ids = new Set();
           let countLables = 0;         
-          console.log(temp);
+          if (projectType.includes("OCRTranscriptionEditing")){
+            addLabelsToBboxes(temp);
+          }
           temp.map((curr) => {
             if(curr.type !== "relation"){
               ids.add(curr.id);
@@ -621,7 +623,6 @@ useEffect(() => {
             if (taskData.annotation_status !== "freezed") {
               setAutoSave(false);
               showLoader();
-              let temp = annotation.serializeAnnotation();
 
               for (let i = 0; i < temp.length; i++) {
                 if(temp[i].type === "relation"){

diff --git a/src/ui/pages/container/Label-Studio/SuperCheckerLSF.jsx b/src/ui/pages/container/Label-Studio/SuperCheckerLSF.jsx
@@ -52,7 +52,7 @@ import styles from "./lsf.module.css";
 import "./lsf.css";
 import { useSelector, useDispatch } from "react-redux";
 import { translate } from "../../../../config/localisation";
-import { labelConfigJS } from "./labelConfigJSX";
+import { addLabelsToBboxes, labelConfigJS } from "./labelConfigJSX";
 
 const StyledMenu = styled((props) => (
   <Menu
@@ -202,7 +202,7 @@ const LabelStudioWrapper = ({
 useEffect(() => {
   getProjectsandTasks(projectId, taskId).then(
     ([labelConfig, taskData, annotations, predictions]) => {
-    let sidePanel = labelConfig?.project_type?.includes("OCRSegmentCategorization");
+    let sidePanel = labelConfig?.project_type?.includes("OCRSegmentCategorization") || labelConfig?.project_type?.includes("OCRTranscriptionEditing");
     let showLabelsOnly = labelConfig?.project_type?.includes("OCRSegmentCategorization");
     let selectAfterCreateOnly = labelConfig?.project_type?.includes("OCRSegmentCategorization");
     let continousLabelingOnly = labelConfig?.project_type?.includes("OCRSegmentCategorization");    
@@ -399,7 +399,10 @@ useEffect(() => {
         onUpdateAnnotation: function (ls, annotation) {
           let temp = annotation.serializeAnnotation();
           let ids = new Set();
-          let countLables = 0;   
+          let countLables = 0;
+          if (projectType.includes("OCRTranscriptionEditing")){
+            addLabelsToBboxes(temp);
+          }   
           temp.map((curr) => {
             if(curr.type !== "relation"){
               ids.add(curr.id);

diff --git a/src/ui/pages/container/Label-Studio/labelConfigJSX.js b/src/ui/pages/container/Label-Studio/labelConfigJSX.js
@@ -54,3 +54,40 @@ export const labelConfigJS = `<View>
     <Relation value="c" selected="true"/>
 </Relations>
 </View>`;
+
+export function addLabelsToBboxes(data) {
+    const bboxIdToLabel = new Map();
+    data.forEach(item => {
+        if (item.type === "labels" && item.value.labels) {
+            const bboxId = item.id;
+            bboxIdToLabel.set(bboxId, item.value.labels[0]);
+        }
+    });
+
+    data.forEach(item => {
+        if (item.type === "rectangle" && !bboxIdToLabel.has(item.id)) {
+            const labelItem = {
+                original_width: item.original_width,
+                original_height: item.original_height,
+                image_rotation: item.image_rotation,
+                value: {
+                    x: item.value.x,
+                    y: item.value.y,
+                    width: item.value.width,
+                    height: item.value.height,
+                    rotation: item.value.rotation,
+                    labels: ["Body"]
+                },
+                id: item.id,
+                from_name: "annotation_labels",
+                to_name: "image_url",
+                type: "labels",
+                origin: "manual"
+            };
+            data.push(labelItem);
+            bboxIdToLabel.set(item.id, "Body");
+        }
+    });
+
+    return data;
+}