Update main.py

main.py

@@ -1,6 +1,5 @@
 from fastapi import FastAPI, UploadFile, File
 from fastapi.responses import FileResponse
-from starlette.middleware.base import BaseHTTPMiddleware
 import cv2
 import subprocess
 import uuid
@@ -16,16 +15,12 @@ async def process_video(file: UploadFile = File(...)):
     input_filename = f"/tmp/input_{uuid.uuid4()}.mp4"
     output_filename = f"/tmp/output_{uuid.uuid4()}.mp4"
     
-    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)}"}
+    with open(input_filename, "wb") as f:
+        f.write(await file.read())
 
     cap = cv2.VideoCapture(input_filename)
     success, frame = cap.read()
     if not success:
-        os.remove(input_filename)
         return {"error": "Não foi possível ler o vídeo."}
     
     roi_box = detect_product_in_frame(frame)
@@ -40,30 +35,9 @@ async def process_video(file: UploadFile = File(...)):
             '-c:a', 'copy',
             output_filename
         ]
-        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)}"}
+        subprocess.run(crop_cmd)
 
         os.remove(input_filename)
-
-        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."}
+        return FileResponse(output_filename, media_type="video/mp4", filename="cropped_video.mp4")
     else:
-        os.remove(input_filename)
         return {"error": "Não foi possível detectar o produto para realizar o crop."}
-
-class CleanupMiddleware(BaseHTTPMiddleware):
-    async def dispatch(self, request, call_next):
-        response = await call_next(request)
-        file_to_delete = response.headers.get("X-Delete-File")
-        if file_to_delete and os.path.exists(file_to_delete):
-            os.remove(file_to_delete)
-        return response
-
-app.add_middleware(CleanupMiddleware)