Browse Source

add mp4 upload

ming 3 years ago
parent
commit
438dd454fc
3 changed files with 9 additions and 4 deletions
  1. BIN
      .DS_Store
  2. BIN
      api/.DS_Store
  3. 9 4
      api/main.py

BIN
.DS_Store


BIN
api/.DS_Store


+ 9 - 4
api/main.py

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