Update app.py

app.py

@@ -1,20 +1,13 @@
 import os
 import gradio as gr
 from pdf2image import convert_from_path
+from PIL import Image
+import cv2
+import numpy as np
 from fpdf import FPDF
-import shutil
-import subprocess
 
-
-# Define path to Poppler (update if needed)
-POPPLER_PATH = "/usr/bin/"  # Adjust based on installation
-
-def check_poppler_installed():
-    """Check if Poppler utilities (pdfinfo) are installed."""
-    pdfinfo_path = shutil.which("pdfinfo")
-    if not pdfinfo_path:
-        raise EnvironmentError("Poppler-utils (pdfinfo) is not installed or not in PATH. "
-                               "Install it using 'apt-get install -y poppler-utils'.")
+# Define path to Poppler
+POPPLER_PATH = "/usr/bin/"
 
 def flatten_pdf_to_images(pdf_path, output_dir, dpi=300):
     """Convert PDF pages to high-resolution images."""
@@ -22,7 +15,7 @@ def flatten_pdf_to_images(pdf_path, output_dir, dpi=300):
         images = convert_from_path(pdf_path, dpi=dpi, poppler_path=POPPLER_PATH)
     except Exception as e:
         raise RuntimeError(f"Error during PDF flattening: {e}")
-    
+
     image_paths = []
     for i, image in enumerate(images):
         output_path = os.path.join(output_dir, f"page_{i + 1}.png")
@@ -30,29 +23,8 @@ def flatten_pdf_to_images(pdf_path, output_dir, dpi=300):
         image_paths.append(output_path)
     return image_paths
 
-def create_comparison_pdf(original_images, edited_images, output_pdf_path):
-    """Generate a placeholder comparison PDF."""
-    pdf = FPDF()
-    pdf.set_auto_page_break(auto=True, margin=15)
-    
-    for orig_img, edited_img in zip(original_images, edited_images):
-        pdf.add_page()
-        pdf.set_font("Arial", size=12)
-
-        # Add original image
-        pdf.cell(200, 10, txt="Original Image", ln=True)
-        pdf.image(orig_img, x=10, y=20, w=90)
-
-        # Add edited image
-        pdf.cell(200, 10, txt="Edited Image", ln=True)
-        pdf.image(edited_img, x=110, y=20, w=90)
-
-    pdf.output(output_pdf_path)
-
 def process_pdfs(original_pdf, edited_pdf):
     """Main function to process PDFs and generate output."""
-    check_poppler_installed()
-    
     output_dir = "./output"
     os.makedirs(output_dir, exist_ok=True)
 
@@ -60,20 +32,22 @@ def process_pdfs(original_pdf, edited_pdf):
     original_images = flatten_pdf_to_images(original_pdf, output_dir)
     edited_images = flatten_pdf_to_images(edited_pdf, output_dir)
 
-    # Generate comparison PDF
+    # Create comparison PDF
     comparison_pdf = os.path.join(output_dir, "comparison_result.pdf")
-    create_comparison_pdf(original_images, edited_images, comparison_pdf)
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_font("Arial", size=12)
+    pdf.cell(200, 10, txt="Comparison PDF Placeholder", ln=True)
+
+    pdf.output(comparison_pdf)
     return comparison_pdf
 
 def gradio_interface(original_pdf, edited_pdf):
     """Gradio interface function."""
-    try:
-        output_pdf = process_pdfs(original_pdf.name, edited_pdf.name)
-        return output_pdf
-    except Exception as e:
-        return str(e)
+    output_pdf = process_pdfs(original_pdf.name, edited_pdf.name)
+    return output_pdf
 
-# Gradio Interface
+# Updated Gradio Interface
 interface = gr.Interface(
     fn=gradio_interface,
     inputs=[
@@ -86,13 +60,4 @@ interface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    # Ensure poppler-utils is available before launching the app
-    try:
-        check_poppler_installed()
-    except EnvironmentError as e:
-        print(str(e))
-        print("Ensure Poppler is installed in the environment for this app to function.")
-        exit(1)
-
     interface.launch()
-