Update app.py

app.py

@@ -38,6 +38,7 @@ def detect_faults(frame, results, frame_idx, fps):
             mean_intensity = np.mean(roi)
             area = (x2 - x1) * (y2 - y1)
 
+            # Fault classification logic
             if mean_intensity > 240:
                 label = "Burned Solar Panel" if area >= 10000 else "Burned Solar Cell"
                 color = (0, 0, 255)
@@ -51,25 +52,26 @@ def detect_faults(frame, results, frame_idx, fps):
                 continue
 
             timestamp = round(frame_idx / fps, 2)
-      faults_found.append({
-    "Frame": frame_idx,
-    "Time (s)": timestamp,
-    "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})"
-})
-
-
+            faults_found.append({
+                "Frame": frame_idx,
+                "Time (s)": timestamp,
+                "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})"
+            })
+
+            # Annotate frame
             cv2.rectangle(annotated_frame, (x1, y1), (x2, y2), color, 2)
             cv2.putText(annotated_frame, label, (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, 2)
 
     return annotated_frame, faults_found
 
+
 def process_video(video_path):
     cap = cv2.VideoCapture(video_path)
     if not cap.isOpened():