Update app.py

app.py

@@ -39,7 +39,6 @@ CONFIG = {
     },
     "PUBLIC_URL_BASE": "https://huggingface.co/spaces/PrashanthB461/AI_Safety_Demo1/resolve/main/static/output/",
     "FRAME_SKIP": 15,  # Process every 15th frame
-    "MAX_PROCESSING_TIME": 25,  # Cap video processing at 25s
     "CONFIDENCE_THRESHOLD": 0.5  # Minimum confidence for violation detection
 }
 
@@ -239,27 +238,20 @@ def process_video(video_data):
 
         violations, snapshots = [], []
         frame_count = 0
-        start_time = time.time()
         fps = video.get(cv2.CAP_PROP_FPS)
-        max_frames = int(60 * fps)  # Process up to 1 minute
 
         # Track one snapshot per violation type
         snapshot_taken = {"no_helmet": False, "no_harness": False, "unsafe_posture": False}
 
         while True:
             ret, frame = video.read()
-            if not ret or frame_count >= max_frames:
+            if not ret:
                 break
 
             if frame_count % CONFIG["FRAME_SKIP"] != 0:
                 frame_count += 1
                 continue
 
-            # Stop if processing time exceeds 25 seconds
-            if time.time() - start_time > CONFIG["MAX_PROCESSING_TIME"]:
-                logger.info("Processing time limit reached")
-                break
-
             results = model(frame, device=device)
             seen_violations = set()
             for result in results:
@@ -268,11 +260,9 @@ 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"Unexpected detection: {label} (cls: {cls}, conf: {conf}) - ignored")
                         continue
                     # Apply confidence threshold
                     if conf < CONFIG["CONFIDENCE_THRESHOLD"]:
-                        logger.info(f"Skipping low-confidence detection: {label} (conf: {conf})")
                         continue
                     if label in seen_violations:
                         continue
@@ -313,7 +303,8 @@ def process_video(video_data):
                 "snapshots": [],
                 "score": 100,
                 "salesforce_record_id": None,
-                "violation_details_url": ""
+                "violation_details_url": "",
+                "message": "No violations detected here."
             }
 
         score = calculate_safety_score(violations)
@@ -325,7 +316,8 @@ def process_video(video_data):
             "snapshots": snapshots,
             "score": score,
             "salesforce_record_id": report_id,
-            "violation_details_url": final_pdf_url
+            "violation_details_url": final_pdf_url,
+            "message": ""
         }
     except Exception as e:
         logger.error(f"Error processing video: {e}")
@@ -334,7 +326,8 @@ def process_video(video_data):
             "snapshots": [],
             "score": 100,
             "salesforce_record_id": None,
-            "violation_details_url": ""
+            "violation_details_url": "",
+            "message": "Error processing video."
         }
 
 # ==========================
@@ -348,14 +341,28 @@ def gradio_interface(video_file):
             video_data = f.read()
         result = process_video(video_data)
 
+        if result.get("message"):
+            # Show message (like "No violations detected here.")
+            return result["message"], f"Safety Score: {result['score']}%", "", "N/A", "N/A"
+
         violation_table = "No violations detected."
         if result["violations"]:
-            header = "| Violation        | Timestamp | Confidence | Bounding Box             | Violation Details       |\n"
-            separator = "|------------------|-----------|------------|--------------------------|-------------------------|\n"
+            header = "| Violation         | Timestamp | Confidence | Violation Details                 |\n"
+            separator = "|-------------------|-----------|------------|---------------------------------|\n"
             rows = []
             for v in result["violations"]:
                 display_name = CONFIG["DISPLAY_NAMES"].get(v["violation"], v["violation"])
-                row = f"| {display_name:<16} | {v['timestamp']:.2f}s  | {v['confidence']:.2f}      | {v['bounding_box']} | {result['violation_details_url']} |"
+                # Provide clearer human-readable violation explanation
+                if v["violation"] == "no_helmet":
+                    details = "Employee not wearing helmet"
+                elif v["violation"] == "no_harness":
+                    details = "Employee not wearing proper harness"
+                elif v["violation"] == "unsafe_posture":
+                    details = "Employee in unsafe posture/zone"
+                else:
+                    details = "Violation detected"
+
+                row = f"| {display_name:<17} | {v['timestamp']:.2f}s  | {v['confidence']:.2f}     | {details:<31} |"
                 rows.append(row)
             violation_table = header + separator + "\n".join(rows)