fix swap face

api/main.py

@@ -80,23 +80,18 @@ async def avatar():
     return FileResponse('static/gen_avatar.html')
 
 @app.post("/swapFace")
-async def swapFace(imgurl):
+async def swapFace(req:swap_req):
     '''
     sf = swap_face(req.imgurl)
     result = sf.run()
     #notify_group(result)hi
     '''
-    im = Image.open(requests.get(imgu, stream=True).raw)
-    im.save(dir_photo+name_hash+"/"+str(img_num)+".jpg")
     try:
-        if get_url_type(imgurl) =='video/mp4':
-            r=requests.get(imgurl)
-        else:
-            im = Image.open(requests.get(imgurl, stream=True).raw)
-            im= im.convert("RGB")
+        im = Image.open(requests.get(req.imgurl, stream=True).raw)
+        im= im.convert("RGB")
     except:
-        return {'msg':"無法辨別圖片網址"+imgurl}
-    x = threading.Thread(target=gen_avatar, args=(name_hash,imgurl))
+        return {'msg':"無法辨別圖片網址"+req.imgurl}
+    x = threading.Thread(target=gen_avatar, args=(name_hash,req.imgurl))
     x.start()
     return '人物生成中，請稍候'
 

api/static/gen_avatar.html

@@ -66,7 +66,7 @@
       <fieldset>
         <h2 class="fs-title">AI ANCHOR</h2>
         <h3  class="fs-subtitle">影像來源</h3>
-        <input id=title type="text" name='t1' class='img_src' value="" placeholder="1" /> <br/>
+        <input id=title type="text" name='t1' class='img_src' value="" placeholder="1" /><input type="file" class="img_uploader img_up"> <br/>
       
         <input id="checker" type="button" name="gen_avatar" class="gen_avatar action-button" value="送出" />