Update app.py

app.py

@@ -1,65 +1,79 @@
-import gradio as gr
-import cv2
 import os
 import random
-import time
+import cv2
+import gradio as gr
 from services.video_service import get_random_video_frame
-from services.metrics_service import update_metrics
 from services.overlay_service import overlay_boxes
+from services.thermal_service import detect_thermal_anomalies
+from services.metrics_service import update_metrics, get_metrics_text
 
-# Constants
+# Global Configurations
 TEMP_IMAGE_PATH = "temp_frame.jpg"
-VIDEO_FOLDER = "data/sample_videos"
-
-# State Variables
 paused = False
-frame_rate = 1  # Default 1 FPS
+frame_rate = 1  # Default frames per second
+
+# Gradio state variables
+global_frame_rate = frame_rate
 
-# Helper Functions
 def monitor_feed():
     global paused
+
+    if paused:
+        return TEMP_IMAGE_PATH, get_metrics_text()
+
     frame = get_random_video_frame()
+
     if frame is None:
-        return None, update_metrics()
-    if not paused:
-        frame = overlay_boxes(frame)
-        cv2.imwrite(TEMP_IMAGE_PATH, frame)
-    return TEMP_IMAGE_PATH, update_metrics()
+        return TEMP_IMAGE_PATH, get_metrics_text()
 
-def toggle_pause():
-    global paused
-    paused = not paused
+    detected_boxes = detect_thermal_anomalies(frame)
+    frame = overlay_boxes(frame, detected_boxes)
+
+    update_metrics(detected=bool(detected_boxes), anomaly_name="Anomaly")
 
-def set_frame_rate(value):
-    global frame_rate
-    frame_rate = int(value)
+    cv2.imwrite(TEMP_IMAGE_PATH, frame)
+
+    return TEMP_IMAGE_PATH, get_metrics_text()
 
 def loop_monitor():
     while True:
-        time.sleep(1 / frame_rate if frame_rate > 0 else 1)
         yield monitor_feed()
 
-# Gradio App
+def pause_stream():
+    global paused
+    paused = True
+
+def resume_stream():
+    global paused
+    paused = False
+
+def update_frame_rate(new_rate):
+    global global_frame_rate
+    global_frame_rate = int(new_rate)
+
 def build_interface():
-    with gr.Blocks() as app:
+    with gr.Blocks(title="Solar Surveillance Thermal Detection") as app:
+        gr.Markdown("# ✨ Solar Surveillance Thermal Anomaly Detection")
+
         with gr.Row():
-            with gr.Column():
-                video_output = gr.Image(label="Live Drone Feed")
-            with gr.Column():
-                metrics_output = gr.Label(label="Live Metrics")
+            video_output = gr.Image(label="Video Feed", elem_id="video-feed")
+            metrics_output = gr.Textbox(label="Live Metrics", elem_id="metrics-box")
 
         with gr.Row():
-            pause_button = gr.Button("Pause / Resume")
-            frame_rate_slider = gr.Slider(1, 10, value=1, step=1, label="Frame Rate (Frames per second)")
+            with gr.Column():
+                pause_button = gr.Button("Pause")
+                resume_button = gr.Button("Resume")
+            with gr.Column():
+                frame_rate_slider = gr.Slider(1, 10, step=1, value=global_frame_rate, label="Frame Rate (fps)")
 
-        pause_button.click(toggle_pause)
-        frame_rate_slider.change(set_frame_rate, inputs=[frame_rate_slider])
+        pause_button.click(pause_stream)
+        resume_button.click(resume_stream)
+        frame_rate_slider.change(update_frame_rate, inputs=[frame_rate_slider])
 
-        app.load(fn=loop_monitor, outputs=[video_output, metrics_output])
+        app.load(loop_monitor, outputs=[video_output, metrics_output])
 
     return app
 
-# Launch
 if __name__ == "__main__":
     demo = build_interface()
-    demo.queue().launch()
+    demo.launch(server_name="0.0.0.0", server_port=7860)