Update app.py

app.py

@@ -5,6 +5,7 @@ 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
+import time
 
 # Globals
 paused = False
@@ -17,23 +18,25 @@ TEMP_IMAGE_PATH = "temp.jpg"
 def monitor_feed():
     global paused
 
-    if paused:
-        frame = cv2.imread(TEMP_IMAGE_PATH)
-        if frame is None:
+    while True:
+        if paused:
+            frame = cv2.imread(TEMP_IMAGE_PATH)
+            if frame is None:
+                frame = get_random_video_frame()
+        else:
             frame = get_random_video_frame()
-    else:
-        frame = get_random_video_frame()
-        detected_boxes = detect_thermal_anomalies(frame)
-        frame = overlay_boxes(frame, detected_boxes)
-        cv2.imwrite(TEMP_IMAGE_PATH, frame)
+            detected_boxes = detect_thermal_anomalies(frame)
+            frame = overlay_boxes(frame, detected_boxes)
+            cv2.imwrite(TEMP_IMAGE_PATH, frame)
 
-    metrics = update_metrics()
-    return frame[:, :, ::-1], metrics  # Convert BGR to RGB for Gradio
+        metrics = update_metrics()
+        yield frame[:, :, ::-1], metrics  # Convert BGR to RGB for Gradio
+        time.sleep(frame_rate)
 
 # Gradio UI
 with gr.Blocks() as app:
-    gr.Markdown("# \ud83c\udf10 Thermal Anomaly Monitoring Dashboard")
-    
+    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")
@@ -41,34 +44,27 @@ with gr.Blocks() as app:
             metrics_output = gr.Label(label="Live Metrics", elem_id="metrics")
 
     with gr.Row():
-        pause_btn = gr.Button("\u23F8\ufe0f Pause")
-        resume_btn = gr.Button("\u25b6\ufe0f Resume")
+        pause_btn = gr.Button("⏸️ Pause")
+        resume_btn = gr.Button("▶️ Resume")
         frame_slider = gr.Slider(0.2, 5, value=1, label="Frame Interval (seconds)")
 
     def toggle_pause():
         global paused
         paused = True
-        return gr.update()
 
     def toggle_resume():
         global paused
         paused = False
-        return gr.update()
 
     def set_frame_rate(val):
         global frame_rate
         frame_rate = val
-        return gr.update()
 
     pause_btn.click(toggle_pause)
     resume_btn.click(toggle_resume)
     frame_slider.change(set_frame_rate, inputs=[frame_slider])
 
-    app.stream(
-        monitor_feed,
-        outputs=[video_output, metrics_output],
-        every=lambda: frame_rate
-    )
+    app.load(monitor_feed, outputs=[video_output, metrics_output])
 
 if __name__ == "__main__":
     app.launch()