Spaces:
Sleeping
Sleeping
File size: 2,618 Bytes
f53e27e aa0b211 f53e27e aa0b211 f53e27e aa0b211 f53e27e aa0b211 76b08f6 799ee9e f53e27e 799ee9e f53e27e aa0b211 f53e27e aa0b211 799ee9e aa0b211 799ee9e 30ed123 aa0b211 799ee9e f53e27e aa0b211 f53e27e aa0b211 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 |
from fastapi import FastAPI, UploadFile, File
from fastapi.responses import FileResponse
from starlette.middleware.base import BaseHTTPMiddleware
import cv2
import subprocess
import uuid
import os
import shutil
import tempfile
app = FastAPI()
os.umask(0o077)
def detect_product_in_frame(frame):
return (100, 50, 400, 400)
@app.post("/process_video/")
async def process_video(file: UploadFile = File(...)):
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")
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)
if roi_box:
x, y, w, h = roi_box
crop_cmd = [
'ffmpeg',
'-i', input_filename,
'-vf', f"crop={w}:{h}:{x}:{y}",
'-c:a', 'copy',
output_filename
]
try:
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": f"Erro ao processar o vídeo com FFmpeg: {e.stderr}"}
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-Dir"] = temp_dir
return response
else:
shutil.rmtree(temp_dir)
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."}
class CleanupMiddleware(BaseHTTPMiddleware):
async def dispatch(self, request, call_next):
response = await call_next(request)
temp_dir_to_delete = response.headers.get("X-Delete-Dir")
if temp_dir_to_delete and os.path.exists(temp_dir_to_delete):
shutil.rmtree(temp_dir_to_delete)
return response
app.add_middleware(CleanupMiddleware)
|