Update src/alerting/alert_system.py

src/alerting/alert_system.py

@@ -1,90 +1,84 @@
 # drive_paddy/alerting/alert_system.py
-import time, os, io, google.generativeai as genai
+import time, os, google.generativeai as genai
 from gtts import gTTS
 
 class BaseAlerter:
-    """Base class for alert systems, handling state and cooldowns."""
     def __init__(self, config):
-        self.config = config.get('alerting', {})
-        self.cooldown = self.config.get('alert_cooldown_seconds', 15) # Increased cooldown for dynamic alerts
+        self.config   = config.get("alerting", {})
+        self.cooldown = self.config.get("alert_cooldown_seconds", 15)
         self.last_alert_time = 0
         self.alert_on = False
 
     def trigger_alert(self, level="Very Drowsy"):
-        """
-        Checks if an alert can be triggered based on cooldown.
-        If so, marks the alert as active and returns the appropriate text.
-        Returns None if on cooldown.
-        """
         raise NotImplementedError
 
     def reset_alert(self):
-        """
-        Resets the alert state. This is crucial to call after an alert has finished playing
-        to allow the next one to be triggered.
-        """
         if self.alert_on:
-            print("Resetting Alert State.")
-            self.alert_on = Falsese
+            print("Resetting alert state.")
+            self.alert_on = False   # ← fixed typo
 
+# ──────────────────────────────────────────────────────────
+# STATIC FILE ALERTER
+# ──────────────────────────────────────────────────────────
 class FileAlertSystem(BaseAlerter):
     def __init__(self, config):
         super().__init__(config)
         self.audio_bytes = None
-        
         try:
-            if os.path.exists(config['alerting']['alert_sound_path']):
-                with open(config['alerting']['alert_sound_path'], "rb") as f: self.audio_bytes = f.read()
-        except Exception as e: print(f"Warning: Could not load audio file. Error: {e}.")
-    
+            p = config["alerting"]["alert_sound_path"]
+            with open(p, "rb") as f:
+                self.audio_bytes = f.read()
+            print("Static alert sound loaded.")
+        except Exception as e:
+            print(f"Could not load alert sound: {e}")
+
     def trigger_alert(self, level="Very Drowsy"):
-        current_time = time.time()
-        if (current_time - self.last_alert_time) > self.cooldown and not self.alert_on and self.audio_bytes:
-            self.last_alert_time = current_time
+        now = time.time()
+        if (now - self.last_alert_time) > self.cooldown and not self.alert_on and self.audio_bytes:
+            self.last_alert_time = now
             self.alert_on = True
-            # For this simple system, we just return a flag to play the audio
-            # The main app will handle the audio data itself.
-            return "play_static_alert" 
+            return self.audio_bytes           # return raw bytes
         return None
 
+# ──────────────────────────────────────────────────────────
+# GEMINI TTS TEXT ALERTER
+# ──────────────────────────────────────────────────────────
 class GeminiAlertSystem(BaseAlerter):
-    """Generates dynamic alert TEXT using Gemini."""
     def __init__(self, config, api_key):
         super().__init__(config)
-        self.model = None
         try:
             genai.configure(api_key=api_key)
-            self.model = genai.GenerativeModel('gemini-pro')
-            print("Successfully initialized GeminiAlertSystem.")
+            self.model = genai.GenerativeModel("gemini-pro")
+            print("Gemini alert system ready.")
         except Exception as e:
-            print(f"Error initializing Gemini: {e}.")
-
-    def get_alert_text(self, level):
-        """Generates just the text for an alert."""
-        if not self.model: return "Warning, driver is drowsy."
+            print(f"Gemini init failed: {e}")
+            self.model = None
 
-        if level == "Slightly Drowsy":
-            prompt = "Generate a short, gentle, and encouraging reminder (under 10 words) for a driver showing minor signs of fatigue. Example: 'Feeling a bit tired? Maybe take a quick break.'"
-        else: # Very Drowsy
-            prompt = "Generate a short, firm, and urgent alert (under 8 words) for a driver who is very drowsy. Be direct and clear. Example: 'Danger! Pull over now!' or 'Wake up! Stay focused!'"
+    def _gen_text(self, level):
+        if not self.model:
+            return "Driver alert! Please check your status."
+        prompt = (
+            "Generate a short, gentle reminder (under 10 words) for a slightly fatigued driver."
+            if level == "Slightly Drowsy"
+            else "Generate a short, urgent alert (under 8 words) for a very drowsy driver."
+        )
         try:
-            response = self.model.generate_content(prompt)
-            return response.text.strip().replace('*', '')
+            return self.model.generate_content(prompt).text.strip().replace("*", "")
         except Exception as e:
-            print(f"Error generating text with Gemini: {e}")
-            return "Driver alert! Please check your status."
+            print(f"Gemini error: {e}")
+            return "Stay alert! Pull over."
 
     def trigger_alert(self, level="Very Drowsy"):
-        """If not on cooldown, generates alert text and returns it."""
-        current_time = time.time()
-        if (current_time - self.last_alert_time) > self.cooldown and not self.alert_on:
-            self.last_alert_time = current_time
+        now = time.time()
+        if (now - self.last_alert_time) > self.cooldown and not self.alert_on:
+            self.last_alert_time = now
             self.alert_on = True
-            print(f"Alert state is ON for level: {level}. Generating text.")
-            return self.get_alert_text(level)
+            return self._gen_text(level)
         return None
 
+# ──────────────────────────────────────────────────────────
 def get_alerter(config, api_key=None):
-    if config.get('alerting', {}).get('gemini_alerts', {}).get('enabled', False) and api_key:
+    gem_cfg = config.get("alerting", {}).get("gemini_alerts", {})
+    if gem_cfg.get("enabled") and api_key:
         return GeminiAlertSystem(config, api_key)
     return FileAlertSystem(config)