class GUI:
"""Set up and manage all the variables for the GUI interface"""
def __init__(self):
self.root = Tk()
self.root.title("Noob Robot")
self.face = None
self.canvas = None
self.rc = None
def setup_widgets(self):
"""Set up all the parts of the GUI"""
self._init_face()
self._init_video_tool()
self._init_chat()
def _init_chat(self):
self.chatVar = StringVar()
self.chatVar.set("Hello world!")
chatFrame = Frame(self.root)
chatFrame.grid(row=0, column=1, padx=5, pady=10)
self.chatLabel = Label(chatFrame, textvariable=self.chatVar)
self.chatLabel.config(font=("Courier", 12))
self.chatLabel.grid(row=3, column=1)
def _init_face(self):
"""Set up the face"""
canvasFrame = Frame(self.root)
canvasFrame.grid(row=1, column=1, padx=5, pady=5)
self.canvas = Canvas(canvasFrame, width=400, height=400)
self.canvas.grid(row=1, column=1)
self.face = Face(self.canvas)
def _init_video_tool(self):
videoFrame = Frame(self.root, bd=5, padx=10, pady=10)
videoFrame.grid(row=2, column=1, padx=5, pady=5)
self.runText = StringVar()
self.runText.set("Start Animation")
self.runBtn = Button(videoFrame,
textvariable=self.runText,
command=self.run_animation,
width=40,
height=2)
self.runBtn.grid(row=1, column=1)
def _init_expression_tool(self):
expressionFrame = Frame(self.root, bd=5, padx=10, pady=10)
expressionFrame.grid(row=1, column=2, padx=5, pady=5, sticky=N)
expressionTitle = Label(expressionFrame,
text="Expression Option",
font="Arial 16 bold")
expressionTitle.grid(row=0, column=1, padx=5, pady=5)
self.smileBtn = Button(expressionFrame,
text="Smile",
command=self.face.smile)
self.smileBtn.grid(row=1, column=1)
self.sadBtn = Button(expressionFrame,
text="Sad",
command=self.face.sad)
self.sadBtn.grid(row=2, column=1)
self.grinBtn = Button(expressionFrame,
text="Grin",
command=self.face.grin)
self.grinBtn.grid(row=3, column=1)
self.smileBtn = Button()
def run_animation(self):
"""
Run the animation of the face
:return:
"""
self.runText.set("Press 'Esc' to stop")
self.rc = FaceRecognizer(self.root, self.canvas, self.runText,
self.runBtn, self.chatVar)
self.rc.start()
self.runText.set("Start Animation")
def stop_animation(self):
self.rc.join()
def on_closing(self):
if self.rc is not None:
self.rc.join()
self.root.destroy()
def run_face_animation(self):
"""
This callback for the Start button moves the eyeball and change the facial emotion
:return:
"""
faceCascade = cv2.CascadeClassifier(
"../haarcascades/haarcascade_frontalface_alt0.xml")
eye1Cascade = cv2.CascadeClassifier(
"../haarcascades/haarcascade_eye1.xml")
cap = cv2.VideoCapture(0)
while True:
ret, frame = cap.read()
gray = cv2.cvtColor(frame, cv2.COLOR_RGB2GRAY)
face_rects = faceCascade.detectMultiScale(gray, 1.3, 5)
for (x, y, w, h) in face_rects:
cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 3)
roi_gray = gray[y:y + h, x:x + w]
roi_color = frame[y:y + h, x:x + w]
eye = eye1Cascade.detectMultiScale(roi_gray)
for (ex, ey, ew, eh) in eye:
cv2.rectangle(roi_color, (ex, ey), (ex + ew, ey + eh),
(255, 0, 0), 3)
cv2.imshow("Video Capture", frame)
v = cv2.waitKey(20)
c = chr(v & 0xFF)
if c == 'q':
break
def run(self):
"""Start the whole GUI"""
self.root.protocol("WM_DELETE_WINDOW", self.on_closing)
self.root.mainloop()
