Spaces:
Runtime error
Runtime error
File size: 2,995 Bytes
19f420a f37553c 19f420a f37553c 19f420a f37553c 19f420a f37553c 19f420a f37553c 19f420a f37553c 19f420a f37553c 19f420a f37553c 19f420a f37553c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
# drive_paddy/alerting/alert_system.py
import time, os, io, google.generativeai as genai
from gtts import gTTS
class BaseAlerter:
def __init__(self, config):
self.config = config['alerting']
self.cooldown = self.config['alert_cooldown_seconds']
self.last_alert_time = 0
self.alert_on = False
def trigger_alert(self, level="Very Drowsy"): raise NotImplementedError
def reset_alert(self):
if self.alert_on: print("Resetting Alert."); self.alert_on = False
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}.")
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; self.alert_on = True
print("Triggering Static Alert!")
return self.audio_bytes
return None
class GeminiAlertSystem(BaseAlerter):
def __init__(self, config, api_key):
super().__init__(config)
try: genai.configure(api_key=api_key); self.model = genai.GenerativeModel('gemini-pro')
except Exception as e: print(f"Error initializing Gemini: {e}."); self.model = None
def _generate_audio_data(self, level):
if not self.model: return None
if level == "Slightly Drowsy":
prompt = "You are an AI driving assistant. Generate a short, gentle reminder (under 10 words) for a driver showing minor signs of fatigue."
else: # Very Drowsy
prompt = "You are an AI driving assistant. Generate a short, firm, and urgent alert (under 10 words) for a driver who is very drowsy."
try:
response = self.model.generate_content(prompt)
alert_text = response.text.strip().replace('*', '')
print(f"Generated Alert Text ({level}): '{alert_text}'")
mp3_fp = io.BytesIO(); tts = gTTS(text=alert_text, lang='en'); tts.write_to_fp(mp3_fp)
mp3_fp.seek(0); return mp3_fp.getvalue()
except Exception as e: print(f"Error generating TTS audio: {e}"); return None
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.model:
self.last_alert_time = current_time; self.alert_on = True
return self._generate_audio_data(level)
return None
def get_alerter(config, api_key=None):
if config.get('gemini_api', {}).get('enabled', False) and api_key: return GeminiAlertSystem(config, api_key)
return FileAlertSystem(config)
|