Update api/ltx_server.py

api/ltx_server.py

@@ -572,6 +572,26 @@ class VideoService:
         result = subprocess.run(cmd, capture_output=True, text=True, check=True)
         return int(result.stdout.strip())    
 
+
+
+    def dividir_latentes(latents_brutos):
+        total = latents_brutos.shape[2]  # dimensão temporal (número de latentes)
+        
+        if total % 2 == 1:  # ÍMPAR
+            # Ex: 11 → primeira 0..5, segunda 5..10
+            cut = total // 2
+            primeira = latents_brutos[:, :, :cut+1, :, :].clone()
+            segunda  = latents_brutos[:, :, cut:, :, :].clone()
+        
+        else:  # PAR
+            # Ex: 12 → primeira 0..5, segunda 5..11
+            cut = total // 2
+            primeira = latents_brutos[:, :, :cut, :, :].clone()
+            segunda  = latents_brutos[:, :, cut-1:, :, :].clone()
+        
+        return primeira, segunda
+
+    
     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)
@@ -652,8 +672,23 @@ class VideoService:
         print("===========CONCATECAO CAUSAL=============")
         print(f"[DEBUG] {nova_lista}")
         return nova_lista
+
+    def _concat_mp4s_no_reencode(self, mp4_a: str, mp4_b: str, out_path: str):
+    # Concat demuxer do ffmpeg (sem reencode)
+    import tempfile, subprocess, shlex, os
+    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}")
+    try:
+        subprocess.check_call(shlex.split(cmd))
+    finally:
+        try: os.remove(list_path)
+        except Exception: pass
         
-    def _concat_mp4s_no_reencode(self, mp4_list: List[str], out_path: str):
+    def _concat_mp4s_no_reencode2(self, mp4_list: List[str], out_path: str):
         """
         Concatena múltiplos MP4s sem reencode usando o demuxer do ffmpeg.
         ATENÇÃO: todos os arquivos precisam ter mesmo codec, fps, resolução etc.
@@ -826,7 +861,16 @@ class VideoService:
                     except Exception:
                          pass
                 
-                    latents_parts_up = self._dividir_latentes_por_tamanho(latents_cpu_up,15,0)
+                    #latents_parts_up = self._dividir_latentes_por_tamanho(latents_cpu_up,15,0)
+
+
+                    lat_aup, lat_bup = dividir_latentes(latents_parts_up)
+                    print(f"[DEBUG] Partição Aup: {tuple(lat_aup.shape)}")
+                    print(f"[DEBUG] Partição Bup: {tuple(lat_bup.shape)}")
+            
+                    latents_parts_up = [lat_aup, lat_bup]
+                
+
                     
                     for latents in latents_parts_up:
                
@@ -898,7 +942,15 @@ class VideoService:
                 except Exception:
                     pass
                     
-                latents_parts_vae = self._dividir_latentes_por_tamanho(latents_cpu_vae,4,1)
+                #latents_parts_vae = self._dividir_latentes_por_tamanho(latents_cpu_vae,4,1)
+
+
+                lat_a, lat_b = dividir_latentes(latents_cpu_vae)
+                print(f"[DEBUG] Partição A: {tuple(lat_a.shape)}")
+                print(f"[DEBUG] Partição B: {tuple(lat_b.shape)}")
+            
+                latents_parts_vae = [lat_a, lat_b]
+
                 
                 for latents in latents_parts_vae:
                     print(f"[DEBUG] Partição {par}: {tuple(latents.shape)}")
@@ -937,8 +989,8 @@ class VideoService:
             total_partes = len(partes_mp4)
             if (total_partes>1):
                 final_vid = os.path.join(results_dir, f"concat_fim_{used_seed}.mp4")
-                partes_mp4_fade = self._gerar_lista_com_transicoes(pasta=results_dir, video_paths=partes_mp4, crossfade_frames=8)
-                self._concat_mp4s_no_reencode(partes_mp4_fade, final_vid)
+                #partes_mp4_fade = self._gerar_lista_com_transicoes(pasta=results_dir, video_paths=partes_mp4, crossfade_frames=8)
+                self._concat_mp4s_no_reencode(partes_mp4, final_vid)
             else:
                 final_vid = partes_mp4[0]