Update app.py

app.py

@@ -84,6 +84,11 @@ def highlight_visual_changes(orig_img, edit_img, mask, start_position):
 
     return overlay, visual_changes, position_counter
 
+# Sanitize text for PDF compatibility
+def sanitize_text(text):
+    """Sanitize text for FPDF by replacing unsupported characters."""
+    return text.encode('latin-1', errors='replace').decode('latin-1')
+
 # Generate separate PDFs for visual and text changes
 def generate_separate_pdfs(original_pdf, edited_pdf):
     original_images = convert_pdf_to_images(original_pdf)
@@ -122,10 +127,10 @@ def generate_separate_pdfs(original_pdf, edited_pdf):
         os.remove(temp_path)
     pdf_visual.add_page()
     pdf_visual.set_font("Arial", size=12)
-    pdf_visual.cell(0, 10, "Visual Changes", ln=True, align="C")
+    pdf_visual.cell(0, 10, sanitize_text("Visual Changes"), ln=True, align="C")
     pdf_visual.ln(10)
     for _, change in visual_changes:
-        pdf_visual.cell(0, 10, change, ln=True)
+        pdf_visual.cell(0, 10, sanitize_text(change), ln=True)
     pdf_visual.output(visual_pdf_path)
 
     # Generate Text Changes PDF
@@ -139,10 +144,10 @@ def generate_separate_pdfs(original_pdf, edited_pdf):
         os.remove(temp_path)
     pdf_text.add_page()
     pdf_text.set_font("Arial", size=12)
-    pdf_text.cell(0, 10, "Text Changes", ln=True, align="C")
+    pdf_text.cell(0, 10, sanitize_text("Text Changes"), ln=True, align="C")
     pdf_text.ln(10)
     for _, change in text_changes:
-        pdf_text.cell(0, 10, change, ln=True)
+        pdf_text.cell(0, 10, sanitize_text(change), ln=True)
     pdf_text.output(text_pdf_path)
 
     return visual_pdf_path, text_pdf_path