Update app.py

app.py

@@ -64,8 +64,8 @@ if device == "cpu":  # Optional check, not strictly needed but can help to ensur
     model.float()  # Ensure model is running on the CPU
 print(f"Model classes: {model.names}")
 
-# Generate PDF report
-def generate_pdf_report(log_entries, detected_issues, chart_path, map_path, metrics):
+# Generate PDF report with Top 5 Images
+def generate_pdf_report(log_entries, detected_issues, chart_path, map_path, metrics, top_images):
     pdf = FPDF()
     pdf.set_auto_page_break(auto=True, margin=15)
     pdf.add_page()
@@ -92,6 +92,13 @@ def generate_pdf_report(log_entries, detected_issues, chart_path, map_path, metr
     pdf.multi_cell(0, 10, txt=json.dumps(metrics, indent=2))
     pdf.ln(5)
 
+    # Add Top 5 Images
+    pdf.cell(200, 10, txt="Top 5 Detected Images:", ln=True)
+    for image_path in top_images:
+        if os.path.exists(image_path):
+            pdf.image(image_path, x=10, y=pdf.get_y(), w=180)
+            pdf.ln(60)  # Space out the images
+
     # Add Chart Image
     if chart_path:
         pdf.cell(200, 10, txt="Detection Trend Chart:", ln=True)
@@ -207,6 +214,8 @@ def process_video(video, resize_width=4000, resize_height=3000, frame_skip=5):
         "metrics": {}
     }
 
+    top_images = []  # Track top 5 images
+
     while True:
         ret, frame = cap.read()
         if not ret:
@@ -262,6 +271,8 @@ def process_video(video, resize_width=4000, resize_height=3000, frame_skip=5):
             if detection_frame_count % SAVE_IMAGE_INTERVAL == 0:
                 captured_frame_path = os.path.join(CAPTURED_FRAMES_DIR, f"detected_{frame_count:06d}.jpg")
                 if cv2.imwrite(captured_frame_path, annotated_frame):
+                    if len(top_images) < 5:  # Keep track of only top 5 images
+                        top_images.append(captured_frame_path)
                     if write_geotag(captured_frame_path, gps_coord):
                         detected_issues.append(captured_frame_path)
                         data_lake_submission["images"].append({
@@ -357,7 +368,7 @@ def process_video(video, resize_width=4000, resize_height=3000, frame_skip=5):
     logs_zip = zip_directory(FLIGHT_LOG_DIR, os.path.join(OUTPUT_DIR, "flight_logs.zip"))
 
     # Generate PDF report
-    pdf_path = generate_pdf_report(log_entries, detected_issues, chart_path, map_path, last_metrics)
+    pdf_path = generate_pdf_report(log_entries, detected_issues, chart_path, map_path, last_metrics, top_images)
 
     return (
         output_path,