Update main.py

main.py

@@ -10,7 +10,6 @@ import tempfile
 
 app = FastAPI()
 
-# Define permissões restritas para os arquivos criados
 os.umask(0o077)
 
 def detect_product_in_frame(frame):
@@ -18,23 +17,25 @@ def detect_product_in_frame(frame):
 
 @app.post("/process_video/")
 async def process_video(file: UploadFile = File(...)):
-    # Cria um diretório temporário exclusivo para a requisição
     temp_dir = tempfile.mkdtemp()
     input_filename = os.path.join(temp_dir, f"input_{uuid.uuid4()}.mp4")
     output_filename = os.path.join(temp_dir, f"output_{uuid.uuid4()}.mp4")
     
-    # Salva o vídeo de entrada no diretório temporário
-    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:
+        shutil.rmtree(temp_dir)
+        return {"error": f"Erro ao salvar o vídeo de entrada: {str(e)}"}
 
     cap = cv2.VideoCapture(input_filename)
     success, frame = cap.read()
+    cap.release()
     if not success:
         shutil.rmtree(temp_dir)
         return {"error": "Não foi possível ler o vídeo."}
     
     roi_box = detect_product_in_frame(frame)
-    cap.release()
 
     if roi_box:
         x, y, w, h = roi_box
@@ -46,24 +47,26 @@ async def process_video(file: UploadFile = File(...)):
             output_filename
         ]
         try:
-            subprocess.run(crop_cmd, check=True)
-        except subprocess.CalledProcessError:
+            result = subprocess.run(crop_cmd, check=True, capture_output=True, text=True)
+            print("FFmpeg output:", result.stdout)
+            print("FFmpeg error (if any):", result.stderr)
+        except subprocess.CalledProcessError as e:
             shutil.rmtree(temp_dir)
-            return {"error": "Erro ao processar o vídeo com FFmpeg."}
+            return {"error": f"Erro ao processar o vídeo com FFmpeg: {e.stderr}"}
+
+        os.remove(input_filename)
 
-        # Verifica se o arquivo de saída foi criado e prepara a resposta
         if os.path.exists(output_filename):
             response = FileResponse(output_filename, media_type="video/mp4", filename="cropped_video.mp4")
             response.headers["X-Delete-Dir"] = temp_dir
             return response
         else:
             shutil.rmtree(temp_dir)
-            return {"error": "O arquivo de saída não foi gerado."}
+            return {"error": "O arquivo de saída não foi gerado pelo FFmpeg."}
     else:
         shutil.rmtree(temp_dir)
         return {"error": "Não foi possível detectar o produto para realizar o crop."}
 
-# Middleware para limpar o diretório temporário após a resposta
 class CleanupMiddleware(BaseHTTPMiddleware):
     async def dispatch(self, request, call_next):
         response = await call_next(request)