Update app.py

app.py

@@ -40,7 +40,7 @@ CONFIG = {
     "PUBLIC_URL_BASE": "https://huggingface.co/spaces/PrashanthB461/AI_Safety_Demo2/resolve/main/static/output/",
     "FRAME_SKIP": 15,
     "MAX_PROCESSING_TIME": 30,
-    "CONFIDENCE_THRESHOLD": 0.25,  # Lowered for better detection
+    "CONFIDENCE_THRESHOLD": 0.1,  # Lowered for debugging
     "IOU_THRESHOLD": 0.5  # For worker tracking
 }
 
@@ -80,12 +80,19 @@ def calculate_iou(box1, box2):
     x1, y1, w1, h1 = box1
     x2, y2, w2, h2 = box2
     
+    # Convert to top-left and bottom-right coordinates
     x1_min, y1_min = x1 - w1/2, y1 - h1/2
     x1_max, y1_max = x1 + w1/2, y1 + h1/2
     x2_min, y2_min = x2 - w2/2, y2 - h2/2
     x2_max, y2_max = x2 + w2/2, y2 + h2/2
     
-    intersection = max(0, x2_min - x1_max) * max(0, y2_min - y1_max)
+    # Calculate intersection
+    x_min = max(x1_min, x2_min)
+    y_min = max(y1_min, y2_min)
+    x_max = min(x1_max, x2_max)
+    y_max = min(y1_max, y2_max)
+    
+    intersection = max(0, x_max - x_min) * max(0, y_max - y_min)
     area1 = w1 * h1
     area2 = w2 * h2
     union = area1 + area2 - intersection
@@ -245,7 +252,7 @@ def process_video(video_data):
         if not video.isOpened():
             raise ValueError("Could not open video file")
 
-        violations, snapshots = [], []
+        violations, snapshots, raw_detections = [], [], []
         frame_count = 0
         start_time = time.time()
         fps = video.get(cv2.CAP_PROP_FPS)
@@ -280,7 +287,15 @@ def process_video(video_data):
                     cls, conf = int(box.cls), float(box.conf)
                     label = CONFIG["VIOLATION_LABELS"].get(cls, None)
                     
-                    logger.info(f"Detection: class={cls}, conf={conf:.2f}, label={label}")
+                    # Log all raw detections
+                    logger.info(f"Raw Detection: class={cls}, conf={conf:.2f}, label={label}")
+                    raw_detections.append({
+                        "frame": frame_count,
+                        "class": cls,
+                        "confidence": round(conf, 2),
+                        "label": label if label in CONFIG["VIOLATION_LABELS"].values() else "unknown",
+                        "timestamp": frame_count / fps
+                    })
                     
                     if label not in CONFIG["VIOLATION_LABELS"].values():
                         logger.info(f"Skipping unknown class: {cls}")
@@ -394,6 +409,7 @@ def process_video(video_data):
             return {
                 "violations": [],
                 "snapshots": [],
+                "raw_detections": raw_detections,
                 "score": 100,
                 "salesforce_record_id": None,
                 "violation_details_url": "",
@@ -407,6 +423,7 @@ def process_video(video_data):
         return {
             "violations": violations,
             "snapshots": snapshots,
+            "raw_detections": raw_detections,
             "score": score,
             "salesforce_record_id": report_id,
             "violation_details_url": final_pdf_url,
@@ -417,6 +434,7 @@ def process_video(video_data):
         return {
             "violations": [],
             "snapshots": [],
+            "raw_detections": [],
             "score": 100,
             "salesforce_record_id": None,
             "violation_details_url": "",
@@ -425,9 +443,9 @@ def process_video(video_data):
 
 def gradio_interface(video_file):
     if not video_file:
-        return "No file uploaded.", "", "No file uploaded.", "", "", []
+        return "No file uploaded.", "", "No file uploaded.", "", "", [], "No raw detections."
     try:
-        yield "Processing video... please wait.", "", "", "", "", []
+        yield "Processing video... please wait.", "", "", "", "", [], "Processing..."
 
         with open(video_file, "rb") as f:
             video_data = f.read()
@@ -435,7 +453,7 @@ def gradio_interface(video_file):
         result = process_video(video_data)
 
         if result.get("message"):
-            yield result["message"], "", "", "", "", []
+            yield result["message"], "", "", "", "", [], "Error in processing."
             return
 
         violation_table = "No violations detected."
@@ -460,19 +478,30 @@ def gradio_interface(video_file):
             )
             snapshot_images = [s["snapshot_base64"] for s in result["snapshots"]]
 
+        raw_detections_text = "No raw detections logged."
+        if result["raw_detections"]:
+            header = "| Frame | Timestamp (s) | Class | Label          | Confidence |\n"
+            separator = "|-------|---------------|-------|----------------|------------|\n"
+            rows = []
+            for d in result["raw_detections"]:
+                row = f"| {d['frame']:<5} | {d['timestamp']:.2f}        | {d['class']:<5} | {d['label']:<14} | {d['confidence']:.2f}      |"
+                rows.append(row)
+            raw_detections_text = header + separator + "\n".join(rows)
+
         yield (
             violation_table,
             f"Safety Score: {result['score']}%",
             snapshots_text,
             f"Salesforce Record ID: {result['salesforce_record_id'] or 'N/A'}",
             result["violation_details_url"] or "N/A",
-            snapshot_images
+            snapshot_images,
+            raw_detections_text
         )
     except Exception as e:
         logger.error(f"Error in Gradio interface: {e}", exc_info=True)
-        yield f"Error: {str(e)}", "", "Error in processing.", "", "", []
+        yield f"Error: {str(e)}", "", "Error in processing.", "", "", [], "Error in processing."
 
-interface = gr.Interface(
+interface	beq gr.Interface(
     fn=gradio_interface,
     inputs=gr.Video(label="Upload Site Video"),
     outputs=[
@@ -481,7 +510,8 @@ interface = gr.Interface(
         gr.Markdown(label="Snapshots"),
         gr.Textbox(label="Salesforce Record ID"),
         gr.Textbox(label="Violation Details URL"),
-        gr.Gallery(label="Violation Snapshots")
+        gr.Gallery(label="Violation Snapshots"),
+        gr.Markdown(label="Raw Detections (Debug)")
     ],
     title="Worksite Safety Violation Analyzer",
     description="Upload site videos to detect safety violations (No Helmet, No Harness, Unsafe Posture). Non-violations are ignored.",