Upload app.py

app.py

@@ -489,14 +489,11 @@ def highlight_pdf(file_path, text_to_highlight, page_numbers):
 def pdf_to_images(pdf_path, page_numbers):
     doc = pymupdf.open(pdf_path)
     images = []
-    for page_number in page_numbers:
+    for page_number in range(1, len(page_numbers)+1):
         page = doc.load_page(page_number - 1)
         pix = page.get_pixmap()
         img = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
-        buf = io.BytesIO()
-        img.save(buf, format="PNG")
-        byte_im = buf.getvalue()
-        images.append(byte_im)
+        images.append(img)
     return images
 
 # Function to display PDF in Streamlit
@@ -517,7 +514,10 @@ def display_highlighted_pdf():
     images = pdf_to_images(highlighted_pdf_path, sources)
 
     for img in images:
-        image_zoom(img)
+        if isinstance(img, Image.Image):  # Ensure img is a Pillow Image object
+            image_zoom(img)
+        else:
+            st.error("The provided image is not a valid Pillow Image object.")
 
 display_highlighted_pdf()