From faea5b2f37a9e832bcb94df01d1e015694cf6380 Mon Sep 17 00:00:00 2001
From: Ramtin Yazdanian <ramtin.yazdanian@gmail.com>
Date: Wed, 15 Oct 2025 13:43:56 +0200
Subject: [PATCH] fix oversight of dangling opened pymupdf file

---
 graphai/core/image/image.py | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/graphai/core/image/image.py b/graphai/core/image/image.py
index b1df3066..2f8bd7b7 100644
--- a/graphai/core/image/image.py
+++ b/graphai/core/image/image.py
@@ -183,17 +183,17 @@ def break_pdf_into_images(token, file_manager):
     if not file_exists(pdf_path):
         print(f'Error: File {pdf_path} does not exist')
         return None
-    pdf_doc = pymupdf.open(pdf_path)
     output_filenames = list()
-    for page in pdf_doc:
-        i = page.number
-        img_dir = file_manager.generate_filepath(token.replace('.', '__') + f'/page_{i}.png')
-        pix = page.get_pixmap()
-        pix.save(img_dir)
-        output_filenames.append({
-            'page': i + 1,
-            'filename': img_dir
-        })
+    with pymupdf.open(pdf_path) as pdf_doc:
+        for page in pdf_doc:
+            i = page.number
+            img_dir = file_manager.generate_filepath(token.replace('.', '__') + f'/page_{i}.png')
+            pix = page.get_pixmap()
+            pix.save(img_dir)
+            output_filenames.append({
+                'page': i + 1,
+                'filename': img_dir
+            })
     return output_filenames