From 2c36d9e8bf83fb45ccbcf7b677ec61a9064079b7 Mon Sep 17 00:00:00 2001
From: kartikvirendrar <kartikrajput357@gmail.com>
Date: Thu, 28 Nov 2024 16:30:49 +0530
Subject: [PATCH] added new project type OCRTextlineSegmentation

---
 backend/projects/annotation_registry.py          |  9 +++++++++
 .../ocr/ocr_textline_segmentation.jsx            |  5 +++++
 backend/projects/project_registry.yaml           | 16 ++++++++++++++++
 3 files changed, 30 insertions(+)
 create mode 100644 backend/projects/label_studio_jsx_files/ocr/ocr_textline_segmentation.jsx

diff --git a/backend/projects/annotation_registry.py b/backend/projects/annotation_registry.py
index 2dddab5d3..dd4c53452 100644
--- a/backend/projects/annotation_registry.py
+++ b/backend/projects/annotation_registry.py
@@ -108,6 +108,15 @@
             "type": "labels",
         },
     },
+    "OCRTextlineSegmentation": {
+        "ocr_transcribed_json": {
+            "to_name": "image_url",
+            "from_name": [
+                "annotation_bboxes",
+            ],
+            "type": ["rectangle"],
+        },
+    },
     "OCRTranscription": {
         "ocr_transcribed_json": {
             "to_name": "image_url",
diff --git a/backend/projects/label_studio_jsx_files/ocr/ocr_textline_segmentation.jsx b/backend/projects/label_studio_jsx_files/ocr/ocr_textline_segmentation.jsx
new file mode 100644
index 000000000..7684b037b
--- /dev/null
+++ b/backend/projects/label_studio_jsx_files/ocr/ocr_textline_segmentation.jsx
@@ -0,0 +1,5 @@
+<View>
+  <Style>.ant-input { font-size: large; }</Style>
+  <Image name="image_url" value="$image_url"/>
+  <Rectangle name="annotation_bboxes" toName="image_url" strokeWidth="3" className="ignore_assertion"/>
+</View>
diff --git a/backend/projects/project_registry.yaml b/backend/projects/project_registry.yaml
index b4c0c8d85..f36c098f9 100644
--- a/backend/projects/project_registry.yaml
+++ b/backend/projects/project_registry.yaml
@@ -99,6 +99,22 @@ OCR:
         fields:
           annotations:
             - ocr_transcribed_json
+    OCRTextlineSegmentation:
+      project_mode: "Annotation"
+      label_studio_jsx_file: "ocr_textline_segmentation.jsx"
+      input_dataset:
+        class: OCRDocument
+        fields:
+          - image_url
+        display_fields:
+          - image_url
+        prediction: ocr_prediction_json
+      output_dataset: 
+        class: OCRDocument
+        save_type: in_place
+        fields:
+          annotations:
+            - ocr_transcribed_json
     OCRTranscriptionEditing:
       project_mode: "Annotation"
       label_studio_jsx_file: "ocr/ocr_transcription.jsx"