| import matplotlib.pyplot as plt |
| import numpy as np |
| from reportlab.lib.pagesizes import letter |
| from reportlab.pdfgen import canvas |
| from reportlab.lib.utils import ImageReader |
| import os |
| from datetime import datetime |
|
|
| def create_percentage_circle(percentage, title): |
| """ |
| Create a circular progress bar for percentage visualization. |
| """ |
| fig, ax = plt.subplots(figsize=(4, 4)) |
| ax.set_aspect("equal") |
| |
| circle = plt.Circle((0.5, 0.5), 0.4, color="lightgray") |
| ax.add_patch(circle) |
| |
| theta = np.linspace(0, 360 * (percentage / 100), 100) |
| x = 0.5 + 0.4 * np.cos(np.radians(theta)) |
| y = 0.5 + 0.4 * np.sin(np.radians(theta)) |
| ax.plot(x, y, color="blue", linewidth=20) |
| |
| ax.text(0.5, 0.5, f"{percentage:.1f}%", ha="center", va="center", fontsize=20) |
| ax.set_title(title, fontsize=12) |
| ax.axis("off") |
| |
| output_path = f"screenshots/circle_{int(percentage)}_{int(np.random.randint(1000))}.png" |
| plt.savefig(output_path, bbox_inches="tight") |
| plt.close() |
| |
| return output_path |
|
|
| def generate_pdf_report(summary, screenshots, output_path): |
| """ |
| Generate a PDF report using reportlab with summary and embedded screenshots. |
| """ |
| c = canvas.Canvas(output_path, pagesize=letter) |
| width, height = letter |
|
|
| |
| c.setFont("Helvetica", 16) |
| c.drawString(100, height - 50, "Road Safety Analysis Report") |
| c.setFont("Helvetica", 12) |
| c.drawString(100, height - 70, f"Generated on: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}") |
|
|
| |
| c.setFont("Helvetica", 12) |
| c.drawString(100, height - 100, "Analysis Summary:") |
| y = height - 120 |
| for line in summary.split('\n'): |
| if line.strip(): |
| c.drawString(120, y, line.strip()) |
| y -= 20 |
| if y < 100: |
| c.showPage() |
| y = height - 50 |
|
|
| |
| if screenshots: |
| c.showPage() |
| c.setFont("Helvetica", 12) |
| c.drawString(100, height - 50, "Incident Screenshots:") |
| y = height - 70 |
| for i, screenshot in enumerate(screenshots[:5]): |
| if os.path.exists(screenshot): |
| img = ImageReader(screenshot) |
| img_width = 200 |
| img_height = img_width * (img.getSize()[1] / img.getSize()[0]) |
| if y - img_height < 50: |
| c.showPage() |
| y = height - 50 |
| c.drawImage(screenshot, 100, y - img_height, width=img_width, height=img_height) |
| c.drawString(100, y - img_height - 20, f"Incident Frame {i+1}") |
| y -= img_height + 30 |
|
|
| c.save() |
