Browse Source

fix indexing bug

ming 3 years ago
parent
commit
7e8121b6e7
1 changed files with 12 additions and 6 deletions
  1. 12 6
      api/main.py

+ 12 - 6
api/main.py

@@ -219,14 +219,20 @@ async def swapFace(req:models.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 file.read()
+                await out_file.write(content) 
+            return {"msg": 'www.choozmo.com:8168/'+tmp_img_sub_folder+img_name+'.mp4'}
+        else:
+            contents = await file.read()
+            image = Image.open(io.BytesIO(contents))
+            image= image.convert("RGB")
+            image.save(img_upload_folder+img_name+'.jpg')
+            return {"msg": 'www.choozmo.com:8168/'+tmp_img_sub_folder+img_name+'.jpg'}
     except Exception as e:
-        print(e)
+        logging.error(traceback.format_exc())
         return {'msg':'檔案無法使用'}
-    return {"msg": 'www.choozmo.com:8168/'+tmp_img_sub_folder+img_name+'.jpg'}
 
 @app.post("/make_anchor_video_gSlide")
 async def make_anchor_video_gSlide(req:models.gSlide_req,token: str = Depends(oauth2_scheme)):