Update app.py

app.py

@@ -9,8 +9,11 @@ from fpdf import FPDF
 
 # Convert PDFs to images
 def convert_pdf_to_images(pdf_path, dpi=300):
-    images = convert_from_path(pdf_path, dpi=dpi)
-    return [cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR) for image in images]
+    try:
+        images = convert_from_path(pdf_path, dpi=dpi, poppler_path="/usr/bin")
+        return [cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR) for image in images]
+    except Exception as e:
+        return f"Error converting PDF to images: {e}"
 
 # Align images
 def align_images(img1, img2):
@@ -49,6 +52,9 @@ def generate_comparison_pdf(original_pdf, edited_pdf):
     original_images = convert_pdf_to_images(original_pdf)
     edited_images = convert_pdf_to_images(edited_pdf)
 
+    if isinstance(original_images, str) or isinstance(edited_images, str):
+        return original_images if isinstance(original_images, str) else edited_images
+
     combined_images = []
     for orig_img, edit_img in zip(original_images, edited_images):
         aligned_img = align_images(orig_img, edit_img)
@@ -69,9 +75,11 @@ def generate_comparison_pdf(original_pdf, edited_pdf):
 
 # Gradio interface function
 def pdf_comparison(original_pdf, edited_pdf):
-    if not os.path.exists("outputs"):
-        os.makedirs("outputs")
-    result_path = generate_comparison_pdf(original_pdf.name, edited_pdf.name)
+    if original_pdf is None or edited_pdf is None:
+        return "Error: Both PDF files must be provided."
+    original_pdf_path = original_pdf.name
+    edited_pdf_path = edited_pdf.name
+    result_path = generate_comparison_pdf(original_pdf_path, edited_pdf_path)
     return result_path
 
 # Gradio app
@@ -87,4 +95,6 @@ interface = gr.Interface(
 )
 
 if __name__ == "__main__":
+    if not os.path.exists("outputs"):
+        os.makedirs("outputs")
     interface.launch()