Update app.py

Fixed the Variable Frame issue

app.py

@@ -2,40 +2,57 @@ import gradio as gr
 import cv2
 import time
 import os
-from services.video_service import get_random_video_frame
+from services.video_service import get_next_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.0  # seconds
+frame_rate = 1
 
+# Constants
+TEMP_IMAGE_PATH = "temp.jpg"
+
+# Core monitor function
 def monitor_feed():
     global paused
+
+    frame = None
+
     if paused:
-        frame = get_random_video_frame()  # Still fetch to keep system active
-    else:
-        frame = get_random_video_frame()
+        if os.path.exists(TEMP_IMAGE_PATH):
+            frame = cv2.imread(TEMP_IMAGE_PATH)
+
+    if frame is None:
+        frame = get_next_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)
-        return frame[:, :, ::-1], metrics  # BGR → RGB
+    else:
+        metrics = update_metrics([])
+
+    frame = cv2.resize(frame, (640, 480))  # Resize frame to fixed window size
 
-    return frame[:, :, ::-1], {"Detections": 0}
+    return frame[:, :, ::-1], metrics  # BGR to RGB
 
+# Gradio UI
 with gr.Blocks() as app:
-    gr.Markdown("# 🌡️ Solar Surveillance Thermal Monitoring")
+    gr.Markdown("# 🌐 Thermal Anomaly Monitoring Dashboard")
 
     with gr.Row():
         with gr.Column(scale=3):
-            video_output = gr.Image(label="Live Video Feed", elem_id="video-feed")
+            video_output = gr.Image(label="Live Video Feed", elem_id="video-feed", width=640, height=480)
         with gr.Column(scale=1):
-            metrics_output = gr.Label(label="Detection Metrics")
+            metrics_output = gr.Label(label="Live Metrics", elem_id="metrics")
 
     with gr.Row():
         pause_btn = gr.Button("⏸️ Pause")
         resume_btn = gr.Button("▶️ Resume")
-        frame_slider = gr.Slider(0.5, 5.0, step=0.1, value=1.0, label="Frame Interval (seconds)")
+        frame_slider = gr.Slider(0.2, 5, value=1, label="Frame Interval (seconds)")
 
     def toggle_pause():
         global paused
@@ -55,8 +72,7 @@ with gr.Blocks() as app:
 
     def streaming_loop():
         while True:
-            frame, metrics = monitor_feed()
-            yield frame, metrics
+            yield monitor_feed()
             time.sleep(frame_rate)
 
     app.load(streaming_loop, outputs=[video_output, metrics_output])