Update app.py

app.py

@@ -40,7 +40,8 @@ CONFIG = {
     "PUBLIC_URL_BASE": "https://huggingface.co/spaces/PrashanthB461/AI_Safety_Demo2/resolve/main/static/output/",
     "FRAME_SKIP": 1,  # Process every frame (set to 1 for full video processing)
     "MAX_PROCESSING_TIME": 30,
-    "CONFIDENCE_THRESHOLD": 0.5
+    "CONFIDENCE_THRESHOLD": 0.5,
+    "TEMPORAL_THRESHOLD": 1.0  # Time threshold in seconds to avoid counting the same violation
 }
 
 # Setup logging
@@ -226,6 +227,7 @@ def process_video(video_data):
         fps = video.get(cv2.CAP_PROP_FPS)
 
         snapshot_taken = {label: False for label in CONFIG["VIOLATION_LABELS"].values()}
+        last_detected = {label: 0 for label in CONFIG["VIOLATION_LABELS"].values()}  # Tracks last violation time
 
         while True:
             ret, frame = video.read()
@@ -254,10 +256,12 @@ def process_video(video_data):
                     if label not in CONFIG["VIOLATION_LABELS"].values() or conf < CONFIG["CONFIDENCE_THRESHOLD"]:
                         continue
                     
-                    if label in seen_violations:
-                        continue  # Avoid duplicates in the same frame
-                    
-                    seen_violations.add(label)
+                    # Skip if the same violation is detected again within the temporal threshold
+                    if time.time() - last_detected[label] < CONFIG["TEMPORAL_THRESHOLD"]:
+                        continue
+
+                    # Update last detected time
+                    last_detected[label] = time.time()
 
                     # Save the violation data
                     violation = {
@@ -389,4 +393,4 @@ interface = gr.Interface(
 
 if __name__ == "__main__":
     logger.info("Launching Safety Analyzer App...")
-    interface.launch()
+    interface.launch()