AI_anchor_openshot/util/swap_face.py

import os 
import time
import requests
from PIL import Image
import threading
class swap_face():
    def __init__(self, imgurl):
        self.imgurl = imgurl
    def run(self):
        name_hash = str(time.time()).replace('.','')
        src_img = 'FaceSwap/src_img/'+name_hash+'.jpg'
        sv_path = '../html/swap_save'+name_hash+'.avi'
        
        try:
            im = Image.open(requests.get(self.imgurl, stream=True).raw)
            im= im.convert("RGB")
            im.save(src_img)
        except:
            return '圖片錯誤'
        
        x = threading.Thread(target=runthreadswap, args=(src_img,sv_path))
        x.start()
        
        return '生成中請稍候'
    def runthreadswap(self.src_img,sv_path):
        os.system('python3 FaceSwap/main_video.py --src_img '+src_img+' --video_path FaceSwap/nina_noGesture_adj.mp4 --correct_color --save_path '+sv_path)    
        while True:
            print('waiting...')
            if os.path.exists(sv_path):
                break
            time.sleep(5)
            print('waiting...')
        notify_group('成功，影片在:www.choozmo.com:8168/swap_save/'+name_hash+'.avi')

    def notify_group(self,msg):
        headers = {
                "Authorization": "Bearer " + "WekCRfnAirSiSxALiD6gcm0B56EejsoK89zFbIaiZQD",
                "Content-Type": "application/x-www-form-urlencoded"
        }
        params = {"message": msg}   
        r = requests.post("https://notify-api.line.me/api/notify",headers=headers, params=params)
        #print(r)