Main code(DDDS).py

import cv2
import mediapipe as mp
import serial
import time
import numpy as np

# --- CONFIGURATION AND SETUP ---
# This is the final configuration based on your requests.

ARDUINO_PORT = 'COM4' 
BAUD_RATE = 9600
EAR_THRESHOLD = 0.23
EYE_CLOSED_DURATION_SECONDS = 1.5

# --- SERIAL COMMUNICATION SETUP ---

print(f"Connecting to Arduino on {ARDUINO_PORT}...")
try:
    arduino = serial.Serial(port=ARDUINO_PORT, baudrate=BAUD_RATE, timeout=1)
    # This sleep is important for the connection to establish.
    time.sleep(2)
    # Send an initial 'Awake' signal to start the motor, as requested.
    print("Connection successful. Sending initial 'Awake' signal...")
    arduino.write(b'1')
    print("Motor activated.")
except serial.SerialException as e:
    print(f"Error: Could not connect to Arduino. {e}")
    exit()

# --- MEDIPIPE FACE MESH SETUP ---

mp_face_mesh = mp.solutions.face_mesh
face_mesh = mp_face_mesh.FaceMesh(
    max_num_faces=1, 
    refine_landmarks=True, 
    min_detection_confidence=0.5, 
    min_tracking_confidence=0.5
)

LEFT_EYE_POINTS = [33, 160, 158, 133, 153, 144]
RIGHT_EYE_POINTS = [362, 385, 387, 263, 373, 380]

def calculate_ear(eye_landmarks, frame_shape):
    coords_points = np.array([(int(lm.x * frame_shape[1]), int(lm.y * frame_shape[0])) for lm in eye_landmarks])
    p2_p6 = np.linalg.norm(coords_points[1] - coords_points[5])
    p3_p5 = np.linalg.norm(coords_points[2] - coords_points[4])
    p1_p4 = np.linalg.norm(coords_points[0] - coords_points[3])
    ear = (p2_p6 + p3_p5) / (2.0 * p1_p4)
    return ear

# --- MAIN VIDEO CAPTURE LOOP ---

cap = cv2.VideoCapture(0)

eye_closed_start_time = None
system_status = "Awake" 

while cap.isOpened():
    ret, frame = cap.read()
    if not ret:
        break

    frame = cv2.flip(frame, 1)
    rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
    results = face_mesh.process(rgb_frame)

    display_status = "Awake" 

    if results.multi_face_landmarks:
        face_landmarks = results.multi_face_landmarks[0].landmark
        
        left_eye_lms = [face_landmarks[i] for i in LEFT_EYE_POINTS]
        right_eye_lms = [face_landmarks[i] for i in RIGHT_EYE_POINTS]

        left_ear = calculate_ear(left_eye_lms, frame.shape)
        right_ear = calculate_ear(right_eye_lms, frame.shape)
        avg_ear = (left_ear + right_ear) / 2.0

        if avg_ear < EAR_THRESHOLD:
            if eye_closed_start_time is None:
                eye_closed_start_time = time.time()
            
            elapsed_time = time.time() - eye_closed_start_time
            
            if elapsed_time >= EYE_CLOSED_DURATION_SECONDS:
                display_status = "Asleep"
                if system_status != "Asleep":
                    system_status = "Asleep"
                    print(f"STATE CHANGE: {system_status}")
                    arduino.write(b'0')
            else:
                display_status = "Blinking"
        else:
            eye_closed_start_time = None
            display_status = "Awake"
            if system_status != "Awake":
                system_status = "Awake"
                print(f"STATE CHANGE: {system_status}")
                arduino.write(b'1')
    
    else:
        display_status = "No Face Detected"
        eye_closed_start_time = None 
        if system_status != "Asleep":
            system_status = "Asleep"
            print(f"STATE CHANGE: Asleep (No face detected)")
            arduino.write(b'0')

    if display_status == "Awake":
        color = (0, 255, 0)
    elif display_status == "Asleep":
        color = (0, 0, 255)
    elif display_status == "Blinking":
        color = (0, 255, 255)
    else: # No Face Detected
        color = (0, 165, 255)

    cv2.putText(frame, display_status, (50, 50), 
                cv2.FONT_HERSHEY_SIMPLEX, 1.5, color, 3)

    cv2.imshow("Eye State Detection", frame)

    if cv2.waitKey(5) & 0xFF == ord('q'):
        break

# --- CLEANUP ---
cap.release()
cv2.destroyAllWindows()
if arduino.is_open:
    print("Exiting program. Sending SHUTDOWN signal...")
    # Send the '2' command for a clean shutdown (motor and buzzer off).
    arduino.write(b'2')
    time.sleep(0.1)
    arduino.close()
    print("Arduino connection closed. Everything should be off.")