from aip import AipFace
import cv2
import time
import base64
from PIL import Image
from io import BytesIO
import pyttsx3
# """ 你的 APPID AK SK """
APP_ID = '1965####'
API_KEY = 'YXL65ekIloykyjrT4kzc####'
SECRET_KEY = 'lFiapBoZ5eBwOFyxMbiwQDmClg1u####'

client = AipFace(APP_ID, API_KEY, SECRET_KEY)

# def frame2base64(frame):
#   img = Image.fromarray(frame) #將每一幀轉(zhuǎn)為Image
#   output_buffer = BytesIO() #創(chuàng)建一個(gè)BytesIO
#   img.save(output_buffer, format='JPEG') #寫入output_buffer
#   byte_data = output_buffer.getvalue() #在內(nèi)存中讀取
#   image = base64.b64encode(byte_data) #轉(zhuǎn)為BASE64
#   return image #轉(zhuǎn)碼成功 返回base64編碼

def generate():
  camera = cv2.VideoCapture(0)
  engine = pyttsx3.init()
  try:
    while True:
      engine = pyttsx3.init()
      ret, img = camera.read()
      cv2.imwrite("E://Ana/face.png",img)
      cv2.imshow("調(diào)用攝像頭", img)
      imageType = "BASE64"
      groupIdList = "1,2,3,4"


      """ 如果有可選參數(shù) """
      options = {}
      options["max_face_num"] = 4
      options["match_threshold"] = 70
      options["quality_control"] = "NORMAL"
      options["liveness_control"] = "NONE"
      # options["user_id"] = "233451"
      options["max_user_num"] = 4

#       """ 帶參數(shù)調(diào)用人臉?biāo)阉?"""
      with open("E://Ana/face.png", 'rb') as fp:
        imageB = base64.b64encode(fp.read())
      image = str(imageB, 'utf-8')
      """ 調(diào)用人臉?biāo)阉?"""
      result = client.search(image, imageType, groupIdList, options)
      engine.runAndWait()
      print(2)
      if result:
        if not result['result']:
          continue
        name = result['result']['user_list'][0]['user_id']#獲取名字
        score = result['result']['user_list'][0]['score']#獲取相似度
        if name == 'cgh_1':
          if score>80:
            print(score)
            print(name)
            engine.say("華來(lái)了")
        elif name == 'yjc_1':
          if score > 80:
            print(score)
            print(name)
            engine.say("楊來(lái)了")
        elif name == 'cjy_1':
          if score > 80:
            print(score)
            print(name)
            engine.say("言來(lái)了")
        elif name == 'hjy_1':
          if score > 80:
            print(score)
            print(name)
            engine.say("怡來(lái)了")
        else:
          print("匹配失敗")
      else:
        continue
  except Exception as e:
    print(e)
  finally:
    # 釋放資源
    engine.runAndWait()
    camera.release()
    cv2.destroyAllWindows()
 
generate()

APP_ID = '1965####'
API_KEY = 'YXL65ekIloykyjrT4kzc####'
SECRET_KEY = 'lFiapBoZ5eBwOFyxMbiwQDmClg1u####'
client = AipFace(APP_ID, API_KEY, SECRET_KEY)