Update main.py

main.py

@@ -16,8 +16,11 @@ async def process_video(file: UploadFile = File(...)):
     input_filename = f"/tmp/input_{uuid.uuid4()}.mp4"
     output_filename = f"/tmp/output_{uuid.uuid4()}.mp4"
     
-    with open(input_filename, "wb") as f:
-        f.write(await file.read())
+    try:
+        with open(input_filename, "wb") as f:
+            f.write(await file.read())
+    except Exception as e:
+        return {"error": f"Erro ao salvar o vídeo de entrada: {str(e)}"}
 
     cap = cv2.VideoCapture(input_filename)
     success, frame = cap.read()
@@ -37,12 +40,20 @@ async def process_video(file: UploadFile = File(...)):
             '-c:a', 'copy',
             output_filename
         ]
-        subprocess.run(crop_cmd)
+        try:
+            subprocess.run(crop_cmd, check=True)
+        except subprocess.CalledProcessError as e:
+            os.remove(input_filename)
+            return {"error": f"Erro ao processar o vídeo com FFmpeg: {str(e)}"}
+
         os.remove(input_filename)
 
-        response = FileResponse(output_filename, media_type="video/mp4", filename="cropped_video.mp4")
-        response.headers["X-Delete-File"] = output_filename
-        return response
+        if os.path.exists(output_filename):
+            response = FileResponse(output_filename, media_type="video/mp4", filename="cropped_video.mp4")
+            response.headers["X-Delete-File"] = output_filename
+            return response
+        else:
+            return {"error": "O arquivo de saída não foi gerado."}
     else:
         os.remove(input_filename)
         return {"error": "Não foi possível detectar o produto para realizar o crop."}