Update app.py

adição de legendas .ass

app.py

@@ -220,3 +220,86 @@ async def get_duration(file: UploadFile = File(...)):
             os.remove(temp_input_path)
             print(f"Cleaned up temporary file: {temp_input_path}")
 
+# ==============================================================================
+# 4. ENDPOINT PARA ADICIONAR LEGENDAS
+# ==============================================================================
+
+@app.post("/add-subtitles/")
+async def add_subtitles(
+    video_file: UploadFile = File(...),
+    subtitle_file: UploadFile = File(...)
+):
+    """
+    Adiciona ('queima') um arquivo de legenda (.ass) a um vídeo.
+
+    Este endpoint espera o upload de dois arquivos: o vídeo base e o
+    arquivo de legenda. Ele usa o filtro 'ass' do FFmpeg para renderizar
+    as legendas diretamente no vídeo.
+
+    Args:
+        video_file (UploadFile): O arquivo de vídeo de entrada (ex: .mp4).
+        subtitle_file (UploadFile): O arquivo de legenda no formato .ass.
+
+    Returns:
+        FileResponse: O vídeo final com as legendas "queimadas".
+    """
+    # --- Setup do Ambiente Temporário ---
+    unique_id = str(uuid.uuid4())
+    temp_processing_dir = os.path.join(TEMP_DIR, unique_id)
+    os.makedirs(temp_processing_dir)
+    
+    # Define os caminhos para os arquivos de entrada e saída
+    video_input_path = os.path.join(temp_processing_dir, video_file.filename)
+    subtitle_input_path = os.path.join(temp_processing_dir, "subtitle.ass") # Nome padronizado
+    final_output_path = os.path.join(temp_processing_dir, f"video_with_subtitles.mp4")
+
+    try:
+        # --- ETAPA 1: Salvar os arquivos recebidos no disco ---
+        print("--- STEP 1 (Subtitles): Saving uploaded files ---")
+        # Salva o vídeo
+        with open(video_input_path, "wb") as buffer:
+            shutil.copyfileobj(video_file.file, buffer)
+        print(f"Video saved to: {video_input_path}")
+        
+        # Salva a legenda
+        with open(subtitle_input_path, "wb") as buffer:
+            shutil.copyfileobj(subtitle_file.file, buffer)
+        print(f"Subtitle saved to: {subtitle_input_path}")
+        
+        # --- ETAPA 2: Construir e Executar o Comando FFmpeg ---
+        print("--- STEP 2 (Subtitles): Executing FFmpeg command ---")
+        
+        # O filtro `ass` precisa de um caminho de arquivo. Como o ffmpeg é executado
+        # a partir da raiz, precisamos fornecer o caminho completo. Para evitar problemas
+        # com espaços ou caracteres especiais, usamos shlex.quote.
+        # O filtro do Windows pode precisar de uma sintaxe de escape diferente,
+        # mas para o ambiente Linux do Hugging Face, isto é o ideal.
+        escaped_subtitle_path = subtitle_input_path.replace("'", "'\\''")
+
+        command = (
+            f"ffmpeg -i {shlex.quote(video_input_path)} "
+            f"-vf \"ass='{escaped_subtitle_path}'\" "
+            f"-c:v libx264 -preset ultrafast -c:a copy "
+            f"{shlex.quote(final_output_path)}"
+        )
+        
+        await run_subprocess(command)
+        
+        print(f"--- SUCCESS: Final video with subtitles created at {final_output_path} ---")
+
+        # --- ETAPA 3: Retornar o arquivo e agendar a limpeza ---
+        cleanup_task = BackgroundTask(cleanup_directory, directory_path=temp_processing_dir)
+        return FileResponse(
+            path=final_output_path,
+            filename="video_com_legenda.mp4",
+            media_type="video/mp4",
+            background=cleanup_task
+        )
+
+    except Exception as e:
+        print(f"\n--- AN ERROR OCCURRED in /add-subtitles/ ---")
+        print(f"Error Details: {e}")
+        print("-------------------------------------------\n")
+        cleanup_directory(directory_path=temp_processing_dir)
+        raise HTTPException(status_code=500, detail=str(e))            
+