Update app.py

app.py

@@ -44,8 +44,8 @@ model = YOLO('./data/best.pt').to(device)
 if device == "cuda":
     model.half()
 
-# Add function to generate the PDF report with images and data
-def generate_pdf_report(summary: str, screenshots: List[str], pdf_path: str):
+# Function to generate the PDF report with images and data
+def generate_pdf_report(summary: str, screenshots: List[str], log_results: List[str], chart_path: str, map_path: str, pdf_path: str):
     c = canvas.Canvas(pdf_path, pagesize=letter)
     width, height = letter
 
@@ -58,13 +58,28 @@ def generate_pdf_report(summary: str, screenshots: List[str], pdf_path: str):
     c.drawString(72, height - 100, "Summary:")
     c.drawString(72, height - 120, summary)
 
-    # Add Screenshots
-    c.drawString(72, height - 160, "Incident Screenshots:")
+    # Add Logs for Top 5 Images
+    c.drawString(72, height - 160, "Log Results for Top 5 Images:")
     y_position = height - 180
+    for log in log_results[:5]:  # Display top 5 logs
+        c.drawString(72, y_position, log)
+        y_position -= 15
+
+    # Add Screenshots (Top 5 Images)
+    c.drawString(72, y_position - 20, "Incident Screenshots:")
+    y_position -= 40
     for screenshot in screenshots[:5]:  # Limit to the top 5 images
         c.drawImage(screenshot, 72, y_position, width=150, height=100)
         y_position -= 120
 
+    # Add Chart Image
+    c.drawString(72, y_position - 20, "Detection Trend Chart:")
+    c.drawImage(chart_path, 72, y_position - 100, width=400, height=300)
+
+    # Add Map Image
+    c.drawString(72, y_position - 140, "Issue Locations Map:")
+    c.drawImage(map_path, 72, y_position - 240, width=400, height=300)
+
     c.save()
 
 def zip_all_outputs(report_path: str, video_path: str, chart_path: str, map_path: str) -> str:
@@ -154,14 +169,16 @@ def update_metrics(detections: List[Dict[str, Any]]) -> Dict[str, Any]:
         "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     }
 
-def generate_line_chart() -> Optional[str]:
-    if not detected_counts:
+def generate_line_chart(detections: List[Dict[str, Any]]) -> Optional[str]:
+    if not detections:
         return None
     plt.figure(figsize=(4, 2))
-    plt.plot(detected_counts[-50:], marker='o', color='#FF8C00')
+    detection_times = [det["frame"] for det in detections]
+    detection_counts = [det["conf"] for det in detections]
+    plt.plot(detection_times, detection_counts, marker='o', color='#FF8C00')
     plt.title("Detections Over Time")
     plt.xlabel("Frame")
-    plt.ylabel("Count")
+    plt.ylabel("Confidence")
     plt.grid(True)
     plt.tight_layout()
     chart_path = os.path.join(OUTPUT_DIR, f"chart_{datetime.now().strftime('%Y%m%d_%H%M%S')}.png")
@@ -311,12 +328,12 @@ def process_video(video, resize_width=4000, resize_height=3000, frame_skip=5):
     log_entries.append(f"Output video: {output_frames} frames, {output_fps:.2f} FPS, {output_duration:.2f} seconds")
 
     # Generate chart and map
-    chart_path = generate_line_chart()
+    chart_path = generate_line_chart(all_detections)
     map_path = generate_map(gps_coordinates[-5:], all_detections)
 
     # Generate the report
     pdf_path = f"{OUTPUT_DIR}/drone_analysis_report_{datetime.now().strftime('%Y%m%d_%H%M%S')}.pdf"
-    generate_pdf_report("Drone analysis completed. Top 5 detections included.", detected_issues, pdf_path)
+    generate_pdf_report("Drone analysis completed. Top 5 detections included.", detected_issues, log_entries, chart_path, map_path, pdf_path)
 
     log_entries.append(f"Processing completed in {total_time:.2f} seconds")