Update app.py

app.py

@@ -50,18 +50,18 @@ CONFIG = {
         "domain": "login"
     },
     "PUBLIC_URL_BASE": "https://huggingface.co/spaces/PrashanthB461/AI_Safety_Demo2/resolve/main/static/output/",
-    "FRAME_SKIP": 50,  # Increased to process fewer frames (1 frame every 5 frames)
+    "FRAME_SKIP": 5,
     "CONFIDENCE_THRESHOLDS": {
-        "no_helmet": 0.5,
+        "no_helmet": 0.6,  # Increased to reduce false positives
         "no_harness": 0.15,
         "unsafe_posture": 0.15,
         "unsafe_zone": 0.15,
         "improper_tool_use": 0.15
     },
     "IOU_THRESHOLD": 0.4,
-    "MIN_VIOLATION_FRAMES":2,  # Reduced to ensure violations are detected with fewer frames
-    "HELMET_CONFIDENCE_THRESHOLD": 0.7,
-    "MAX_PROCESSING_TIME": 120  # Enforce 30-second processing limit
+    "MIN_VIOLATION_FRAMES": 2,
+    "HELMET_CONFIDENCE_THRESHOLD": 0.65,  # Lowered to mark more workers as compliant
+    "MAX_PROCESSING_TIME": 30
 }
 
 # Setup logging
@@ -379,7 +379,9 @@ def process_video(video_data):
                         helmet_compliance[worker_id]["no_helmet_frames"] += 1
                         if detection["confidence"] < CONFIG["HELMET_CONFIDENCE_THRESHOLD"]:
                             helmet_compliance[worker_id]["compliant"] = True
-                            logger.debug(f"Worker {worker_id} marked as helmet compliant due to low no_helmet confidence")
+                            logger.debug(f"Worker {worker_id} marked as helmet compliant due to low no_helmet confidence: {detection['confidence']}")
+                        else:
+                            logger.debug(f"Worker {worker_id} potential no_helmet violation with confidence: {detection['confidence']}")
                         if helmet_compliance[worker_id]["compliant"]:
                             logger.debug(f"Worker {worker_id} has helmet, skipping no_helmet violation")
                             continue
@@ -439,9 +441,8 @@ def process_video(video_data):
                         if worker_id in helmet_compliance and helmet_compliance[worker_id]["compliant"]:
                             logger.debug(f"Skipping no_helmet for worker {worker_id} due to helmet compliance")
                             continue
-                        if helmet_compliance[worker_id]["no_helmet_frames"] < CONFIG["MIN_VIOLATION_FRAMES"] * 2:
-                            logger.debug(f"Skipping no_helmet for worker {worker_id}, not enough persistent detections")
-                            continue
+                        # Removed the stricter persistence check for no_helmet
+                        logger.info(f"Confirmed no_helmet for worker {worker_id} with {len(worker_dets)} detections")
                     
                     best_detection = max(worker_dets, key=lambda x: x["confidence"])
                     violations.append(best_detection)