close pdf (#10)

explosion · Dec 23, 2024 · bc295d2 · bc295d2
1 parent 4034df2
commit bc295d2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/prodigy_pdf/spans.py b/prodigy_pdf/spans.py
@@ -79,8 +79,10 @@ def pdf_to_images(path: Path) -> List[str]:
             pil_image.save(buffered, format="JPEG")
             img_str = base64.b64encode(buffered.getvalue())
         images.append(f"data:image/png;base64,{img_str.decode('utf-8')}")
-    return images
-
+    try:
+     return images
+    finally:
+        pdf.close()
 
 def disable_tokens(stream: StreamType, disabled: List[str]) -> StreamType:
     for eg in stream: