S7 updates

.gitattributes

@@ -2,4 +2,3 @@
 *.png binary
 *.traineddata binary
 *.whl binary
-benchmarks/assets/imageproc/*.png filter=lfs diff=lfs merge=lfs -text

.pre-commit-config.yaml

@@ -18,7 +18,7 @@ repos:
       - id: trailing-whitespace
 
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.9.2
+    rev: v0.9.3
     hooks:
       - id: ruff
         args: [--fix]

src/main.py

@@ -1,4 +1,5 @@
 import tesserocr  # noqa #  Note: Somehow needed, otherwise the binary has an issue with tesserocr
+from PIL import Image  # noqa #  Note: Somehow needed, otherwise the binary has an issue with tesserocr
 import logging
 import os
 import sys
@@ -7,7 +8,6 @@
 
 import keyboard
 from beautifultable import BeautifulTable
-from PIL import Image  # noqa #  Note: Somehow needed, otherwise the binary has an issue with tesserocr
 
 import src.logger
 from src import __version__

src/utils/ocr/read.py

@@ -4,7 +4,7 @@
 
 import cv2
 import numpy as np
-from tesserocr import OEM, RIL, PyTessBaseAPI
+import tesserocr
 
 from src.config import BASE_DIR
 from src.config.data import COLORS
@@ -24,13 +24,15 @@ def __init__(self):
         self._apis = []
         self._lock = threading.Lock()
 
-    def get_api(self, user: uuid.UUID) -> PyTessBaseAPI:
+    def get_api(self, user: uuid.UUID) -> tesserocr.PyTessBaseAPI:
         with self._lock:
             for api in self._apis:
                 if not api["user"]:
                     api["user"] = user
                     return api["api"]
-            new_api = PyTessBaseAPI(psm=3, oem=OEM.LSTM_ONLY, path=str(self.tessdata_path), lang=IniConfigLoader().general.language)
+            new_api = tesserocr.PyTessBaseAPI(
+                psm=3, oem=tesserocr.OEM.LSTM_ONLY, path=str(self.tessdata_path), lang=IniConfigLoader().general.language
+            )
             new_api.SetVariable("debug_file", "/dev/null")
             self._apis.append({"api": new_api, "user": user})
             return new_api
@@ -69,7 +71,7 @@ def image_to_text(img: np.ndarray, line_boxes: bool = False, do_pre_proc: bool =
     text = api.GetUTF8Text().strip()
     res = OcrResult(original_text=text, text=text, word_confidences=api.AllWordConfidences(), mean_confidence=api.MeanTextConf())
     if line_boxes:
-        line_boxes_res = api.GetComponentImages(RIL.TEXTLINE, True)
+        line_boxes_res = api.GetComponentImages(tesserocr.RIL.TEXTLINE, True)
         _APILoader().release_api(my_id)
         return res, line_boxes_res
     _APILoader().release_api(my_id)