gc.collect

README.md

@@ -34,7 +34,7 @@ services:
       - ./pdf/done:/app/done
 ```
 
-1. Run `docker compose up` to OCR the PDFs and move them into `./pdf/done`
+3. Run `docker compose up` to OCR the PDFs and move them into `./pdf/done`
 
 ## Quick Start
 

src/main.py

@@ -1,3 +1,8 @@
+"""
+This script OCRs PDFs
+"""
+
+import gc
 import os
 import sys
 from pathlib import Path
@@ -10,7 +15,7 @@
 
 def process_pdfs(base: Path = Path(".")):
     """
-    Process all PDF files in the "todo" directory and save the OCR results in the "done" directory.
+    Process all PDF files in "todo" and save the results in "done".
 
     :param base: The base directory containing the "todo" and "done" directories.
     """
@@ -24,23 +29,19 @@ def predict(input_file: Path, output_file: Path):
 
         # Create a PDF file to store the OCR results
         doc = fitz.open()
-        print(f"Created new PDF file: {output_file}")
+
         # Perform OCR on the images
         for image in convert_from_path(input_file, dpi=300, fmt="jpeg"):
-            print("Converted PDF page to image")
             prediction = pytesseract.image_to_pdf_or_hocr(image, extension="pdf")
-            print("Extracted text from image")
             doc.insert_pdf(fitz.open("pdf", prediction))
-            print("Inserted text into PDF")
+            gc.collect()
 
         # Save the OCR results to a new PDF file
         doc.save(output_file, garbage=4, deflate=True)
         doc.close()
-        print(f"Saved OCR results to {output_file}")
         input_file.unlink()
         print(f"Processed {relative_path}")
 
-    # parallel:
     Parallel(n_jobs=-1)(
         delayed(predict)(Path(root) / file, Path(root.replace("todo", "done")) / file)
         for root, _, files in os.walk(base / "todo")