Update app.py

app.py

@@ -37,7 +37,7 @@ class AlertManager:
     def __init__(self, config):
         self.cooldown_seconds = config.get("alert_cooldown_seconds", 5)
         self.last_alert_time = 0
-        self.is_alert_active = False
+        # --- FIX: The is_alert_active flag is no longer needed ---
         self._load_sound(config.get("alert_sound_path"))
 
     def _load_sound(self, wav_path):
@@ -56,23 +56,27 @@ class AlertManager:
             self.alert_data = None
 
     def trigger_alert(self, level, lighting):
-        if self.is_alert_active:
-            if level == "Awake":
-                logging.info("✅ Alert state reset. User is Awake. Re-arming system.")
-                self.is_alert_active = False
-            return None
-
+        """
+        Triggers a repeating alert if drowsiness persists, governed by a cooldown.
+        """
         is_drowsy = level != "Awake"
         is_good_light = lighting != "Low"
-        is_ready = (time.monotonic() - self.last_alert_time) > self.cooldown_seconds
 
-        if self.alert_data is not None and is_drowsy and is_good_light and is_ready:
-            self.last_alert_time = time.monotonic()
-            self.is_alert_active = True
-            logging.info("🔊 Drowsiness detected! Firing alert.")
-            return (self.sample_rate, self.alert_data.copy())
+        # --- FIX: New simplified logic for repeating alerts ---
+        # Check for drowsiness conditions first.
+        if is_drowsy and is_good_light and self.alert_data is not None:
+            # Then, check if the cooldown period has passed since the last alert.
+            cooldown_passed = (time.monotonic() - self.last_alert_time) > self.cooldown_seconds
+            if cooldown_passed:
+                # If conditions are met, fire the alert and reset the timer.
+                self.last_alert_time = time.monotonic()
+                logging.info(f"🔊 Drowsiness alert! Repeating every {self.cooldown_seconds}s until 'Awake'.")
+                return (self.sample_rate, self.alert_data.copy())
+
+        # If conditions are not met (e.g., user is Awake or cooldown hasn't passed), do nothing.
         return None
 
+        
 alert_manager = AlertManager(CFG.get("alerting", {}))
 
 # ───────────────────────────── Frame Processing for Tab 1 (Image Stream) - UPDATED
@@ -82,7 +86,7 @@ def process_live_frame(frame):
 
     t0 = time.perf_counter()
     try:
-        # --- FIX: Call with draw_visuals=True ---
+        # Call with draw_visuals=True
         processed, indic = detector.process_frame(frame, draw_visuals=True)
     except Exception as e:
         logging.error(f"Error processing frame: {e}")
@@ -103,7 +107,6 @@ def process_live_frame(frame):
     return processed, status_txt, gr.Audio(value=audio_payload, autoplay=True) if audio_payload else None
 
 
-# ───────────────────────────── Frame Processing for Tab 2 (Analysis-Only) - UPDATED
 def process_for_stats_only(frame):
     """
     Processes a frame but does not return any video/image output.
@@ -114,7 +117,7 @@ def process_for_stats_only(frame):
 
     t0 = time.perf_counter()
     try:
-        # --- FIX: Call with draw_visuals=False. The first returned value will be None. ---
+        # Call with draw_visuals=False. The first returned value will be None.
         _, indic = detector.process_frame(frame, draw_visuals=False)
     except Exception as e:
         logging.error(f"Error processing frame: {e}")