Update app.py

app.py

@@ -3,7 +3,7 @@ import streamlit as st
 from openai import OpenAI
 import os
 from dotenv import load_dotenv
-import pdf2image
+import fitz
 from PIL import Image
 import io
 import tempfile
@@ -14,15 +14,25 @@ load_dotenv()
 client = OpenAI(api_key=os.getenv('OPENAI_API_KEY'))
 
 def convert_pdf_to_images(pdf_file):
-    """Convert PDF to list of images"""
+    """Convert PDF to list of images using PyMuPDF"""
+    images = []
     with tempfile.NamedTemporaryFile(delete=False, suffix='.pdf') as tmp_file:
         tmp_file.write(pdf_file.getvalue())
         pdf_path = tmp_file.name
-    
-    images = pdf2image.convert_from_path(pdf_path)
+        
+        pdf_document = fitz.open(pdf_path)
+        for page_number in range(pdf_document.page_count):
+            page = pdf_document[page_number]
+            pix = page.get_pixmap()
+            img_data = pix.tobytes("png")
+            image = Image.open(io.BytesIO(img_data))
+            images.append(image)
+            
+        pdf_document.close()
     os.unlink(pdf_path)
     return images
 
+
 def format_response(text):
     """Format the analysis response with clean styling"""
     formatted_text = ""