Update app.py

app.py

@@ -3,7 +3,7 @@ import base64
 import requests
 from PIL import Image
 from io import BytesIO
-from pdf2image import convert_from_bytes
+import fitz  # PyMuPDF
 
 # Configuration - Replace with your API key
 GEMINI_API_KEY = st.secrets["GEMINI_API_KEY"]
@@ -34,13 +34,13 @@ def encode_file(uploaded_file):
         file_bytes = uploaded_file.getvalue()
         
         if uploaded_file.type == "application/pdf":
-            # Specify the path to poppler binaries
-            poppler_path = "/usr/bin"  # Adjust this path if necessary
-            images = convert_from_bytes(file_bytes, first_page=1, last_page=1, poppler_path=poppler_path)
-            if not images:
-                raise ValueError("Failed to convert PDF to image")
+            # Convert PDF to image using PyMuPDF
+            pdf = fitz.open(stream=BytesIO(file_bytes))
+            page = pdf[0]  # Get the first page
+            pix = page.get_pixmap()
+            img = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
             img_byte_arr = BytesIO()
-            images[0].save(img_byte_arr, format='JPEG')
+            img.save(img_byte_arr, format='JPEG')
             return base64.b64encode(img_byte_arr.getvalue()).decode('utf-8')
         
         return base64.b64encode(file_bytes).decode('utf-8')
@@ -89,10 +89,11 @@ def process_document():
 
             # Generate preview
             if uploaded_file.type == "application/pdf":
-                # Specify the path to poppler binaries
-                poppler_path = "/usr/bin"  # Adjust this path if necessary
-                images = convert_from_bytes(uploaded_file.getvalue(), first_page=1, last_page=1, poppler_path=poppler_path)
-                st.session_state.doc_preview = images[0]
+                pdf = fitz.open(stream=BytesIO(uploaded_file.getvalue()))
+                page = pdf[0]  # Get the first page
+                pix = page.get_pixmap()
+                img = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
+                st.session_state.doc_preview = img
             else:
                 st.session_state.doc_preview = Image.open(uploaded_file)