FearBooth/main.py

import math
import time
import numpy as np
import cv2 #pip install opencv-python ||| pip3 install opencv-contrib-python==4.4.0.46
from gpiozero import Button
from escpos.printer import Serial

from PIL import Image
from PIL import ImageDraw
from PIL import ImageFont

from deepface import DeepFace #pip install deepface
button = Button(2)

cam = cv2.VideoCapture(0)
cv2.namedWindow("webcam", cv2.WND_PROP_FULLSCREEN)
cv2.setWindowProperty("webcam",cv2.WND_PROP_FULLSCREEN,cv2.WINDOW_FULLSCREEN)
printer = Serial(devfile='/dev/serial0',baudrate=19200,bytesize=8,parity='N',stopbits=1.00,dsrdtr=True)
# printer.set(density=10)
cameraMode = False
TIMER = 5

startScreen = cv2.imread("noFace.png")

if cam.isOpened():
    while True:
        ret, img = cam.read()
        # img = cv2.rotate(img, cv2.ROTATE_90_CLOCKWISE)
        if cameraMode and ret:
            prev = time.time() 
            while TIMER >= 0:
                ret, img = cam.read()
                # img = cv2.rotate(img, cv2.ROTATE_90_CLOCKWISE)
                # cv2.putText(img, str(TIMER), (200, 250), cv2.FONT_HERSHEY_SIMPLEX, 7, (0, 255, 255), 4, cv2.LINE_AA)
                img = cv2.flip(img, 0)
                img = cv2.rotate(img, cv2.ROTATE_90_COUNTERCLOCKWISE) 
                img = cv2.cvtColor(img,cv2.COLOR_BGR2RGB) 
                img = Image.fromarray(img)
                draw = ImageDraw.Draw(img)
                
                fontBIG = ImageFont.truetype("HalloweenFont.ttf", 350)
                fontSmall = ImageFont.truetype("HalloweenFont.ttf", 60)
                if TIMER>0:
                    draw.text((130, 150), str(TIMER), font=fontBIG,fill=(255,0,0,255))
                else:
                    
                    draw.text((30, 480), "ANALYZING...", font=fontSmall,fill=(255,0,0,255))
                img = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
                img = cv2.rotate(img, cv2.ROTATE_90_CLOCKWISE)

                cv2.imshow('webcam',img)
                # print(str(TIMER))
                cur = time.time() #current time
                if cur-prev >= 1: 
                    prev = cur
                    TIMER = TIMER-1
                key = cv2.waitKey(5) & 0xFF
                if ord('q') == key:
                    break
            else:
                ret, img = cam.read()
                img = cv2.flip(img, 0)
                img = cv2.rotate(img, cv2.ROTATE_90_COUNTERCLOCKWISE) 
                # img = cv2.rotate(img, cv2.ROTATE_90_CLOCKWISE)
                try:
                    predictions = DeepFace.analyze(img,actions=['emotion'])
                    fearValue = predictions[0]["emotion"]["fear"]
                    surpriseValue = predictions[0]["emotion"]["surprise"]
                    fearPoint = max(fearValue,surpriseValue)
                    print("FEAR:" + str(round(fearPoint,2)))
                    print("SURPRISE:" + str(round(surpriseValue,2)))
                    imgGray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
                    imgGray = cv2.equalizeHist(imgGray) 
                    cv2.imwrite('scared.jpg', imgGray) 
                    cv2.imwrite("/home/gogo/Desktop/Photos/Fear_{}.jpg".format(cur),img)
                    img = cv2.cvtColor(img,cv2.COLOR_BGR2RGB) 
                    img = Image.fromarray(img)
                    draw = ImageDraw.Draw(img)
                    font_size = 75
                    font = ImageFont.truetype("HalloweenFont.ttf", font_size)
                    text = "FEAR LEVEL"
                    draw.text((30, 452), str(text), font=font,fill=(255,0,0,255))
                    img = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)

                    cv2.rectangle(img,(30,550),(30+math.floor(int(fearPoint)*420/100),600),(255,255,255), -1)
                    cv2.rectangle(img,(30,550),(450,600),(0,0,255), 8)
                    img = cv2.rotate(img, cv2.ROTATE_90_CLOCKWISE) 

                    cv2.imshow('webcam',img)
                    cv2.waitKey(2000)
                    if fearPoint>0:
                        #rotoImg = cv2.rotate(img, cv2.ROTATE_90_CLOCKWISE)
                        # cv2.imshow('webcam',img)
                        basewidth = 384
                        imgCrop = Image.open('scared.jpg')
                        wpercent = (basewidth/float(imgCrop.size[0]))
                        hsize = int((float(imgCrop.size[1])*float(wpercent)))
                        imgCrop = imgCrop.resize((basewidth,hsize), Image.Resampling.LANCZOS)
                        imgCrop = imgCrop.save("cropScared.jpg")
                        # cv2.waitKey(2000)
                        printer.set(align='center',font='a',width=2,height=2)
                    
                        printer.image("cropScared.jpg",high_density_vertical=True,high_density_horizontal=False,impl="bitImageRaster")
                        printer.text("Fear Level\n" + "%" + str(round(fearPoint,1))+"\n")  
                        # printer.text("Surprise Level: \n" + str(round(surpriseValue,2))+"/100\n")  
                        # printer.text("Overall: \n" + str(round(fearPoint,2))+"/100\n")  
                        if fearPoint>80:
                            printer.text("(Scream Queen)\n")
                        if fearPoint<5:
                            printer.text("(The Killer)\n")
                        printer.text("\n\n\n\n")
                        #printer.set(align='center',font='b',width=1,height=1)
                        #printer.text("Spooky Night 2023")
                        #printer.text("2023\n")
                        # cv2.waitKey(5000)
                except:
                    cameraMode = False
                    TIMER = 5
                
                #print(30+math.floor(int(fearPoint)*580/100))
                # ft.putText(img=img,text='TEST',org=(15, 70),fontHeight=60,color=(255,  255, 255),thickness=-1,line_type=cv2.LINE_AA,bottomLeftOrigin=True)
                #cv2.rectangle(img,(30,400),(610,450),(255,255,255), 5)
                #cv2.rectangle(img,(30,400),(30+math.floor(int(fearPoint)*580/100),450),(255,255,255), -1)

                # cv2.waitKey(5000)



                cameraMode = False
                TIMER = 5
        else:
            cv2.imshow('webcam',startScreen) 
        
        key = cv2.waitKey(5) & 0xFF
        if ord('t') == key or button.is_pressed:
            cameraMode = True
        if ord('q') == key:
            break
           
cam.release()