|
@@ -118,10 +118,15 @@ async def swapFace(req:swap_req):
|
|
|
async def create_upload_file(file: UploadFile = File(...)):
|
|
|
img_name = str(time.time()).replace('.','')
|
|
|
try:
|
|
|
- contents = await file.read()
|
|
|
- image = Image.open(io.BytesIO(contents))
|
|
|
- image= image.convert("RGB")
|
|
|
- image.save(img_upload_folder+img_name+'.jpg')
|
|
|
+ if file.content_type=='video/mp4':
|
|
|
+ async with aiofiles.open(img_upload_folder+img_name+'.mp4', 'wb') as out_file:
|
|
|
+ content = await in_file.read()
|
|
|
+ await out_file.write(content)
|
|
|
+ else :
|
|
|
+ contents = await file.read()
|
|
|
+ image = Image.open(io.BytesIO(contents))
|
|
|
+ image= image.convert("RGB")
|
|
|
+ image.save(img_upload_folder+img_name+'.jpg')
|
|
|
except:
|
|
|
return {'msg':'檔案無法使用'}
|
|
|
return {"msg": 'www.choozmo.com:8168/'+tmp_img_sub_folder+img_name+'.jpg'}
|