use a context manager

explosion · Oct 5, 2023 · 0bcad08 · 0bcad08
1 parent 99f0206
commit 0bcad08
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/prodigy_pdf/__init__.py b/prodigy_pdf/__init__.py
@@ -11,9 +11,9 @@
 
 def page_to_image(page: pdfium.PdfPage) -> str:
     pil_image = page.render().to_pil()
-    buffered = BytesIO()
-    pil_image.save(buffered, format="JPEG")
-    img_str = base64.b64encode(buffered.getvalue())
+    with BytesIO() as buffered:
+        pil_image.save(buffered, format="JPEG")
+        img_str = base64.b64encode(buffered.getvalue())
     return f"data:image/png;base64,{img_str.decode('utf-8')}"