Update deformes4D_engine.py

deformes4D_engine.py

@@ -250,7 +250,57 @@ class Deformes4DEngine:
             new_full_latents = self._generate_latent_tensor_internal(conditioning_items, current_ltx_params, target_resolution_tuple, total_frames_to_generate)
 
 
+            
+            # --- CÓDIGO ADICIONADO PARA ITERAR E CONTAR FRAMES ---
+            logger.info("Iniciando a contagem de frames para cada tensor latente individual.")
+            total_frames_from_latents = 0
+            for i in range(new_full_latents.shape[2]):
+                single_latent_tensor = new_full_latents[:, :, i:i+1, :, :]
+                temp_video_path = os.path.join(self.workspace_dir, f"temp_latent_video_{i}.mp4")
+                
+                # Gerar um vídeo temporário para o tensor latente individual
+                pixel_tensor_single = self.latents_to_pixels(single_latent_tensor)
+                self.save_video_from_tensor(pixel_tensor_single, temp_video_path, fps=24)
+                del pixel_tensor_single
+                gc.collect()
 
+                # Contar os frames do vídeo temporário com ffprobe
+                try:
+                    frame_count_result = subprocess.run(
+                        ["ffprobe", "-v", "error", "-select_streams", "v:0", "-count_frames", "-show_entries", "stream=nb_read_frames", "-of", "default=noprint_wrappers=1:nokey=1", temp_video_path],
+                        capture_output=True, text=True, check=True
+                    )
+                    frames = int(frame_count_result.stdout.strip())
+                    logger.info(f"Latente {i}: Vídeo temporário gerado com {frames} frames.")
+                    total_frames_from_latents += frames
+                except (subprocess.CalledProcessError, ValueError, FileNotFoundError) as e:
+                    logger.error(f"Falha ao contar frames para o latente {i} com ffprobe: {e}")
+                finally:
+                    if os.path.exists(temp_video_path):
+                        os.remove(temp_video_path)
+            
+            logger.info(f"Contagem total de frames a partir dos latentes individuais: {total_frames_from_latents}")
+
+            # Gerar e contar frames para o vídeo completo a partir do tensor latente inteiro
+            full_latent_video_path = os.path.join(self.workspace_dir, "temp_full_latent_video.mp4")
+            full_pixel_tensor = self.latents_to_pixels(new_full_latents)
+            self.save_video_from_tensor(full_pixel_tensor, full_latent_video_path, fps=24)
+            del full_pixel_tensor
+            gc.collect()
+
+            try:
+                full_frame_count_result = subprocess.run(
+                    ["ffprobe", "-v", "error", "-select_streams", "v:0", "-count_frames", "-show_entries", "stream=nb_read_frames", "-of", "default=noprint_wrappers=1:nokey=1", full_latent_video_path],
+                    capture_output=True, text=True, check=True
+                )
+                total_frames_full_video = int(full_frame_count_result.stdout.strip())
+                logger.info(f"Vídeo completo gerado a partir do tensor latente inteiro possui {total_frames_full_video} frames.")
+            except (subprocess.CalledProcessError, ValueError, FileNotFoundError) as e:
+                logger.error(f"Falha ao contar frames do vídeo completo com ffprobe: {e}")
+            finally:
+                if os.path.exists(full_latent_video_path):
+                    os.remove(full_latent_video_path)
+            # --- FIM DO CÓDIGO ADICIONADO ---```