Update app.py

app.py

@@ -8,10 +8,17 @@ from huggingface_hub import snapshot_download
 import pytesseract
 import io
 import math
+import os
+import re
+import tempfile
 from collections import defaultdict
 import matplotlib
 matplotlib.use("Agg")
 import matplotlib.pyplot as plt
+from reportlab.lib.pagesizes import letter
+from reportlab.lib.units import inch
+from reportlab.lib.styles import getSampleStyleSheet
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Image as RLImage
 
 _msi_model = None
 
@@ -240,6 +247,8 @@ def identify_hotspots(saliency_map, img_h, img_w, top_n=4):
                        "intensity": round(avg_intensity * 100, 1), "cx": int(cx), "cy": int(cy)})
     zones.sort(key=lambda z: -z["intensity"])
     top_zones = zones[:top_n]
+    for idx, z in enumerate(top_zones):
+        z["intensity_rank"] = idx + 1
     top_zones.sort(key=lambda z: (z["cy"], z["cx"]))
     return top_zones
 
@@ -397,6 +406,76 @@ def draw_issue_markers(overlay_bgr, issue_boxes):
     return img
 
 
+def markdown_to_paragraphs(md_text, styles):
+    story = []
+    for raw_line in md_text.split("\n"):
+        line = raw_line.strip()
+        if not line:
+            story.append(Spacer(1, 6))
+            continue
+        if line.startswith("## "):
+            story.append(Paragraph(line[3:], styles['Heading1']))
+        elif line.startswith("### "):
+            story.append(Paragraph(line[4:], styles['Heading2']))
+        elif line.startswith("---"):
+            story.append(Spacer(1, 10))
+        elif line.startswith("- ") or line.startswith("* "):
+            text = line[2:]
+            text = re.sub(r'\*\*(.*?)\*\*', r'<b>\1</b>', text)
+            text = re.sub(r'\*(.*?)\*', r'<i>\1</i>', text)
+            story.append(Paragraph("&bull; " + text, styles['Normal']))
+        elif re.match(r'^\d+\.\s', line):
+            num, text = line.split('.', 1)
+            text = re.sub(r'\*\*(.*?)\*\*', r'<b>\1</b>', text.strip())
+            story.append(Paragraph(f"{num}. {text}", styles['Normal']))
+        else:
+            text = re.sub(r'\*\*(.*?)\*\*', r'<b>\1</b>', line)
+            text = re.sub(r'\*(.*?)\*', r'<i>\1</i>', text)
+            text = re.sub(r'\[([^\]]+)\]\(([^)]+)\)', r'<link href="\2"><u>\1</u></link>', text)
+            story.append(Paragraph(text, styles['Normal']))
+    return story
+
+
+def export_pdf(heatmap_img, chart_img, summary_md):
+    if heatmap_img is None or not summary_md:
+        return None
+    styles = getSampleStyleSheet()
+    tmp_dir = tempfile.gettempdir()
+    pdf_path = os.path.join(tmp_dir, "design_analysis_report.pdf")
+    doc = SimpleDocTemplate(pdf_path, pagesize=letter, topMargin=0.6 * inch, bottomMargin=0.6 * inch,
+                             leftMargin=0.7 * inch, rightMargin=0.7 * inch)
+    story = [Paragraph("Design Analyzer Report", styles['Title']), Spacer(1, 12)]
+
+    max_width = 6.1 * inch
+    heatmap_path = os.path.join(tmp_dir, "heatmap_export_temp.png")
+    heatmap_img.save(heatmap_path)
+    ratio = heatmap_img.height / heatmap_img.width
+    story.append(RLImage(heatmap_path, width=max_width, height=max_width * ratio))
+    story.append(Spacer(1, 14))
+
+    if chart_img is not None:
+        chart_path = os.path.join(tmp_dir, "chart_export_temp.png")
+        chart_img.save(chart_path)
+        chart_ratio = chart_img.height / chart_img.width
+        story.append(RLImage(chart_path, width=max_width, height=max_width * chart_ratio))
+        story.append(Spacer(1, 14))
+
+    story.extend(markdown_to_paragraphs(summary_md, styles))
+    doc.build(story)
+    return pdf_path
+
+
+def draw_scan_path(overlay_bgr, zones):
+    img = overlay_bgr.copy()
+    ordered = sorted(zones, key=lambda z: z["intensity_rank"])
+    for i in range(len(ordered) - 1):
+        pt1 = (ordered[i]["cx"], ordered[i]["cy"])
+        pt2 = (ordered[i + 1]["cx"], ordered[i + 1]["cy"])
+        cv2.arrowedLine(img, pt1, pt2, (255, 255, 255), 4, tipLength=0.06, line_type=cv2.LINE_AA)
+        cv2.arrowedLine(img, pt1, pt2, (20, 20, 20), 2, tipLength=0.06, line_type=cv2.LINE_AA)
+    return img
+
+
 def draw_zone_labels(overlay_bgr, zones):
     img = overlay_bgr.copy()
     for i, z in enumerate(zones, 1):
@@ -429,6 +508,7 @@ def analyze(pil_image, overlay_strength):
     tips = generate_recommendations(saliency_map, scores, read, h, w)
     chart_image = create_score_chart(scores['focus_score'], scores['spread_score'], readability_score)
 
+    overlay_bgr = draw_scan_path(overlay_bgr, zones)
     overlay_bgr = draw_zone_labels(overlay_bgr, zones)
     overlay_bgr = draw_issue_markers(overlay_bgr, read.get("issue_boxes", []))
     overlay_rgb = cv2.cvtColor(overlay_bgr, cv2.COLOR_BGR2RGB)
@@ -499,9 +579,10 @@ with gr.Blocks(title="Design Analyzer") as demo:
         "screens, banner ads, print, packaging.\n"
         "- **Attention + readability in one pass**, not two separate tools.\n"
         "- **No inflated claims.** A cited, transparent process instead "
-        "of a marketing number.\n\n"
-        "Coming soon: exportable PDF reports, before/after comparisons, "
-        "and attention scan-path visualization."
+        "of a marketing number.\n"
+        "- **Scan-path arrows** show the predicted order attention flows, "
+        "not just isolated hotspots.\n\n"
+        "Coming soon: before/after comparisons."
     )
     with gr.Row():
         with gr.Column():
@@ -512,6 +593,9 @@ with gr.Blocks(title="Design Analyzer") as demo:
             image_output = gr.Image(label="Heatmap result")
             chart_output = gr.Image(label="Scores at a glance")
             text_output = gr.Markdown()
+            pdf_btn = gr.Button("Download PDF Report")
+            pdf_output = gr.File(label="PDF Report")
     analyze_btn.click(fn=analyze, inputs=[image_input, strength_slider], outputs=[image_output, chart_output, text_output])
+    pdf_btn.click(fn=export_pdf, inputs=[image_output, chart_output, text_output], outputs=pdf_output)
 
 demo.launch()