Update app.py

app.py

@@ -57,31 +57,21 @@ def compare_visual_changes(orig_img, edit_img):
 
     return overlay
 
-# Generate visual comparison report with configurable dimensions
-def generate_visual_report(images, output_path, page_width=210, page_height=297):
-    """
-    Generate a PDF report with visual changes.
-    :param images: List of images to include in the report.
-    :param output_path: Path to save the output PDF.
-    :param page_width: Width of the PDF page in millimeters (default is A4: 210mm).
-    :param page_height: Height of the PDF page in millimeters (default is A4: 297mm).
-    """
-    pdf = FPDF(format=(page_width, page_height))  # Set custom page size
+# Generate visual comparison report
+def generate_visual_report(images, output_path):
+    pdf = FPDF()
     for img in images:
         temp_path = "temp_image.png"
         cv2.imwrite(temp_path, img)
         pdf.add_page()
-        # Adjust image placement and scaling to fit the custom dimensions
-        img_width = min(page_width - 20, page_width)  # Allow small margins
-        img_height = img.shape[0] * (img_width / img.shape[1])  # Preserve aspect ratio
-        pdf.image(temp_path, x=10, y=10, w=img_width, h=min(img_height, page_height - 20))
+        pdf.image(temp_path, x=10, y=10, w=190)
         os.remove(temp_path)
 
     pdf.output(output_path)
     return output_path
 
 # Perform only visual comparison
-def generate_visual_comparison(original_pdf, edited_pdf, page_width=210, page_height=297):
+def generate_visual_comparison(original_pdf, edited_pdf):
     original_images = convert_pdf_to_images(original_pdf)
     edited_images = convert_pdf_to_images(edited_pdf)
 
@@ -91,16 +81,16 @@ def generate_visual_comparison(original_pdf, edited_pdf, page_width=210, page_he
         highlighted_img = compare_visual_changes(orig_img, aligned_img)
         visual_combined_images.append(np.hstack((orig_img, highlighted_img)))
 
-    # Generate visual changes report with custom dimensions
+    # Generate visual changes report
     visual_report_path = generate_visual_report(
-        visual_combined_images, "outputs/visual_changes.pdf", page_width, page_height
+        visual_combined_images, "outputs/visual_changes.pdf"
     )
 
     return visual_report_path
 
 # Gradio interface function
-def pdf_visual_comparison(original_pdf, edited_pdf, page_width, page_height):
-    visual_path = generate_visual_comparison(original_pdf.name, edited_pdf.name, page_width, page_height)
+def pdf_visual_comparison(original_pdf, edited_pdf):
+    visual_path = generate_visual_comparison(original_pdf.name, edited_pdf.name)
     return visual_path
 
 # Gradio interface
@@ -108,15 +98,13 @@ interface = gr.Interface(
     fn=pdf_visual_comparison,
     inputs=[
         gr.File(label="Upload Original PDF", file_types=[".pdf"]),
-        gr.File(label="Upload Edited PDF", file_types=[".pdf"]),
-        gr.Number(label="Page Width (mm)", value=210),
-        gr.Number(label="Page Height (mm)", value=297),
+        gr.File(label="Upload Edited PDF", file_types=[".pdf"])
     ],
     outputs=[
         gr.File(label="Download Visual Changes Report")
     ],
     title="PDF Visual Comparison Tool",
-    description="Upload two PDFs: the original and the edited version. Specify PDF dimensions and generate a visual changes report."
+    description="Upload two PDFs: the original and the edited version. The tool generates a visual changes report."
 )
 
 if __name__ == "__main__":