Python 居眠り防止機能を作ってみた!ver2

import cv2

import time

#音を出すために使う
import pygame

#mp3ファイルの音源を再生するための関数
def add_sound():
    
    #再生したいmp3ファイル
    soundfile = "beam4.mp3"
    
    pygame.mixer.init()
    
    #mp3ファイルを読み込む
    pygame.mixer.music.load(soundfile)
    
    #再生開始
    pygame.mixer.music.play(1)
    
    #再生開始後，音源の長さだけ待つ（0.25秒）
    time.sleep(0.25)
    
    #停止
    pygame.mixer.music.stop()

def snooz_alery():

    #カメラを起動する
    cap = cv2.VideoCapture(0)
    
    #始まりの時間を指定
    start_time = time.time()
    
    
    while True:
    
        #----------------------------------------------------------------------------------------------------
        
        #カスケードファイルを用意
        cascade_file = "haarcascade_eye_tree_eyeglasses.xml"
        
        #カスケードファイルを読み込む
        cascade = cv2.CascadeClassifier(cascade_file)
        
        #----------------------------------------------------------------------------------------------------
    
        #1フレームずつ取得する
        ret, frame = cap.read()
        
        #左右を反転する
        frame = cv2.flip(frame, 1)
        
        #もし，取得できなかったら，
        if not ret:
            
            #終了
            break
        
        #画像を縮小
        frame = cv2.resize(frame, (500, 300))
        
        #グレイスケールに変換
        gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
        
        #目の認識を実行
        eyes_list = cascade.detectMultiScale(gray, minNeighbors=10)
        
        #----------------------------------------------------------------------------------------------------
        
        #目の部分を四角で囲む
        for (x,y,w,h) in eyes_list:
            
            #赤色を用意
            red = (0,0,255)
            
            #赤色の枠で囲む
            cv2.rectangle(frame, (x,y), (x+w,y+h),red, 1)
            
        #----------------------------------------------------------------------------------------------------
        
        #もし，目が検出できなかった時，
        if len(eyes_list) == 0:
               
            #経過時間を指定する
            end_time = time.time()
            
            #始まりの時間から経過時間を表示する
            print(end_time-start_time)
            
            #もし，3秒経過したら，
            if end_time - start_time >= 3:
               
                #mp3ファイルの音源を再生（今回作った関数を読み込む） 
                add_sound()
            
                #「寝るな!」と表示
                cv2.putText(frame,"Don't Sleep!",(10,70),cv2.FONT_HERSHEY_PLAIN,1.5,(0,0,255),2,cv2.LINE_AA)
              
        #もし，目を検出したら，
        elif len(eyes_list) > 0:
            
            #時間をもとに戻す
            start_time = end_time
            
        #----------------------------------------------------------------------------------------------------
         
        #結果を出力
        cv2.imshow("human_body_detect", frame)
        
        #frame_diffの結果を描写
        #cv2.imshow("the_difference", frame_diff)
        
        #もし，エンターが押されたら終了
        key = cv2.waitKey(1)
        if key == 13:
            break
        
    
    cap.release()
    
    cv2.destroyAllWindows()

if __name__ == "__main__":
    
    snooz_alery()