Update utils.py

utils.py

@@ -1,40 +1,94 @@
 import matplotlib.pyplot as plt
 import numpy as np
+from latex import build_pdf
+import os
 
 def create_percentage_circle(percentage, title):
     """
     Create a circular progress bar for percentage visualization.
-    Args:
-        percentage: Value between 0 and 100.
-        title: Title for the plot.
-    Returns:
-        Path to saved image.
     """
     fig, ax = plt.subplots(figsize=(4, 4))
     ax.set_aspect("equal")
     
-    # Draw background circle
     circle = plt.Circle((0.5, 0.5), 0.4, color="lightgray")
     ax.add_patch(circle)
     
-    # Draw progress arc
     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)
     
-    # Add percentage text
     ax.text(0.5, 0.5, f"{percentage:.1f}%", ha="center", va="center", fontsize=20)
-    
-    # Add title
     ax.set_title(title, fontsize=12)
-    
-    # Remove axes
     ax.axis("off")
     
-    # Save image
     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
+    return output_path
+
+def generate_pdf_report(summary, screenshots, output_path):
+    """
+    Generate a PDF report using LaTeX with summary and embedded screenshots.
+    """
+    # Create LaTeX content
+    latex_content = r"""
+\documentclass[a4paper,12pt]{article}
+\usepackage[utf8]{inputenc}
+\usepackage{graphicx}
+\usepackage{geometry}
+\geometry{margin=1in}
+\usepackage{hyperref}
+\usepackage{caption}
+\usepackage{subcaption}
+
+% Preamble for document setup
+\title{Road Safety Analysis Report}
+\author{Generated by Road Safety AI}
+\date{\today}
+
+\begin{document}
+
+\maketitle
+
+\section*{Analysis Summary}
+\begin{itemize}
+    % Insert summary lines dynamically
+    """
+    for line in summary.split('\n'):
+        if line.strip():
+            latex_content += f"\item {line.strip()}\n"
+
+    latex_content += r"""
+\end{itemize}
+
+\section*{Incident Screenshots}
+\begin{figure}[h]
+    \centering
+"""
+
+    for i, screenshot in enumerate(screenshots[:5]):  # Limit to 5 images
+        latex_content += f"""
+    \begin{subfigure}[t]{0.45\textwidth}
+        \includegraphics[width=\textwidth]{{{screenshot}}}
+        \caption{Incident Frame {i+1}}
+    \end{subfigure}
+    """
+        if (i + 1) % 2 == 0 and i < len(screenshots) - 1:
+            latex_content += r" \hfill "
+
+    latex_content += r"""
+    \caption*{Captured incidents during video analysis}
+\end{figure}
+
+\end{document}
+"""
+
+    # Write LaTeX to a file
+    tex_file = "templates/report.tex"
+    with open(tex_file, "w") as f:
+        f.write(latex_content)
+
+    # Build PDF
+    build_pdf(pdf_file=output_path, tex_file=tex_file, clean_tex=True)