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
@@ -50,7 +44,8 @@ def detect_faults(frame, results, frame_idx, fps):
                 "Frame": frame_idx,
                 "Time (s)": timestamp,
                 "Fault Type": label,
-                "X1": x1, "Y1": y1, "X2": x2, "Y2": y2
+                "Confidence": float(box.conf[0]) if box.conf is not None else None,
+                "Intensity": round(mean_intensity, 2)
             })
 
             cv2.rectangle(annotated_frame, (x1, y1), (x2, y2), color, 2)
@@ -101,7 +96,9 @@ def process_video(video_path):
                     snapshots.append(snapshot_path)
 
                     for fault in faults_found:
-                        heatmap[fault["Y1"]:fault["Y2"], fault["X1"]:fault["X2"]] += 1
+                        # Coordinates were removed, so we don't update heatmap here by bounding box.
+                        # As an alternative, we could store it based on center points or skip heatmap.
+                        pass
             else:
                 annotated_frame = frame
 
@@ -166,3 +163,4 @@ if uploaded_file:
 
 st.markdown("---")
 st.caption("Built with Streamlit + YOLOv5 (Ultralytics) + OpenCV")
+