Update api/seedvr_server.py

api/seedvr_server.py

@@ -22,22 +22,31 @@ if str(SEEDVR_REPO_PATH) not in sys.path:
 # Tenta importar as funções necessárias APÓS a modificação do path.
 # Se falhar, a aplicação não pode continuar.
 try:
-    from inference_cli import run_inference_logic, save_frames_to_vide
+    from inference_cli import run_inference_logic, save_frames_to_video
+except ImportError as e:
+    print(f"ERRO FATAL: Não foi possível importar de 'inference_cli.py'.")
+    print(f"Verifique se o repositório em '{SEEDVR_REPO_PATH}' está correto e completo.")
+    raise e
 
-    self.SEEDVR_ROOT = SEEDVR_REPO_PATH
-    self.CKPTS_ROOT = Path("/data/seedvr_models_fp16") 
-    self.OUTPUT_ROOT = Path(os.getenv("OUTPUT_ROOT", "/app/outputs"))
-    self.INPUT_ROOT = Path(os.getenv("INPUT_ROOT", "/app/inputs"))
-    self.HF_HOME_CACHE = Path(os.getenv("HF_HOME", "/data/.cache/huggingface"))
-    self.REPO_URL = os.getenv("SEEDVR_GIT_URL", "https://github.com/numz/ComfyUI-SeedVR2_VideoUpscaler")
-    self.NUM_GPUS_TOTAL = 4# int(os.getenv("NUM_GPUS", "8"))
+class SeedVRServer:
+    def __init__(self, **kwargs):
+        """
+        Inicializa o servidor, define os caminhos e prepara o ambiente.
+        """
+        self.SEEDVR_ROOT = SEEDVR_REPO_PATH
+        self.CKPTS_ROOT = Path("/data/seedvr_models_fp16") 
+        self.OUTPUT_ROOT = Path(os.getenv("OUTPUT_ROOT", "/app/outputs"))
+        self.INPUT_ROOT = Path(os.getenv("INPUT_ROOT", "/app/inputs"))
+        self.HF_HOME_CACHE = Path(os.getenv("HF_HOME", "/data/.cache/huggingface"))
+        self.REPO_URL = os.getenv("SEEDVR_GIT_URL", "https://github.com/numz/ComfyUI-SeedVR2_VideoUpscaler")
+        self.NUM_GPUS_TOTAL = int(os.getenv("NUM_GPUS", "8"))
         
-    print("🚀 SeedVRServer (Modo de Chamada Direta) inicializando...")
-    for p in [self.CKPTS_ROOT, self.OUTPUT_ROOT, self.INPUT_ROOT, self.HF_HOME_CACHE]:
-        p.mkdir(parents=True, exist_ok=True)
+        print("🚀 SeedVRServer (Modo de Chamada Direta) inicializando...")
+        for p in [self.CKPTS_ROOT, self.OUTPUT_ROOT, self.INPUT_ROOT, self.HF_HOME_CACHE]:
+            p.mkdir(parents=True, exist_ok=True)
         
-    self.setup_dependencies()
-    print("✅ SeedVRServer (Modo de Chamada Direta) pronto.")
+        self.setup_dependencies()
+        print("✅ SeedVRServer (Modo de Chamada Direta) pronto.")
 
     def setup_dependencies(self):
         """ Garante que o repositório e os modelos estão presentes. """
@@ -69,22 +78,6 @@ try:
                 )
         print("[SeedVRServer] Checkpoints (FP16) estão no local correto.")
         
-except ImportError as e:
-    print(f"ERRO FATAL: Não foi possível importar de 'inference_cli.py'.")
-    print(f"Verifique se o repositório em '{SEEDVR_REPO_PATH}' está correto e completo.")
-    raise e
-
-
-
-
-
-class SeedVRServer:
-    def __init__(self, **kwargs):
-        """
-        Inicializa o servidor, define os caminhos e prepara o ambiente.
-        """
-        
-            
     def run_inference_direct(
         self,
         file_path: str, *,
@@ -110,7 +103,7 @@ class SeedVRServer:
             batch_size=sp_size,
             model="seedvr2_ema_7b_sharp_fp16.safetensors",
             preserve_vram=True,
-            debug=true, # Mantém o debug ativo para logs detalhados.
+            debug=False, # Mantém o debug ativo para logs detalhados.
             cuda_device=",".join(map(str, range(self.NUM_GPUS_TOTAL))),
             skip_first_frames=0,
             load_cap=0,