Update api/ltx_server.py

api/ltx_server.py

@@ -435,10 +435,66 @@ class VideoService:
         print("================PODA CAUSAL=================")
         return chunks
 
-        
+
+    def _concat_crossfade_cascade(self, video_paths: List[str], crossfade_frames: int = 8, output_path: str = "/tmp/final_output.mp4"):
+        """
+        Concatena uma lista de vídeos aplicando crossfade de 'crossfade_frames' entre cada par.
+        Args:
+            video_paths: lista de caminhos dos vídeos a concatenar
+            crossfade_frames: quantidade de frames para o crossfade
+            output_path: caminho do vídeo final gerado
+        """
+        if not video_paths or len(video_paths) == 1:
+            # Apenas copiar se houver 1 ou nenhum vídeo
+            if video_paths:
+                subprocess.run(f"cp '{video_paths[0]}' '{output_path}'", shell=True, check=True)
+            return output_path
+    
+        # Lista temporária de vídeos intermediários
+        temp_videos = []
+    
+        # Começamos com o primeiro vídeo
+        prev_video = video_paths[0]
+    
+        for i in range(1, len(video_paths)):
+            next_video = video_paths[i]
+    
+            # Cria arquivo temporário para resultado do crossfade
+            temp_out = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
+            temp_out_path = temp_out.name
+            temp_out.close()
+    
+            # Comando FFmpeg
+            cmd = f"""
+            ffmpeg -y -i "{prev_video}" -i "{next_video}" -filter_complex "
+            [0:v]trim=0:-{crossfade_frames},setpts=PTS-STARTPTS[v_prev_main];
+            [0:v]trim=-{crossfade_frames},setpts=PTS-STARTPTS[v_prev_fade];
+            [1:v]trim=0:{crossfade_frames},setpts=PTS-STARTPTS[v_next_fade];
+            [1:v]trim={crossfade_frames}:,setpts=PTS-STARTPTS[v_next_main];
+            [v_prev_fade][v_next_fade]blend=all_expr='A*(1-T/{crossfade_frames})+B*(T/{crossfade_frames})'[crossfade];
+            [v_prev_main][crossfade][v_next_main]concat=n=3:v=1:a=0[v]
+            " -map "[v]" -c:v libx264 -pix_fmt yuv420p "{temp_out_path}"
+            """
+    
+            # Executa FFmpeg
+            subprocess.run(cmd, shell=True, check=True)
+    
+            # Atualiza prev_video para o próximo crossfade
+            prev_video = temp_out_path
+            temp_videos.append(temp_out_path)
+    
+        # Move o último vídeo processado para o destino final
+        os.rename(prev_video, output_path)
+    
+        # Remove vídeos temporários antigos
+        for tmp in temp_videos[:-1]:
+            if os.path.exists(tmp):
+                os.remove(tmp)
+    
+        return output_path
 
     
-    def _concat_crossfade_cascade(self, video_paths: List[str], output_path: str, crossfade_frames: int = 8):
+    def _concat_crossfade_cascade1(self, video_paths: List[str], output_path: str, crossfade_frames: int = 8):
         """
         Concatena vídeos em cascata com crossfade, usando FFmpeg.