[IDWA-OCR-72] Install OCR into an executable

OCR/ocr/assets/labels.json

@@ -0,0 +1,4 @@
+{
+  "255,0,0": "nbs_patient_id",
+  "0,0,255": "nbs_cas_id"
+}

OCR/ocr/main.py

@@ -1,4 +1,5 @@
 import os
+import sys
 
 from ocr.services.image_segmenter import ImageSegmenter
 from ocr.services.image_ocr import ImageOCR
@@ -9,10 +10,15 @@
 def main():
 
     segmentation_template = os.path.join(
-        path, "../tests/assets/form_segmention_template.png"
+        path, "assets/form_segmention_template.png"
     )
-    raw_image = os.path.join(path, "../tests/assets/form_filled.png")
-    labels_path = os.path.join(path, "../tests/assets/labels.json")
+    # I think this could be an area to enhance later on with a more fleshed out CLI
+    if sys.argv[1] is None:
+        raw_image = os.path.join(path, "assets/form_filled.png")
+    elif sys.argv[1] is not None:
+        raw_image = sys.argv[1]
+
+    labels_path = os.path.join(path, "assets/labels.json")
 
     segmenter = ImageSegmenter(raw_image, segmentation_template, labels_path)
     segments = segmenter.segment()

OCR/ocr/pyinstaller.py

@@ -0,0 +1,18 @@
+import PyInstaller.__main__
+from pathlib import Path
+
+HERE = Path(__file__).parent.absolute()
+path_to_main = str(HERE / "main.py")
+
+#This function installs/packages the main OCR function as an executable 
+#while also including the needed assets ie. the mapping labels and segmentation template
+def install():
+    PyInstaller.__main__.run([
+        path_to_main,
+        '--onefile',
+        '--windowed',
+        # SOURCE:DESTINATION
+        '--add-data=ocr/assets/form_segmention_template.png:assets/',
+        '--add-data=ocr/assets/labels.json:assets/',
+        # other pyinstaller options... 
+    ])

OCR/pyproject.toml

@@ -23,3 +23,4 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry.scripts]
 main = "ocr.main:main"
+build = "ocr.pyinstaller:install"