Update app.py

app.py

@@ -9,6 +9,8 @@ from services.anomaly_service import track_anomalies, predict_anomaly
 from models.solar_model import load_solar_model
 from models.windmill_model import load_windmill_model
 from config.settings import VIDEO_FOLDER
+from PIL import Image
+import io
 
 # Custom CSS for styling to match the screenshot
 st.markdown(
@@ -19,7 +21,6 @@ st.markdown(
         font-size: 24px;
         font-weight: bold;
         color: #333;
-F;
     }
     .status {
         text-align: center;
@@ -50,6 +51,11 @@ F;
         color: #333;
         margin-bottom: 5px;
     }
+    .snapshot-img {
+        max-width: 100%;
+        height: auto;
+        margin-bottom: 10px;
+    }
     </style>
     """,
     unsafe_allow_html=True
@@ -64,6 +70,13 @@ if 'frame_numbers' not in st.session_state:
     st.session_state.frame_numbers = []
 if 'total_detected' not in st.session_state:
     st.session_state.total_detected = 0
+if 'snapshots' not in st.session_state:
+    st.session_state.snapshots = []
+
+# Create snapshots directory if it doesn't exist
+SNAPSHOT_FOLDER = "./snapshots"
+if not os.path.exists(SNAPSHOT_FOLDER):
+    os.makedirs(SNAPSHOT_FOLDER)
 
 def main():
     # Header
@@ -148,15 +161,28 @@ def main():
             num_anomalies = len(faults)
 
             # Draw bounding boxes and labels
+            annotated_frame = frame_rgb.copy()
             for fault in faults:
                 x, y = int(fault['location'][0]), int(fault['location'][1])
-                cv2.rectangle(frame_rgb, (x-30, y-30), (x+30, y+30), (255, 0, 0), 2)
-                cv2.putText(frame_rgb, f"{fault['type']}", (x, y-40), 
+                cv2.rectangle(annotated_frame, (x-30, y-30), (x+30, y+30), (255, 0, 0), 2)
+                cv2.putText(annotated_frame, f"{fault['type']}", (x, y-40), 
                             cv2.FONT_HERSHEY_SIMPLEX, 0.7, (255, 0, 0), 2)
 
+            # Save snapshot if faults are detected
+            if faults:
+                timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+                fault_types = "_".join([fault['type'].replace(" ", "_") for fault in faults])
+                snapshot_filename = f"snapshot_{timestamp}_frame_{frame_count}_{fault_types}.png"
+                snapshot_path = os.path.join(SNAPSHOT_FOLDER, snapshot_filename)
+                cv2.imwrite(snapshot_path, cv2.cvtColor(annotated_frame, cv2.COLOR_RGB2BGR))
+                st.session_state.snapshots.append({
+                    "path": snapshot_path,
+                    "log": f"{timestamp} - Frame {frame_count} - Anomalies: {num_anomalies} ({fault_types})"
+                })
+
             # Update video feed with timestamp
             timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-            video_placeholder.image(frame_rgb, channels="RGB", caption=f"{timestamp}")
+            video_placeholder.image(annotated_frame, channels="RGB", caption=f"{timestamp}")
 
             # Update logs and metrics
             log_entry = f"{timestamp} - Frame {frame_count} - Anomalies: {num_anomalies}"
@@ -165,20 +191,23 @@ def main():
             st.session_state.anomaly_counts.append(num_anomalies)
             st.session_state.frame_numbers.append(frame_count)
 
-            # Keep only the last 100 frames for trends
+            # Keep only the last 100 frames for trends and last 5 snapshots
             if len(st.session_state.frame_numbers) > 100:
                 st.session_state.frame_numbers.pop(0)
                 st.session_state.anomaly_counts.pop(0)
+            if len(st.session_state.snapshots) > 5:
+                st.session_state.snapshots.pop(0)
 
             # Update live logs
             with logs_placeholder.container():
                 for log in st.session_state.logs[::-1]:
                     st.markdown(f'<div class="log-entry">{log}</div>', unsafe_allow_html=True)
 
-            # Update last 5 captured events
+            # Update last 5 captured events with snapshots
             with events_placeholder.container():
-                for log in st.session_state.logs[-5:][::-1]:
-                    st.markdown(f'<div class="log-entry">{log}</div>', unsafe_allow_html=True)
+                for snapshot in st.session_state.snapshots[::-1]:
+                    st.markdown(f'<div class="log-entry">{snapshot["log"]}</div>', unsafe_allow_html=True)
+                    st.image(snapshot["path"], caption="Fault Snapshot", use_column_width=True)
 
             # Update live metrics
             metrics_placeholder.markdown(