Browse Source

remove db que

ming 3 years ago
parent
commit
86fd40a904
1 changed files with 6 additions and 11 deletions
  1. 6 11
      api/main.py

+ 6 - 11
api/main.py

@@ -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')