|
@@ -59,7 +59,6 @@ tmp_img_sub_folder = 'tmp_img/'
|
|
|
img_upload_folder = '/var/www/html/'+tmp_img_sub_folder
|
|
|
video_dest = '/var/www/html/'+video_sub_folder
|
|
|
avatar_dest = '/var/www/html/'+avatar_sub_folder
|
|
|
-tmp_src_face ='../../face_swap/tmp_face/'
|
|
|
|
|
|
class swap_req(BaseModel):
|
|
|
imgurl: str
|
|
@@ -95,10 +94,7 @@ async def swapFace(req:swap_req):
|
|
|
except:
|
|
|
return {'msg':"無法辨別圖片網址"+req.imgurl}
|
|
|
name_hash = str(time.time()).replace('.','')
|
|
|
- im = Image.open(requests.get(req.imgurl, stream=True).raw)
|
|
|
- im= im.convert("RGB")
|
|
|
- im.save(tmp_src_face+name_hash+'.jpg')
|
|
|
-
|
|
|
+
|
|
|
x = threading.Thread(target=gen_avatar, args=(name_hash,req.imgurl))
|
|
|
x.start()
|
|
|
return {'msg':'人物生成中,請稍候'}
|
|
@@ -181,15 +177,14 @@ def gen_video(name_hash,name,text_content, image_urls,avatar,client_id):
|
|
|
os.remove(tmp_video_dir+name_hash+'.mp4')
|
|
|
|
|
|
|
|
|
-def gen_avatar(name_hash,imgurl):
|
|
|
-
|
|
|
+def gen_avatar(name_hash, imgurl):
|
|
|
c = rpyc.connect("localhost", 8868)
|
|
|
c._config['sync_request_timeout'] = None
|
|
|
remote_svc = c.root
|
|
|
- status = remote_svc.call_avatar(name_hash) # method call
|
|
|
- if status:
|
|
|
- shutil.copy(tmp_avatar_dir+name_hash+'.mp4',avatar_dest+name_hash+'.mp4')
|
|
|
- os.remove(tmp_avatar_dir+name_hash+'.mp4')
|
|
|
+ my_answer = remote_svc.call_avatar(name_hash,imgurl) # method call
|
|
|
+
|
|
|
+ shutil.copy(tmp_avatar_dir+name_hash+'.mp4',avatar_dest+name_hash+'.mp4')
|
|
|
+ os.remove(tmp_avatar_dir+name_hash+'.mp4')
|
|
|
|
|
|
|
|
|
|