trustgraph-ocr 1.2.3__tar.gz → 1.2.5__tar.gz

{trustgraph_ocr-1.2.3 → trustgraph_ocr-1.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-ocr
-Version: 1.2.3
+Version: 1.2.5
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Author-email: "trustgraph.ai" <security@trustgraph.ai>
 Project-URL: Homepage, https://github.com/trustgraph-ai/trustgraph

{trustgraph_ocr-1.2.3 → trustgraph_ocr-1.2.5}/trustgraph/decoding/ocr/pdf_decoder.py

@@ -6,12 +6,16 @@ PDF document as text as separate output objects.
 
 import tempfile
 import base64
+import logging
 import pytesseract
 from pdf2image import convert_from_bytes
 
 from ... schema import Document, TextDocument, Metadata
 from ... base import FlowProcessor, ConsumerSpec, ProducerSpec
 
+# Module logger
+logger = logging.getLogger(__name__)
+
 default_ident = "pdf-decoder"
 
 class Processor(FlowProcessor):
@@ -41,15 +45,15 @@ class Processor(FlowProcessor):
             )
         )
 
-        print("PDF OCR inited")
+        logger.info("PDF OCR processor initialized")
 
     async def on_message(self, msg, consumer, flow):
 
-        print("PDF message received", flush=True)
+        logger.info("PDF message received")
 
         v = msg.value()
 
-        print(f"Decoding {v.metadata.id}...", flush=True)
+        logger.info(f"Decoding {v.metadata.id}...")
 
         blob = base64.b64decode(v.data)
 
@@ -60,7 +64,7 @@ class Processor(FlowProcessor):
             try:
                 text = pytesseract.image_to_string(page, lang='eng')
             except Exception as e:
-                print(f"Page did not OCR: {e}")
+                logger.warning(f"Page did not OCR: {e}")
                 continue
 
             r = TextDocument(
@@ -70,7 +74,7 @@ class Processor(FlowProcessor):
 
             await flow("output").send(r)
 
-        print("Done.", flush=True)
+        logger.info("PDF decoding complete")
 
     @staticmethod
     def add_args(parser):

trustgraph_ocr-1.2.5/trustgraph/ocr_version.py

@@ -0,0 +1 @@
+__version__ = "1.2.5"

{trustgraph_ocr-1.2.3 → trustgraph_ocr-1.2.5}/trustgraph_ocr.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-ocr
-Version: 1.2.3
+Version: 1.2.5
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Author-email: "trustgraph.ai" <security@trustgraph.ai>
 Project-URL: Homepage, https://github.com/trustgraph-ai/trustgraph

trustgraph_ocr-1.2.3/trustgraph/ocr_version.py

@@ -1 +0,0 @@
-__version__ = "1.2.3"