import cv2 as cv
import numpy as np
import time
import pygame
import streamlit as st

classes = ['Closed', 'Open']  # Class labels for eyes only

# Load face and eye cascade classifiers
face_cascade = cv.CascadeClassifier("haarcascade files/haarcascade_frontalface_default.xml")
eye_cascade = cv.CascadeClassifier("haarcascade files/haarcascade_eye.xml")

# Initialize alarm
pygame.mixer.init()
pygame.mixer.music.load("alarm.wav")

# Track eye closure duration
closed_start_time = None
alarm_triggered = False

# Prepare the frame for eye detection
def prepare_frame(frame):
    global closed_start_time, alarm_triggered
    gray_frame = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)  # Convert to grayscale for better detection
    faces = face_cascade.detectMultiScale(gray_frame, scaleFactor=1.1, minNeighbors=5)
    eye_status = 'Closed'
    
    for (x, y, w, h) in faces:
        face_roi = frame[y:y+h, x:x+w]
        roi_gray = gray_frame[y:y+h, x:x+w]
        eyes = eye_cascade.detectMultiScale(roi_gray, scaleFactor=1.1, minNeighbors=5)
        
        # If two or more eyes detected, classify as 'Open'
        if len(eyes) >= 2:
            eye_status = 'Open'
            if alarm_triggered:
                pygame.mixer.music.stop()
                alarm_triggered = False
            closed_start_time = None
        else:
            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()
                    alarm_triggered = True
        
        # Draw a rectangle around the entire face
        cv.rectangle(frame, (x, y), (x+w, y+h), (255, 0, 0), 2)
        break  # Stop after detecting the first face to avoid multiple rectangles

    cv.putText(frame, eye_status, (20, 50), cv.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
    return frame, eye_status

# Streamlit interface
st.title("Real-time Eye Detection and Drowsiness Alert")
st.markdown("**Close your eyes for 3 seconds to trigger the alarm.**")

FRAME_WINDOW = st.image([])
run = st.button("Start Detection")
stop = st.button("Stop Detection")

cap = cv.VideoCapture(0)

while run:
    ret, frame = cap.read()
    if not ret:
        st.warning("Failed to access webcam.")
        break

    frame = cv.flip(frame, 1)  # Flip frame horizontally
    frame, eye_status = prepare_frame(frame)
    frame = cv.cvtColor(frame, cv.COLOR_BGR2RGB)
    FRAME_WINDOW.image(frame)

    if stop:
        cap.release()
        cv.destroyAllWindows()
        break

cap.release()
cv.destroyAllWindows()