Update app.py

app.py

@@ -1,56 +1,41 @@
 import gradio as gr
 import cv2
-import os
 import time
+import os
 from services.video_service import get_random_video_frame
 from services.thermal_service import detect_thermal_anomalies
 from services.overlay_service import overlay_boxes
 from services.metrics_service import update_metrics
 
-# Globals
 paused = False
-frame_rate = 1
-
-# Constants
-TEMP_IMAGE_PATH = "temp.jpg"
+frame_rate = 1.0  # seconds
 
-# Core monitor function
 def monitor_feed():
     global paused
-
-    frame = None
-
     if paused:
-        if os.path.exists(TEMP_IMAGE_PATH):
-            frame = cv2.imread(TEMP_IMAGE_PATH)
-
-    if frame is None:
+        frame = get_random_video_frame()  # Still fetch to keep system active
+    else:
         frame = get_random_video_frame()
-
-    if not paused:
         detected_boxes = detect_thermal_anomalies(frame)
         frame = overlay_boxes(frame, detected_boxes)
-        cv2.imwrite(TEMP_IMAGE_PATH, frame)
         metrics = update_metrics(detected_boxes)
-    else:
-        metrics = update_metrics([])
+        return frame[:, :, ::-1], metrics  # BGR → RGB
 
-    return frame[:, :, ::-1], metrics  # BGR to RGB
+    return frame[:, :, ::-1], {"Detections": 0}
 
-# Gradio UI
 with gr.Blocks() as app:
-    gr.Markdown("# 🌐 Thermal Anomaly Monitoring Dashboard")
+    gr.Markdown("# 🌡️ Solar Surveillance Thermal Monitoring")
 
     with gr.Row():
         with gr.Column(scale=3):
             video_output = gr.Image(label="Live Video Feed", elem_id="video-feed")
         with gr.Column(scale=1):
-            metrics_output = gr.Label(label="Live Metrics", elem_id="metrics")
+            metrics_output = gr.Label(label="Detection Metrics")
 
     with gr.Row():
         pause_btn = gr.Button("⏸️ Pause")
         resume_btn = gr.Button("▶️ Resume")
-        frame_slider = gr.Slider(0.2, 5, value=1, label="Frame Interval (seconds)")
+        frame_slider = gr.Slider(0.5, 5.0, step=0.1, value=1.0, label="Frame Interval (seconds)")
 
     def toggle_pause():
         global paused
@@ -71,11 +56,10 @@ with gr.Blocks() as app:
     def streaming_loop():
         while True:
             frame, metrics = monitor_feed()
-            video_output.update(value=frame)
-            metrics_output.update(value=metrics)
+            yield frame, metrics
             time.sleep(frame_rate)
 
-    app.load(streaming_loop)
+    app.load(streaming_loop, outputs=[video_output, metrics_output])
 
 if __name__ == "__main__":
     app.launch()