Update deformes4D_engine.py

deformes4D_engine.py

@@ -100,23 +100,23 @@ class Deformes4DEngine:
         self.save_video_from_tensor(pixel_tensor, silent_video_path, fps=24)
         del pixel_tensor; gc.collect()
         
-        try:
-            result = subprocess.run(
-                ["ffprobe", "-v", "error", "-show_entries", "format=duration", "-of", "default=noprint_wrappers=1:nokey=1", silent_video_path],
-                capture_output=True, text=True, check=True)
-            frag_duration = float(result.stdout.strip())
-        except (subprocess.CalledProcessError, ValueError, FileNotFoundError):
-             logger.warning(f"ffprobe falhou em {os.path.basename(silent_video_path)}. Calculando duração manualmente.")
-             num_pixel_frames = latent_tensor.shape[2] * 8
-             frag_duration = num_pixel_frames / 24.0
-
-        video_with_audio_path = audio_specialist_singleton.generate_audio_for_video(
-            video_path=silent_video_path, prompt=audio_prompt,
-            duration_seconds=frag_duration)
-        
-        if os.path.exists(silent_video_path):
-             os.remove(silent_video_path)
-        return video_with_audio_path
+        #try:
+        #    result = subprocess.run(
+        #        ["ffprobe", "-v", "error", "-show_entries", "format=duration", "-of", "default=noprint_wrappers=1:nokey=1", silent_video_path],
+        #        capture_output=True, text=True, check=True)
+        #    frag_duration = float(result.stdout.strip())
+        #except (subprocess.CalledProcessError, ValueError, FileNotFoundError):
+        #     logger.warning(f"ffprobe falhou em {os.path.basename(silent_video_path)}. Calculando duração manualmente.")
+        #     num_pixel_frames = latent_tensor.shape[2] * 8
+        #     frag_duration = num_pixel_frames / 24.0
+        #
+        #video_with_audio_path = audio_specialist_singleton.generate_audio_for_video(
+        #    video_path=silent_video_path, prompt=audio_prompt,
+        #    duration_seconds=frag_duration)
+        #
+        #if os.path.exists(silent_video_path):
+        #     os.remove(silent_video_path)
+        return silent_video_path
 
     def _generate_latent_tensor_internal(self, conditioning_items, ltx_params, target_resolution, total_frames_to_generate):
         final_ltx_params = {
@@ -227,6 +227,8 @@ class Deformes4DEngine:
             current_ltx_params = {**base_ltx_params, "handler_strength": handler_strength, "motion_prompt": motion_prompt}
             total_frames_to_generate = self._quantize_to_multiple(int(seconds_per_fragment * 24), 8) + 1
 
+            
+            
             # --- NOVA LÓGICA: Preparação das instruções de condicionamento ---
             if is_first_fragment:
                 img_start = self._preprocess_image_for_latent_conversion(Image.open(start_keyframe_path).convert("RGB"), target_resolution_tuple)
@@ -246,6 +248,38 @@ class Deformes4DEngine:
                     conditioning_items.append(LatentConditioningItem(destination_latent, total_frames_to_generate - 1, destination_convergence_strength))
             
             new_full_latents = self._generate_latent_tensor_internal(conditioning_items, current_ltx_params, target_resolution_tuple, total_frames_to_generate)
+
+
+            # --- [INÍCIO] Bloco de Verificação de Frames por Chunk ---
+            logger.info("--- [VERIFICAÇÃO DE CHUNKS INDIVIDUAIS] ---")
+            total_chunks_verificados = new_full_latents.shape[2]
+            for chunk_idx in range(total_chunks_verificados):
+                try:
+                    # Isola o chunk atual
+                    single_chunk_latent = new_full_latents[:, :, chunk_idx:chunk_idx+1, :, :]
+                    
+                    # Gera um nome de arquivo temporário para o vídeo do chunk
+                    temp_video_base_name = f"debug_chunk_{chunk_idx}"
+                    
+                    # Converte o latente do chunk em um vídeo MP4
+                    temp_video_path = self._generate_video_from_latents(single_chunk_latent, temp_video_base_name)
+                    
+                    # Conta os frames no vídeo gerado
+                    if os.path.exists(temp_video_path):
+                        with imageio.get_reader(temp_video_path) as reader:
+                            frame_count = reader.count_frames()
+                        logger.info(f"  - VERIFICADO: Chunk {chunk_idx} gerou um vídeo com {frame_count} frames.")
+                        # Apaga o vídeo de debug
+                        os.remove(temp_video_path)
+                    else:
+                        logger.warning(f"  - FALHA: Não foi possível gerar o vídeo para o Chunk {chunk_idx}.")
+                
+                except Exception as e:
+                    logger.error(f"  - ERRO ao verificar Chunk {chunk_idx}: {e}")
+            logger.info("--- [FIM DA VERIFICAÇÃO] ---")
+            # --- [FIM] Bloco de Verificação ---
+            
+
             
             # --- NOVA LÓGICA: Preparação movida para o final do loop ---
             is_last_fragment = (i == num_transitions_to_generate - 1)