Update app.py

app.py

@@ -1,9 +1,3 @@
-# Full code integrating:
-# - Fault timestamps
-# - CSV export
-# - Clickable snapshots
-# - Heatmap overlay
-
 import streamlit as st
 import cv2
 import numpy as np
@@ -58,10 +52,6 @@ def detect_faults(frame, results, frame_idx, fps):
                 "Fault Type": label,
                 "Confidence": round(float(box.conf[0]), 2) if box.conf is not None else None,
                 "Intensity": round(mean_intensity, 2),
-                "X1": x1,
-                "Y1": y1,
-                "X2": x2,
-                "Y2": y2,
                 "Box": f"({x1}, {y1}, {x2}, {y2})"
             })
 
@@ -71,7 +61,6 @@ def detect_faults(frame, results, frame_idx, fps):
 
     return annotated_frame, faults_found
 
-
 def process_video(video_path):
     cap = cv2.VideoCapture(video_path)
     if not cap.isOpened():
@@ -115,7 +104,8 @@ def process_video(video_path):
                     snapshots.append(snapshot_path)
 
                     for fault in faults_found:
-                        heatmap[fault["Y1"]:fault["Y2"], fault["X1"]:fault["X2"]] += 1
+                        x1, y1, x2, y2 = map(int, fault["Box"].strip("()").split(","))
+                        heatmap[y1:y2, x1:x2] += 1
             else:
                 annotated_frame = frame