Update audio_specialist.py

audio_specialist.py

@@ -87,7 +87,7 @@ class AudioSpecialist:
         gc.collect()
         if torch.cuda.is_available(): torch.cuda.empty_cache()
 
-    def generate_audio_for_video(self, video_path: str, prompt: str, duration_seconds: float) -> str:
+    def generate_audio_for_video(self, video_path: str, prompt: str, duration_seconds: float, output_path_override: str = None) -> str:
         """
         Gera áudio para um arquivo de vídeo, aplicando um prompt negativo para evitar fala.
 
@@ -143,8 +143,8 @@ class AudioSpecialist:
                 audio_waveform = audios.float().cpu()[0]
                 
                 fragment_name = Path(video_path).stem
-                output_video_path = os.path.join(self.workspace_dir, f"{fragment_name}_com_audio.mp4")
-                
+                output_video_path = output_path_override if output_path_override else os.path.join(self.workspace_dir, f"{fragment_name}_com_audio.mp4")
+
                 make_video(video_info, Path(output_video_path), audio_waveform, sampling_rate=self.seq_cfg.sampling_rate)
                 logger.info(f"--- Fragmento com áudio salvo em: {os.path.basename(output_video_path)}")
                 logger.info("------------------------------------------------------")