Update api/ltx_server.py

api/ltx_server.py

@@ -436,10 +436,10 @@ class VideoService:
         print("================PODA CAUSAL=================")
         return chunks
 
-
-    def _get_total_frames(self, video_path):
+    def get_total_frames(video_path: str) -> int:
         cmd = [
-            "ffprobe", "-v", "error",
+            "ffprobe",
+            "-v", "error",
             "-select_streams", "v:0",
             "-count_frames",
             "-show_entries", "stream=nb_read_frames",
@@ -448,7 +448,8 @@ class VideoService:
         ]
         result = subprocess.run(cmd, capture_output=True, text=True, check=True)
         return int(result.stdout.strip())    
-
+        
+        
     def _gerar_lista_com_transicoes(self, pasta: str, video_paths: list[str], crossfade_frames: int = 8) -> list[str]:
         """
         Gera uma nova lista de vídeos aplicando transições suaves (blend frame a frame)
@@ -470,17 +471,19 @@ class VideoService:
     
             # --- PODA ---
             video_podado = os.path.join(pasta, f"{base}_podado_{i}.mp4")
-            total_frames = self._get_total_frames(video_clone)
-            trim_ini = 0
-            trim_end = total_frames - poda   # para cortar últimos frames
-            cmd_poda = (
+            total_frames = get_total_frames(video_clone)
+            trim_ini = total_frames - poda  # posição inicial do fade
+            video_fade_fim = os.path.join(pasta, f"{base}_fade_fim_{i}.mp4")
+            
+            cmd_fim = (
                 f'ffmpeg -y -hide_banner -loglevel error -i "{video_clone}" '
-                f'-vf "trim=start_frame={trim_ini}:end_frame={trim_end},setpts=PTS-STARTPTS" '
-                f'-an "{video_podado}"'
+                f'-vf "trim=start_frame={trim_ini}:end_frame={total_frames},setpts=PTS-STARTPTS" '
+                f'-an "{video_fade_fim}"'
             )
-            subprocess.run(cmd_poda, shell=True, check=True)
-            nova_lista.append(video_podado)
-            print(f"[DEBUG] [{i}] Podado -> {video_podado}")
+            subprocess.run(cmd_fim, shell=True, check=True)
+            video_anterior_fade_fim = video_fade_fim
+            print(f"[DEBUG] Fade_fim preparado -> {video_fade_fim}")
+            
             
             # --- FADE_INI ---
             video_fade_ini = None