Update app.py

app.py

@@ -14,6 +14,7 @@ import base64
 from io import BytesIO
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
+import beautifulsoup4
 
 MODEL_NAME = "cmckinle/sdxl-flux-detector"
 LABELS = ["AI", "Real"]
@@ -201,18 +202,26 @@ def generate_pdf(accuracy, roc_score, report_html, confusion_matrix_plot):
     buffer = BytesIO()
     c = canvas.Canvas(buffer, pagesize=letter)
     
+    # Add content to PDF
     c.drawString(100, 750, f"Model Results")
     c.drawString(100, 730, f"Accuracy: {accuracy:.2f}")
     c.drawString(100, 710, f"ROC Score: {roc_score:.2f}")
     
     y_position = 690
-    for line in report_html.replace("<br>", "\n").splitlines():
-        if y_position < 50:
+    # Convert report_html to plain text if it's HTML content
+    from bs4 import BeautifulSoup
+    soup = BeautifulSoup(report_html, "html.parser")
+    report_text = soup.get_text()
+
+    # Add each line of the report text
+    for line in report_text.splitlines():
+        if y_position < 50:  # Create a new page if space runs out
             c.showPage()
             y_position = 750
         c.drawString(100, y_position, line.strip())
         y_position -= 20
 
+    # Save Confusion Matrix Plot as an Image and Add it to the PDF
     img_buffer = BytesIO()
     confusion_matrix_plot.savefig(img_buffer, format="png")
     img_buffer.seek(0)
@@ -222,6 +231,7 @@ def generate_pdf(accuracy, roc_score, report_html, confusion_matrix_plot):
     buffer.seek(0)
     return buffer
 
+
 def load_url(url):
     try:
         urllib.request.urlretrieve(url, "temp_image.png")