Update app.py

app.py

@@ -13,6 +13,7 @@ from io import BytesIO
 import base64
 import logging
 from retrying import retry
+import uuid
 
 # ==========================
 # Configuration
@@ -24,7 +25,7 @@ CONFIG = {
         0: "no_helmet",
         1: "no_harness",
         2: "unsafe_posture",
-        3: "unsafe_zone"  # Ignored in scoring and table
+        3: "unsafe_zone"  # Ignored in processing
     },
     "DISPLAY_NAMES": {  # Mapping for user-friendly violation names
         "no_helmet": "Missing Helmet",
@@ -108,15 +109,18 @@ def generate_violation_pdf(violations, score):
         y_position -= 0.3 * inch
         c.drawString(1 * inch, y_position, "Violation Details:")
         y_position -= 0.3 * inch
-        for v in violations:
-            display_name = CONFIG["DISPLAY_NAMES"].get(v["violation"], v["violation"])
-            text = f"{display_name} at {v['timestamp']:.2f}s (Confidence: {v['confidence']})"
-            c.drawString(1 * inch, y_position, text)
-            y_position -= 0.3 * inch
-            if y_position < 1 * inch:
-                c.showPage()
-                c.setFont("Helvetica", 10)
-                y_position = 10 * inch
+        if not violations:
+            c.drawString(1 * inch, y_position, "No violations detected.")
+        else:
+            for v in violations:
+                display_name = CONFIG["DISPLAY_NAMES"].get(v["violation"], v["violation"])
+                text = f"{display_name} at {v['timestamp']:.2f}s (Confidence: {v['confidence']})"
+                c.drawString(1 * inch, y_position, text)
+                y_position -= 0.3 * inch
+                if y_position < 1 * inch:
+                    c.showPage()
+                    c.setFont("Helvetica", 10)
+                    y_position = 10 * inch
 
         c.showPage()
         c.save()
@@ -260,7 +264,7 @@ def process_video(video_data):
                     label = CONFIG["VIOLATION_LABELS"].get(cls, f"unknown_class_{cls}")
                     # Only process specified violations
                     if label not in ["no_helmet", "no_harness", "unsafe_posture"]:
-                        logger.warning(f"Ignoring detection: {label} (cls: {cls}, conf: {conf}) - not a target violation")
+                        logger.info(f"Ignoring detection: {label} (cls: {cls}, conf: {conf}) - not a target violation")
                         continue
                     # Apply confidence threshold
                     if conf < CONFIG["CONFIDENCE_THRESHOLD"]: