Update ltx_upscaler_manager_helpers.py

ltx_upscaler_manager_helpers.py

@@ -1,4 +1,3 @@
-#--- START OF MODIFIED FILE app_fluxContext_Ltx/ltx_upscaler_manager_helpers.py ---
 # ltx_upscaler_manager_helpers.py
 # Gerente de Pool para o revezamento de workers de Upscaling.
 # Este arquivo é parte do projeto Euia-AducSdr e está sob a licença AGPL v3.
@@ -49,22 +48,21 @@ class LtxUpscalerPoolManager:
             self.current_worker_index = (self.current_worker_index + 1) % len(self.workers)
             return worker_to_use
 
-    def upscale_latents_to_video(self, latent_path: str, output_path: str, video_fps: int):
-        """Seleciona um worker livre e faz o upscale de um arquivo de latentes para vídeo."""
+    def upscale_video_fragment(self, video_path_low_res: str, output_path: str, video_fps: int):
+        """Seleciona um worker livre e faz o upscale de um vídeo para vídeo."""
         worker_to_use = self._get_worker()
-        try {
-            print(f"UPSCALER POOL MANAGER: Worker em {worker_to_use.device} iniciando upscale de {os.path.basename(latent_path)}...")
-            result_path = worker_to_use.upscale_latents_to_video(latent_path, output_path, video_fps)
-            print(f"UPSCALER POOL MANAGER: Upscale de {os.path.basename(latent_path)} concluído.")
+        try:
+            print(f"UPSCALER POOL MANAGER: Worker em {worker_to_use.device} iniciando upscale de {os.path.basename(video_path_low_res)}...")
+            result_path = worker_to_use.upscale_video_fragment(video_path_low_res, output_path, video_fps)
+            print(f"UPSCALER POOL MANAGER: Upscale de {os.path.basename(video_path_low_res)} concluído.")
             return result_path
         finally:
-            # A limpeza do worker será feita na próxima chamada
             pass
 
     def decode_single_latent_frame(self, latent_frame_tensor: torch.Tensor) -> Image.Image:
         """Seleciona um worker livre e decodifica um único frame latente para uma imagem PIL."""
         worker_to_use = self._get_worker()
-        try {
+        try:
             print(f"UPSCALER POOL MANAGER: Worker em {worker_to_use.device} decodificando frame latente...")
             image = worker_to_use.decode_single_latent_frame(latent_frame_tensor)
             print(f"UPSCALER POOL MANAGER: Decodificação concluída.")
@@ -72,6 +70,4 @@ class LtxUpscalerPoolManager:
         finally:
             pass
 
-# --- Instância Singleton do Gerenciador de Upscaling ---
-ltx_upscaler_manager_singleton = LtxUpscalerPoolManager(device_ids=['cuda:0', 'cuda:1'])
-#--- END OF MODIFIED FILE app_fluxContext_Ltx/ltx_upscaler_manager_helpers.py ---
+ltx_upscaler_manager_singleton = LtxUpscalerPoolManager(device_ids=['cuda:0', 'cuda:1'])