Update app.py

Solving issues

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import cv2
 import time
 import os
+from datetime import datetime
 from services.video_service import get_next_video_frame
 from services.thermal_service import detect_thermal_anomalies
 from services.overlay_service import overlay_boxes
@@ -10,6 +11,7 @@ from services.metrics_service import update_metrics
 # Globals
 paused = False
 frame_rate = 1
+frame_count = 0
 
 # Constants
 TEMP_IMAGE_PATH = "temp.jpg"
@@ -17,6 +19,7 @@ TEMP_IMAGE_PATH = "temp.jpg"
 # Core monitor function
 def monitor_feed():
     global paused
+    global frame_count
 
     frame = None
 
@@ -30,18 +33,26 @@ def monitor_feed():
     if not paused:
         detected_boxes = detect_thermal_anomalies(frame)
         frame = overlay_boxes(frame, detected_boxes)
-        cv2.imwrite(TEMP_IMAGE_PATH, frame)
+        cv2.imwrite(TEMP_IMAGE_PATH, frame, [int(cv2.IMWRITE_JPEG_QUALITY), 95])
         metrics = update_metrics(detected_boxes)
     else:
         metrics = update_metrics([])
 
     frame = cv2.resize(frame, (640, 480))  # Resize frame to fixed window size
 
+    # Add frame count and timestamp on frame
+    frame_count += 1
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    cv2.putText(frame, f"Frame: {frame_count}", (10, 25), cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0, 255, 0), 2)
+    cv2.putText(frame, f"{timestamp}", (10, 50), cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0, 255, 0), 2)
+
     return frame[:, :, ::-1], metrics  # BGR to RGB
 
 # Gradio UI
-with gr.Blocks() as app:
-    gr.Markdown("# 🌐 Thermal Anomaly Monitoring Dashboard")
+with gr.Blocks(theme=gr.themes.Soft()) as app:
+    gr.Markdown("# 🌐 Thermal Anomaly Monitoring Dashboard", elem_id="main-title")
+
+    status_text = gr.Markdown("**Status:** 🟢 Running", elem_id="status-banner")
 
     with gr.Row():
         with gr.Column(scale=3):
@@ -57,17 +68,19 @@ with gr.Blocks() as app:
     def toggle_pause():
         global paused
         paused = True
+        return "**Status:** ⏸️ Paused"
 
     def toggle_resume():
         global paused
         paused = False
+        return "**Status:** 🟢 Running"
 
     def set_frame_rate(val):
         global frame_rate
         frame_rate = val
 
-    pause_btn.click(toggle_pause)
-    resume_btn.click(toggle_resume)
+    pause_btn.click(toggle_pause, outputs=status_text)
+    resume_btn.click(toggle_resume, outputs=status_text)
     frame_slider.change(set_frame_rate, inputs=[frame_slider])
 
     def streaming_loop():