Update app.py

app.py

@@ -61,34 +61,51 @@ def compare_images(img1, img2):
 def highlight_changes(img, mask):
     overlay = img.copy()
     contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-    for cnt in contours:
+    summary = []  # To store details of changes
+    for idx, cnt in enumerate(contours, start=1):
         if cv2.contourArea(cnt) > 100:  # Filter based on area to reduce false positives
             x, y, w, h = cv2.boundingRect(cnt)
             cv2.rectangle(overlay, (x, y), (x + w, y + h), (0, 0, 255), 2)  # Red for changes
-    return overlay
+            # Add change details to the summary
+            summary.append(f"Change {idx}: Area at (x={x}, y={y}, width={w}, height={h})")
+    return overlay, summary
 
-# Generate comparison PDF
+# Generate comparison PDF with summary
 def generate_comparison_pdf(original_pdf, edited_pdf):
     original_images = convert_pdf_to_images(original_pdf)
     edited_images = convert_pdf_to_images(edited_pdf)
     combined_images = []
+    summary = []  # Summary of all changes
     for orig_img, edit_img in zip(original_images, edited_images):
         aligned_img = align_images(orig_img, edit_img)
         diff_mask = compare_images(orig_img, aligned_img)
-        highlighted_img = highlight_changes(edit_img, diff_mask)
+        highlighted_img, page_summary = highlight_changes(edit_img, diff_mask)
+        summary.extend(page_summary)  # Collect all changes
         # Ensure dimensions match
         height = min(orig_img.shape[0], highlighted_img.shape[0])
         orig_img_resized = orig_img[:height]
         highlighted_img_resized = highlighted_img[:height]
         combined_images.append(np.hstack((orig_img_resized, highlighted_img_resized)))
+
     output_path = "outputs/comparison_result.pdf"
     pdf = FPDF()
+
+    # Add each comparison image to the PDF
     for img in combined_images:
         temp_path = "temp_image.png"
         cv2.imwrite(temp_path, img)
         pdf.add_page()
         pdf.image(temp_path, x=10, y=10, w=190)
         os.remove(temp_path)
+
+    # Add summary page to the PDF
+    pdf.add_page()
+    pdf.set_font("Arial", size=12)
+    pdf.cell(0, 10, "Summary of Changes", ln=True, align="C")
+    pdf.ln(10)  # Add a line break
+    for idx, change in enumerate(summary, start=1):
+        pdf.cell(0, 10, f"{idx}. {change}", ln=True)
+
     pdf.output(output_path)
     return output_path
 
@@ -114,6 +131,8 @@ interface = gr.Interface(
         gr.File(label="Upload Edited PDF", file_types=[".pdf"])
     ],
     outputs=gr.File(label="Download Comparison Report"),
+    title="PDF Comparison Tool with Summary",
+    description="Upload two PDFs: the original and the edited version. The tool will highlight changes and provide a summary in the output PDF."
 )
 
 if __name__ == "__main__":