Update app.py

app.py

@@ -38,10 +38,12 @@ detector = get_detector(CFG)
 # It handles its own state (last alert time) internally.
 class AlertManager:
     def __init__(self, config):
-        self.cooldown_seconds = config.get("alert_cooldown_seconds", 1)
+        self.cooldown_seconds = config.get("alert_cooldown_seconds", 5)
         self.last_alert_time = 0
         self.alert_data = None
         self.sample_rate = None
+        # --- NEW: State variable to track if an alert is active ---
+        self.is_alert_active = False
         self._load_sound(config.get("alert_sound_path"))
 
     def _load_sound(self, wav_path):
@@ -49,6 +51,7 @@ class AlertManager:
             logging.warning("No 'alert_sound_path' found in config.")
             return
         try:
+            # Load as int16 to avoid the Gradio conversion warning
             data, sr = sf.read(wav_path, dtype="int16")
             self.alert_data = data
             self.sample_rate = sr
@@ -58,18 +61,34 @@ class AlertManager:
             self.alert_data = None
 
     def trigger_alert(self, level, lighting):
-        """Checks conditions and returns audio payload if an alert should fire."""
+        """
+        Checks conditions and returns audio payload if a new alert should fire.
+        This is now stateful.
+        """
+        # --- NEW LOGIC: Part 1 ---
+        # If an alert is currently active, we do nothing until the user is 'Awake'.
+        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 # Important: Return None to prevent any sound
+
+        # --- ORIGINAL LOGIC (with a small change) ---
+        # If no alert is active, check for conditions to fire a new one.
         is_drowsy = level != "Awake"
         is_good_light = lighting != "Low"
+        # The time-based cooldown is still useful to prevent flickering alerts.
         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()
-            logging.info("🔊 Alert conditions met! Triggering sound.")
+            # --- NEW LOGIC: Part 2 ---
+            # Set the alert to active so it doesn't fire again immediately.
+            self.is_alert_active = True
+            logging.info("🔊 Drowsiness detected! Firing alert and setting state to active.")
             return (self.sample_rate, self.alert_data.copy())
-        
-        return None
 
+        return None
 # Initialize the alert manager
 alert_manager = AlertManager(CFG["alerting"])