Update api/ltx_server.py

api/ltx_server.py

@@ -22,6 +22,7 @@ LTXV_FRAME_LOG_EVERY=8
 
 
 # --- 1. IMPORTAÇÕES ---
+import os, subprocess, shlex, tempfile
 import torch
 import numpy as np
 import random
@@ -479,9 +480,64 @@ class VideoService:
     
         print("================PODA CAUSAL=================")
         return chunks
-
-
         
+    def _concat_crossfade_videos_por_frames(self, mp4_list, out_path, fps=24, crossfade_frames=8):
+        """
+        Concatena múltiplos vídeos MP4 aplicando crossfade (blend) controlado por número de frames.
+        Não inclui áudio.
+        
+        Args:
+            mp4_list (List[str]): Lista de vídeos a concatenar.
+            out_path (str): Caminho do vídeo final.
+            fps (int): Taxa de quadros (todos devem ter o mesmo FPS).
+            crossfade_frames (int): Número de frames usados no fade entre vídeos consecutivos.
+        """
+        if len(mp4_list) < 2:
+            raise ValueError("Forneça ao menos dois vídeos para concatenar com crossfade.")
+        
+        inputs = " ".join([f"-i '{v}'" for v in mp4_list])
+        filter_parts = []
+        concat_refs = []
+    
+        for i in range(len(mp4_list)):
+            v = f"[{i}:v]"
+            filter_parts.append(f"{v}setpts=PTS-STARTPTS[v{i}]")
+    
+        # Agora criamos os blends entre pares consecutivos
+        for i in range(len(mp4_list) - 1):
+            a = f"[v{i}]"
+            b = f"[v{i+1}]"
+    
+            fade = f"[v{i}b{i+1}]blend=all_expr='A*(1-T/{crossfade_frames})+B*(T/{crossfade_frames})'[xf{i}]"
+            filter_parts.append(
+                f"{a}trim=end_frame={-crossfade_frames},setpts=PTS-STARTPTS[v{i}pre];"
+                f"{a}trim=start_frame={-crossfade_frames},setpts=PTS-STARTPTS[v{i}fade];"
+                f"{b}trim=end_frame={crossfade_frames},setpts=PTS-STARTPTS[v{i+1}fade];"
+                f"{a}trim=end_frame={-crossfade_frames},setpts=PTS-STARTPTS[v{i}pre];"
+                f"[v{i}fade][v{i+1}fade]{fade};"
+            )
+            concat_refs.append(f"[v{i}pre][xf{i}]")
+    
+        # Último vídeo (sem blend)
+        concat_refs.append(f"[v{len(mp4_list) - 1}]")
+    
+        filter_complex = (
+            ";".join(filter_parts)
+            + ";"
+            + "".join(concat_refs)
+            + f"concat=n={len(mp4_list) * 2 - 1}:v=1:a=0[v]"
+        )
+    
+        cmd = (
+            f"ffmpeg -y {inputs} -filter_complex \"{filter_complex}\" "
+            f"-map \"[v]\" -c:v libx264 -pix_fmt yuv420p \"{out_path}\""
+        )
+    
+        print(f"[DEBUG] Crossfade concat:\n{cmd}\n")
+        subprocess.run(cmd, shell=True, check=True)
+        print(f"✅ Vídeo final criado: {out_path}")
+
+    
     def _concat_mp4s_no_reencode(self, mp4_list: List[str], out_path: str):
         """
         Concatena múltiplos MP4s sem reencode usando o demuxer do ffmpeg.
@@ -736,8 +792,14 @@ class VideoService:
                     print(f"[DEBUG] Falha no move; usando tmp como final: {e}")
 
             final_concat = os.path.join(results_dir, f"concat_fim_{used_seed}.mp4")
-            self._concat_mp4s_no_reencode(partes_mp4, final_concat)
-
+            #self._concat_mp4s_no_reencode(partes_mp4, final_concat)
+            self._concat_crossfade_videos_por_frames(
+                partes_mp4, 
+                final_concat,
+                fps=24,
+                crossfade_frames=8,
+            )
+            
             self._log_gpu_memory("Fim da Geração")
             return final_concat, used_seed