ocr-stream 2.2.6__tar.gz → 2.2.8__tar.gz

{ocr_stream-2.2.6 → ocr_stream-2.2.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ocr_stream
-Version: 2.2.6
+Version: 2.2.8
 Summary: Extrair textos de documentos digitalizados e imagens.
 Author: Bruno Chaves
 Requires-Python: >=3.11
@@ -11,7 +11,7 @@ Requires-Dist: openpyxl
 Requires-Dist: pandas
 Requires-Dist: requests>=2.32.4
 Requires-Dist: opencv-python
-Requires-Dist: convert_stream>=2.2.4
+Requires-Dist: convert_stream>=2.2.7
 Requires-Dist: soup_files
 
 # ocrlib

{ocr_stream-2.2.6 → ocr_stream-2.2.8}/ocr_stream/ocr/__init__.py

@@ -20,7 +20,7 @@ from ocr_stream.bin_tess import BinTesseract
 try:
     import pyocr.tesseract
 except:
-    pyocr.tesseract = None
+    pyocr = None
 
 
 class TextRecognized(object):
@@ -150,7 +150,6 @@ class IPytesseract(ABCOcrTesseract):
 # Modulo OCR pyocr
 # ======================================================================#
 
-
 class IPyOcr(ABCOcrTesseract):
 
     def __init__(self, bin_tess: BinTesseract, *, lang: str = None, tess_data_dir: str = None):
@@ -163,7 +162,7 @@ class IPyOcr(ABCOcrTesseract):
         # The tools are returned in the recommended order of usage
         self._pyOcr = pyocr_modules[0]
         langs: list[str] = self._pyOcr.get_available_languages()
-        if lang in lang:
+        if lang in langs:
             self.lang = lang
         else:
             self.lang = langs[0]

{ocr_stream-2.2.6 → ocr_stream-2.2.8}/ocr_stream/recognize.py

@@ -13,7 +13,11 @@ class RecognizeImage(object):
         Reconhecer textos em imagens com pytesseract|pyocr usando o tesseract
     """
 
-    def __init__(self, bin_tess: BinTesseract, *, lib_ocr: LibOcr = DEFAULT_LIB_OCR):
+    def __init__(
+                self,
+                bin_tess: BinTesseract = BinTesseract(), *,
+                lib_ocr: LibOcr = DEFAULT_LIB_OCR
+            ):
         super().__init__()
         self.module_ocr: TesseractOcr = TesseractOcr.create(
             bin_tess,
@@ -65,29 +69,36 @@ class RecognizeImage(object):
 
 class RecognizePdf(object):
 
-    def __init__(self, bin_tess: BinTesseract, *, lib_ocr: LibOcr = DEFAULT_LIB_OCR):
+    def __init__(
+                self,
+                bin_tess: BinTesseract = BinTesseract(), *,
+                lib_ocr: LibOcr = DEFAULT_LIB_OCR,
+            ):
         self.recognize_image: RecognizeImage = RecognizeImage(bin_tess, lib_ocr=lib_ocr)
         self.pbar: ProgressBarAdapter = CreatePbar().get()
 
     def set_pbar(self, p: ProgressBarAdapter):
         self.pbar = p
 
-    def recognize_page_pdf(self, page: PageDocumentPdf) -> DocumentPdf:
+    def recognize_page_pdf(self, page: PageDocumentPdf, *, dpi: int = 350) -> DocumentPdf:
         """
             Converte a página em Imagem, reconhece o texto com OCR e
         retorna uma nova página com o texto reconhecido.
         """
         _doc = DocumentPdf.create_from_pages([page])
         _conv = ConvertPdfToImages.create(_doc)
-        img: ImageObject = _conv.to_images()[0]
+        img: ImageObject = _conv.to_images(dpi=dpi)[0]
         text_recognized: TextRecognized = self.recognize_image.image_recognize(img)
+        del _doc
+        del _conv
+        del img
         return text_recognized.to_document()
 
-    def recognize_document(self, doc: DocumentPdf) -> DocumentPdf:
+    def recognize_document(self, doc: DocumentPdf, *, dpi: int = 350) -> DocumentPdf:
         _pages: list[PageDocumentPdf] = []
         _pages.clear()
         _conv = ConvertPdfToImages.create(doc)
-        images: list[ImageObject] = _conv.to_images()
+        images: list[ImageObject] = _conv.to_images(dpi=dpi)
         max_num: int = len(images)
         for num, img in enumerate(images):
             self.pbar.update(
@@ -101,9 +112,9 @@ class RecognizePdf(object):
         new_doc = DocumentPdf.create_from_pages(_pages)
         return new_doc
 
-    def recognize_file_pdf(self, file: File) -> DocumentPdf:
+    def recognize_file_pdf(self, file: File, *, dpi: int = 300) -> DocumentPdf:
         file_doc = DocumentPdf.create_from_file(file)
-        recognized_doc = self.recognize_document(file_doc)
+        recognized_doc = self.recognize_document(file_doc, dpi=dpi)
         del file_doc
         return recognized_doc
 

{ocr_stream-2.2.6 → ocr_stream-2.2.8}/ocr_stream.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ocr_stream
-Version: 2.2.6
+Version: 2.2.8
 Summary: Extrair textos de documentos digitalizados e imagens.
 Author: Bruno Chaves
 Requires-Python: >=3.11
@@ -11,7 +11,7 @@ Requires-Dist: openpyxl
 Requires-Dist: pandas
 Requires-Dist: requests>=2.32.4
 Requires-Dist: opencv-python
-Requires-Dist: convert_stream>=2.2.4
+Requires-Dist: convert_stream>=2.2.7
 Requires-Dist: soup_files
 
 # ocrlib

{ocr_stream-2.2.6 → ocr_stream-2.2.8}/ocr_stream.egg-info/requires.txt

@@ -4,5 +4,5 @@ openpyxl
 pandas
 requests>=2.32.4
 opencv-python
-convert_stream>=2.2.4
+convert_stream>=2.2.7
 soup_files

{ocr_stream-2.2.6 → ocr_stream-2.2.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ocr_stream"
-version = "2.2.6"
+version = "2.2.8"
 authors = [
   { name="Bruno Chaves"},
 ]
@@ -20,7 +20,7 @@ dependencies = [
     "pandas",
     "requests>=2.32.4",
     "opencv-python",
-    "convert_stream>=2.2.4",
+    "convert_stream>=2.2.7",
     "soup_files",
 ]