Update app.py

app.py

@@ -1,16 +1,14 @@
 import streamlit as st
 import numpy as np
 import cv2 as cv
+from tensorflow.keras.models import load_model
 from tensorflow.keras.preprocessing.image import img_to_array
 import pygame
+import os 
 import time
 
-# Initialize alarm
-pygame.mixer.init()
-pygame.mixer.music.load("alarm.wav")
-
 # Load the pre-trained CNN model for drowsiness detection
-#model = load_model('drowiness_new6.h5')
+#model = load_model('drowsiness.h5')
 
 # Define labels
 labels = ['Closed', 'Open']
@@ -19,9 +17,18 @@ labels = ['Closed', 'Open']
 face_cascade = cv.CascadeClassifier(cv.data.haarcascades + "haarcascade_frontalface_default.xml")
 eye_cascade = cv.CascadeClassifier(cv.data.haarcascades + "haarcascade_eye.xml")
 
-st.title("Real-time Eye Detection and Drowsiness Alert")
-st.markdown("**Close your eyes for 3 seconds to trigger the alarm.**")
+# Function to initialize audio
+
+def initialize_audio():
+    try:
+        if not os.environ.get("DISPLAY"):
+            raise RuntimeError("No display found. Skipping audio initialization.")
+        pygame.mixer.init()
+        pygame.mixer.music.load("alarm.wav")
+    except Exception as e:
+        print(f"Audio initialization failed: {e}")
 
+initialize_audio()
 
 # Initialize webcam
 cam = cv.VideoCapture(0)
@@ -29,10 +36,6 @@ FRAME_WINDOW = st.image([])
 run = st.button("Start")
 stop = st.button("Stop")
 
-# Variables to track eye closure
-drowsiness_start_time = None
-alarm_triggered = False
-
 # Function to preprocess the frame for model input
 def preprocess_frame(frame):
     frame = cv.resize(frame, (24, 24))  # Resize to match model input shape
@@ -42,6 +45,8 @@ def preprocess_frame(frame):
     return frame
 
 running = False
+closed_start_time = None
+alarm_triggered = False
 
 if run:
     running = True
@@ -74,18 +79,21 @@ while running:
                 center = (x + ex + ew // 2, y + ey + eh // 2)
                 axes = (ew // 2, eh // 2)
                 cv.ellipse(frame, center, axes, 0, 0, 360, (0, 255, 0), 2)
-            drowsiness_start_time = None
+            closed_start_time = None
             if alarm_triggered:
                 pygame.mixer.music.stop()
                 alarm_triggered = False
         else:
-            if drowsiness_start_time is None:
-                drowsiness_start_time = time.time()
-            elif time.time() - drowsiness_start_time >= 2:
+            if closed_start_time is None:
+                closed_start_time = time.time()
+            elif time.time() - closed_start_time >= 3:
                 if not alarm_triggered:
-                    pygame.mixer.music.play()
+                    try:
+                        pygame.mixer.music.play()
+                    except Exception as e:
+                        print("Alarm triggered, but no audio device found.")
                     alarm_triggered = True
-        
+
     # Display eye status text
     cv.putText(frame, f'Eyes: {eye_status}', (30, 30), cv.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
     
@@ -97,7 +105,6 @@ while running:
         running = False
         cam.release()
         cv.destroyAllWindows()
-        pygame.mixer.quit()
         break
 
 cam.release()