import gradio as gr
import requests
import base64
import cv2
import json
import numpy as np

# token获取
def get_token(client_id, client_secret):
    # client_id为官网获取的API Key，client_secret为官网获取的Secret Key.下行client_id=后的马赛克为你的API Key，client_secret=后的马赛克为你的Secret Key
    url = f"https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id={client_id}&client_secret={client_secret}"  # 注意此处的&符号
    response = requests.get(url)
    resultJson = response.json()
    return resultJson['access_token']

def face_merge(image_template,image_target,steps):
    success,encode_image_temp = cv2.imencode(".jpg",image_template) #注意这句编码，否则图像质量不合格
    image_temp = base64.b64encode(encode_image_temp)
    success,encode_image_targ = cv2.imencode(".jpg",image_target) #注意这句编码，否则图像质量不合格
    image_targ = base64.b64encode(encode_image_targ)
    request_url = 'https://aip.baidubce.com/rest/2.0/face/v1/merge'
    access_token = get_token('gZBF8TaiLfewLu4XdWIVe5H1', 'O7zkY3cc1OYu2GHwqvsyxI4YaCgdRn7A')
    request_url = request_url + "?access_token=" + access_token
    headers ={'content-type':'application/json'}
    
    for i in range(steps,-1,-1):
        params = {
            "image_template": {               # 将其中一张图片设置为模板（相当于底层图片）
                "image": str(image_temp,'utf-8'),
                "image_type": "BASE64",
                "quality_control": "NORMAL"
            },
            "image_target": {                 # 将其中一张图片设置为目标（相当于人脸信息“叠加”到模板上）
                "image": str(image_targ,'utf-8'),
                "image_type": "BASE64",
                "quality_control": "NORMAL"
            },
            "merge_degree": "HIGH",            # 融合程度
            "position": 2,                     #左上角水印
            "language": 1,                     #英语水印
            "version": 4.0,
            "alpha": i/steps,
        }  
        params=json.dumps(params)
        result = requests.post(request_url, data=params, headers=headers).json()  # 经过调用百度云接口服务器返回的内容（融合结果）
        if result['error_code'] == 0:
            res = result["result"]["merge_image"]
            res = base64.b64decode(res)
            nparr = np.frombuffer(res,np.uint8)
            img_np = cv2.imdecode(nparr,cv2.IMREAD_COLOR)
            yield img_np
        else:
            print(str(result['error_code']) + result['error_msg'])
            return result['error_code']


with gr.Blocks() as demo:
    gr.Markdown("Face merge.")

    with gr.Row():
        image_template = gr.Image(label="模板图片")
        image_target = gr.Image(label="目标图片")
    with gr.Row():
        steps = gr.Slider(1, 10, value=1, step=1, label="变化步数")
        image_output = gr.Image(label="融合结果")
    image_button = gr.Button("Merge")

    image_button.click(face_merge, inputs=[image_template,image_target,steps], outputs=image_output)

gr.close_all()
demo.queue()
demo.launch()