add save src face

api/main.py

@@ -94,8 +94,10 @@ async def swapFace(req:swap_req):
     except:
         return {'msg':"無法辨別圖片網址"+req.imgurl}
     name_hash = str(time.time()).replace('.','')
-    
-    x = threading.Thread(target=gen_avatar, args=(name_hash,req.imgurl))
+    im = Image.open(requests.get(imgurl, stream=True).raw)
+    im= im.convert("RGB")
+    im.save('../../face_swap/tmp_face/'+name_hash+'.jpg')
+    x = threading.Thread(target=gen_avatar, args=(name_hash))
     x.start()
     return {'msg':'人物生成中，請稍候'}
 
@@ -177,17 +179,12 @@ 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):
 
-    im = Image.open(requests.get(imgurl, stream=True).raw)
-    im= im.convert("RGB")
-    im.save('../../face_swap/tmp_face/'+name_hash+'.jpg')
     c = rpyc.connect("localhost", 8868)
     c._config['sync_request_timeout'] = None
     remote_svc = c.root
     my_answer = remote_svc.call_avatar(name_hash) # method call
-
-
     shutil.copy(tmp_avatar_dir+name_hash+'.mp4',avatar_dest+name_hash+'.mp4')
     os.remove(tmp_avatar_dir+name_hash+'.mp4')