Update app.py

app.py

@@ -4,7 +4,7 @@ import os
 import numpy as np
 from datetime import datetime
 import matplotlib.pyplot as plt
-import time  # Added import for time module
+import time
 from services.detection_service import detect_faults_solar, detect_faults_windmill
 from services.anomaly_service import track_anomalies, predict_anomaly
 from models.solar_model import load_solar_model
@@ -68,7 +68,7 @@ st.markdown(
 # Initialize session state
 if 'paused' not in st.session_state:
     st.session_state.paused = False
-    st.session_state.frame_rate = 1.0
+    st.session_state.frame_rate = 0.2  # Adjusted default to reduce flicker
     st.session_state.frame_count = 0
     st.session_state.logs = []
     st.session_state.anomaly_counts = []
@@ -264,7 +264,7 @@ def main():
         with col6:
             resume_btn = st.button("▶️ Resume")
         with col7:
-            frame_rate = st.slider("Frame Interval (seconds)", 0.0005, 0.5, st.session_state.frame_rate)
+            frame_rate = st.slider("Frame Interval (seconds)", 0.1, 1.0, st.session_state.frame_rate)  # Adjusted range
 
     # Handle button clicks and slider
     if pause_btn:
@@ -279,10 +279,46 @@ def main():
     video_path = os.path.join(VIDEO_FOLDER, video_file)
     while True:
         if st.session_state.paused:
-            frame, metrics, logs, chart, snapshots = monitor_feed(video_path, detection_type, model)
-            if frame is None:
-                st.success("Video processing completed.")
-                break
+            # Skip processing if paused and last frame exists
+            if st.session_state.last_frame is not None:
+                frame, metrics, logs, chart, snapshots = monitor_feed(video_path, detection_type, model)
+                if frame is None:
+                    st.success("Video processing completed.")
+                    break
+                # Update UI only if necessary
+                video_placeholder.image(frame, channels="RGB", width=640)
+                metrics_formatted = metrics.replace("\n", "<br>")
+                metrics_placeholder.markdown(
+                    f'<div class="metrics-text">{metrics_formatted}</div>',
+                    unsafe_allow_html=True
+                )
+                logs_formatted = logs.replace("\n", "</div><div class='log-entry'>")
+                logs_placeholder.markdown(
+                    f'<div class="log-entry">{logs_formatted}</div>',
+                    unsafe_allow_html=True
+                )
+                trends_placeholder.image(chart)
+                with gallery_placeholder.container():
+                    cols = st.columns(5)
+                    for i, col in enumerate(cols):
+                        with col:
+                            if i < len(snapshots):
+                                st.image(snapshots[i], width=100)
+                                st.markdown(f'<div class="log-entry">{st.session_state.snapshots[i]["log"]}</div>', unsafe_allow_html=True)
+                            else:
+                                st.empty()
+                prediction = predict_anomaly(st.session_state.anomaly_counts)
+                if prediction:
+                    prediction_placeholder.warning("**Prediction:** Potential issue detected - anomaly spike detected!")
+                else:
+                    prediction_placeholder.empty()
+                time.sleep(st.session_state.frame_rate)
+                continue  # Avoid rerun when paused
+            else:
+                frame, metrics, logs, chart, snapshots = monitor_feed(video_path, detection_type, model)
+                if frame is None:
+                    st.success("Video processing completed.")
+                    break
         else:
             frame, metrics, logs, chart, snapshots = monitor_feed(video_path, detection_type, model)
             if frame is None:
@@ -291,7 +327,6 @@ def main():
 
         # Update UI
         video_placeholder.image(frame, channels="RGB", width=640)
-        # Fix for f-string backslash issue
         metrics_formatted = metrics.replace("\n", "<br>")
         metrics_placeholder.markdown(
             f'<div class="metrics-text">{metrics_formatted}</div>',