Update api/ltx_server.py

api/ltx_server.py

@@ -447,21 +447,37 @@ class VideoService:
         
         return primeira, segunda
         
-    def _concat_mp4s_no_reencode(self, mp4_a: str, mp4_b: str, out_path: str):
-        # Concat demuxer do ffmpeg (sem reencode)
+    def _concat_mp4s_no_reencode(self, mp4_paths: List[str], out_path: str):
+        # Valida entradas
+        if not mp4_paths or len(mp4_paths) < 2:
+            raise ValueError("Forneça pelo menos dois MP4s para concatenação.")  # [memory:5]
+        abs_paths = []
+        for p in mp4_paths:
+            if not os.path.exists(p):
+                raise FileNotFoundError(f"Arquivo não encontrado: {p}")  # [memory:5]
+            abs_paths.append(os.path.abspath(p))  # [memory:5]
+    
+        # Cria lista para o demuxer concat (sem reencode)
+        list_path = None
         with tempfile.NamedTemporaryFile("w", delete=False, suffix=".txt") as f:
-            f.write(f"file '{os.path.abspath(mp4_a)}'\n")
-            f.write(f"file '{os.path.abspath(mp4_b)}'\n")
-            list_path = f.name
-            cmd = f"ffmpeg -y -f concat -safe 0 -i {list_path} -c copy {out_path}"
-            print(f"[DEBUG] Concat: {cmd}")
+            for ap in abs_paths:
+                # Escapa aspas simples no caminho se necessário
+                ap_escaped = ap.replace("'", r"'\''")  # [memory:5]
+                f.write(f"file '{ap_escaped}'\n")  # [memory:5]
+            list_path = f.name  # [memory:5]
+    
+        cmd = f"ffmpeg -y -f concat -safe 0 -i {list_path} -c copy {out_path}"  # [memory:5]
+        print(f"[DEBUG] Concat: {cmd}")  # [memory:8]
+    
         try:
-            subprocess.check_call(shlex.split(cmd))
+            subprocess.check_call(shlex.split(cmd))  # [memory:5]
         finally:
-            try: os.remove(list_path)
-            except Exception: pass
-            
-    # --- 6. GERAÇÃO ---
+            if list_path:
+                try:
+                    os.remove(list_path)  # [memory:5]
+                except Exception:
+                    pass  # [memory:5]    # --- 6. GERAÇÃO ---
+  
     def generate(
         self,
         prompt,
@@ -690,12 +706,6 @@ 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[0], partes_mp4[1], final_concat)
-            self._concat_mp4s_no_reencode(final_concat, partes_mp4[2], final_concat)
-            self._concat_mp4s_no_reencode(final_concat, partes_mp4[3], final_concat)
-
-            
             
             self._log_gpu_memory("Fim da Geração")
             return final_concat, used_seed