Update app.py

app.py

@@ -235,7 +235,10 @@ def calculate_safety_score(violations):
         "no_harness": 30,
         "unsafe_posture": 20
     }
-    score = 100 - sum(penalties.get(v["violation"], 0) for v in violations)
+    total_penalty = sum(penalties.get(v["violation"], 0) for v in violations)
+    logger.info(f"Total Penalty: {total_penalty}")
+    score = 100 - total_penalty
+    logger.info(f"Calculated Score: {score}")
     return max(score, 0)
 
 def process_video(video_data):
@@ -255,7 +258,7 @@ def process_video(video_data):
         fps = video.get(cv2.CAP_PROP_FPS)
 
         snapshot_taken = {label: False for label in CONFIG["VIOLATION_LABELS"].values()}
-        workers = []  # List to track workers: [{"id": int, "violations": set(), "bbox": list, "last_frame": int}]
+        workers = []  # List to track workers
 
         while True:
             ret, frame = video.read()
@@ -276,6 +279,11 @@ def process_video(video_data):
                 for box in result.boxes:
                     cls, conf = int(box.cls), float(box.conf)
                     label = CONFIG["VIOLATION_LABELS"].get(cls, None)
+                    
+                    # Log detected violations
+                    if label:
+                        logger.info(f"Violation Detected: {label} with confidence: {conf}")
+
                     if label not in CONFIG["VIOLATION_LABELS"].values() or conf < CONFIG["CONFIDENCE_THRESHOLD"]:
                         continue
 
@@ -288,7 +296,7 @@ def process_video(video_data):
                         "frame": frame_count
                     })
 
-            # Assign detections to workers
+            # Process detections and workers
             for detection in current_detections:
                 matched_worker = None
                 max_iou = 0
@@ -310,19 +318,7 @@ def process_video(video_data):
                             "timestamp": detection["timestamp"],
                             "worker_id": matched_worker["id"]
                         })
-                        # Take snapshot if not already taken for this violation type
-                        if not snapshot_taken[detection["violation"]]:
-                            snapshot_path = os.path.join(CONFIG["OUTPUT_DIR"], f"snapshot_{frame_count}_{detection['violation']}.jpg")
-                            cv2.imwrite(snapshot_path, frame)
-                            with open(snapshot_path, "rb") as img_file:
-                                img_base64 = base64.b64encode(img_file.read()).decode('utf-8')
-                            snapshots.append({
-                                "violation": detection["violation"],
-                                "frame": frame_count,
-                                "snapshot_url": f"{CONFIG['PUBLIC_URL_BASE']}{os.path.basename(snapshot_path)}",
-                                "snapshot_base64": f"data:image/jpeg;base64,{img_base64}"
-                            })
-                            snapshot_taken[detection["violation"]] = True
+                        snapshot_taken[detection["violation"]] = True
                     matched_worker["bbox"] = detection["bounding_box"]
                     matched_worker["last_frame"] = frame_count
                 else:
@@ -342,19 +338,6 @@ def process_video(video_data):
                         "timestamp": detection["timestamp"],
                         "worker_id": worker_id
                     })
-                    # Take snapshot if not already taken for this violation type
-                    if not snapshot_taken[detection["violation"]]:
-                        snapshot_path = os.path.join(CONFIG["OUTPUT_DIR"], f"snapshot_{frame_count}_{detection['violation']}.jpg")
-                        cv2.imwrite(snapshot_path, frame)
-                        with open(snapshot_path, "rb") as img_file:
-                            img_base64 = base64.b64encode(img_file.read()).decode('utf-8')
-                        snapshots.append({
-                            "violation": detection["violation"],
-                            "frame": frame_count,
-                            "snapshot_url": f"{CONFIG['PUBLIC_URL_BASE']}{os.path.basename(snapshot_path)}",
-                            "snapshot_base64": f"data:image/jpeg;base64,{img_base64}"
-                        })
-                        snapshot_taken[detection["violation"]] = True
 
             frame_count += 1
 
@@ -444,7 +427,7 @@ def gradio_interface(video_file):
 interface = gr.Interface(
     fn=gradio_interface,
     inputs=gr.Video(label="Upload Site Video"),
-    outputs=[
+    outputs=[ 
         gr.Markdown(label="Detected Safety Violations"),
         gr.Textbox(label="Compliance Score"),
         gr.Markdown(label="Snapshots"),
@@ -458,4 +441,4 @@ interface = gr.Interface(
 
 if __name__ == "__main__":
     logger.info("Launching Safety Analyzer App...")
-    interface.launch()
+    interface.launch()