Update api/ltx_server.py

api/ltx_server.py

@@ -436,6 +436,19 @@ class VideoService:
         print("================PODA CAUSAL=================")
         return chunks
 
+
+    def _get_total_frames(self, video_path):
+        cmd = [
+            "ffprobe", "-v", "error",
+            "-select_streams", "v:0",
+            "-count_frames",
+            "-show_entries", "stream=nb_read_frames",
+            "-of", "default=nokey=1:noprint_wrappers=1",
+            video_path
+        ]
+        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)
@@ -457,16 +470,18 @@ class VideoService:
     
             # --- PODA ---
             video_podado = os.path.join(pasta, f"{base}_podado_{i}.mp4")
-            trim_ini = poda if i > 0 else 0
-            trim_fim = -poda if i < total_partes - 1 else 0
-            filtro_trim = f"trim=start_frame={trim_ini}"
-            if trim_fim != 0:
-                filtro_trim += f":end_frame=nb_frames+{trim_fim}"
-            cmd_poda = f'ffmpeg -y -hide_banner -loglevel error -i "{video_clone}" -vf "{filtro_trim},setpts=PTS-STARTPTS" -an "{video_podado}"'
+            total_frames = self.get_total_frames(video_clone)
+            trim_ini = 0
+            trim_end = total_frames - poda   # para cortar últimos frames
+            cmd_poda = (
+                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}"'
+            )
             subprocess.run(cmd_poda, shell=True, check=True)
             nova_lista.append(video_podado)
             print(f"[DEBUG] [{i}] Podado -> {video_podado}")
-    
+            
             # --- FADE_INI ---
             video_fade_ini = None
             if i > 0: