Update app.py

app.py

@@ -61,7 +61,7 @@ import re
 import gradio as gr
 import torch
 from transformers import DonutProcessor, VisionEncoderDecoderModel
-from pdf2image import convert_from_bytes
+from pdf2image import convert_from_path
 import io
 
 processor = DonutProcessor.from_pretrained("pacman2223/univ-docu-model-v3")
@@ -72,14 +72,17 @@ model.to(device)
 def pdf_to_images(pdf_file):
     if pdf_file is None:
         return None
-    images = convert_from_bytes(pdf_file.read())
-    return images
+    pdf_path = pdf_file.name  # Get the file path
+    return convert_from_path(pdf_path)
 
 def process_document(pdf_file, page_number, question):
     if pdf_file is None:
         return "Please upload a PDF file."
     
     images = pdf_to_images(pdf_file)
+    if images is None:
+        return "Failed to process the PDF file."
+    
     if page_number < 1 or page_number > len(images):
         return f"Invalid page number. The PDF has {len(images)} pages."
     
@@ -121,6 +124,8 @@ def update_page_slider(pdf_file):
     if pdf_file is None:
         return gr.Slider(minimum=1, maximum=1, value=1, step=1, label="Page Number")
     images = pdf_to_images(pdf_file)
+    if images is None:
+        return gr.Slider(minimum=1, maximum=1, value=1, step=1, label="Page Number")
     return gr.Slider(minimum=1, maximum=len(images), value=1, step=1, label="Page Number")
 
 with gr.Blocks() as demo: