Create app.py

app.py

@@ -0,0 +1,112 @@
+from fastapi import FastAPI, UploadFile, File, Form, HTTPException
+from fastapi.responses import FileResponse, JSONResponse
+import os
+import asyncio
+import uuid
+import shutil
+
+app = FastAPI()
+
+TEMP_DIR = "/tmp/ffmpeg_processing"
+
+@app.on_event("startup")
+async def startup_event():
+    os.makedirs(TEMP_DIR, exist_ok=True)
+
+@app.on_event("shutdown")
+async def shutdown_event():
+    # Limpa o diretório temporário ao desligar, opcional para Spaces persistentes
+    shutil.rmtree(TEMP_DIR, ignore_errors=True)
+    os.makedirs(TEMP_DIR, exist_ok=True) # Recria para a próxima inicialização
+
+@app.post("/process-video/")
+async def process_video(
+    file: UploadFile = File(...),
+    output_filename: str = Form(...),
+    ffmpeg_command_args: str = Form(...)
+):
+    """
+    Processa um vídeo usando FFmpeg.
+    Espera um arquivo de vídeo e argumentos de comando FFmpeg.
+    """
+    unique_id = str(uuid.uuid4())
+    input_path = os.path.join(TEMP_DIR, f"{unique_id}_{file.filename}")
+    output_path = os.path.join(TEMP_DIR, f"{unique_id}_{output_filename}")
+
+    try:
+        # Salva o arquivo enviado
+        with open(input_path, "wb") as buffer:
+            shutil.copyfileobj(file.file, buffer)
+
+        # Monta o comando FFmpeg
+        # CUIDADO: ffmpeg_command_args deve ser sanitizado ou controlado no n8n
+        # para evitar injeção de comandos maliciosos.
+        command = f"ffmpeg -i {input_path} {ffmpeg_command_args} {output_path}"
+        
+        # Executa o comando
+        process = await asyncio.create_subprocess_shell(
+            command,
+            stdout=asyncio.subprocess.PIPE,
+            stderr=asyncio.subprocess.PIPE
+        )
+        stdout, stderr = await process.communicate()
+
+        if process.returncode != 0:
+            print(f"FFmpeg Stdout: {stdout.decode()}")
+            print(f"FFmpeg Stderr: {stderr.decode()}")
+            raise HTTPException(status_code=500, detail=f"FFmpeg error: {stderr.decode()}")
+
+        # Retorna o arquivo processado
+        return FileResponse(
+            path=output_path,
+            filename=output_filename,
+            media_type="application/octet-stream" # ou o tipo MIME correto do seu output
+        )
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
+    finally:
+        # Limpa os arquivos temporários
+        if os.path.exists(input_path):
+            os.remove(input_path)
+        if os.path.exists(output_path):
+            os.remove(output_path)
+
+@app.post("/get-duration/")
+async def get_duration(file: UploadFile = File(...)):
+    """
+    Obtém a duração de um arquivo de áudio/vídeo usando ffprobe.
+    """
+    unique_id = str(uuid.uuid4())
+    input_path = os.path.join(TEMP_DIR, f"{unique_id}_{file.filename}")
+
+    try:
+        with open(input_path, "wb") as buffer:
+            shutil.copyfileobj(file.file, buffer)
+
+        command = f"ffprobe -i {input_path} -show_entries format=duration -v quiet -of csv='p=0'"
+        process = await asyncio.create_subprocess_shell(
+            command,
+            stdout=asyncio.subprocess.PIPE,
+            stderr=asyncio.subprocess.PIPE
+        )
+        stdout, stderr = await process.communicate()
+
+        if process.returncode != 0:
+            print(f"FFprobe Stdout: {stdout.decode()}")
+            print(f"FFprobe Stderr: {stderr.decode()}")
+            raise HTTPException(status_code=500, detail=f"FFprobe error: {stderr.decode()}")
+
+        duration = float(stdout.decode().strip())
+        return JSONResponse(content={"duration_seconds": duration})
+
+    except ValueError:
+        raise HTTPException(status_code=400, detail="Could not parse duration from ffprobe output.")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
+    finally:
+        if os.path.exists(input_path):
+            os.remove(input_path)
+
+@app.get("/")
+async def read_root():
+    return {"message": "FFmpeg FastAPI service is running!"}