add save src face

api/main.py

@@ -178,20 +178,24 @@ def gen_video(name_hash,name,text_content, image_urls,avatar,client_id):
 
 
 def gen_avatar(name_hash, imgurl):
+    im = Image.open(requests.get(imgurl, stream=True).raw)
+    im= im.convert("RGB")
+    im.save('../../face_swap/tmp_face/'+name_hash+'.jpg')
+    
     while True:
-
         directory= os.listdir('../../face_swap/tmp_face/') 
-        if len(directory) == 0 and not os.path.isdir('../../face_swap/queueFlag'):
+        if len(directory) != 0 and not os.path.isdir('../../face_swap/queueFlag'):
             os.mkdir('../../face_swap/queueFlag')
             c = rpyc.connect("localhost", 8868)
             c._config['sync_request_timeout'] = None
             remote_svc = c.root
-            my_answer = remote_svc.call_avatar(name_hash,imgurl) # method call
+            q_name = directory[0]
+            my_answer = remote_svc.call_avatar(q_name,imgurl) # method call
         
-            shutil.copy(tmp_avatar_dir+name_hash+'.mp4',avatar_dest+name_hash+'.mp4')
-            os.remove(tmp_avatar_dir+name_hash+'.mp4')
+            shutil.copy(tmp_avatar_dir+q_name+'.mp4',avatar_dest+q_name+'.mp4')
+            os.remove(tmp_avatar_dir+q_name+'.mp4')
             shutil.rmtree('../../face_swap/queueFlag')
-            os.remove('../../face_swap/tmp_face/'+name_hash+'.jpg')
+            os.remove('../../face_swap/tmp_face/'+q_name+'.jpg')
         else:
             break