Update api/seedvr_server.py

api/seedvr_server.py

@@ -22,31 +22,22 @@ 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_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
+    from inference_cli import run_inference_logic, save_frames_to_vide
 
-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"))
+    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"))
         
-        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. """
@@ -78,6 +69,22 @@ class SeedVRServer:
                 )
         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, *,
@@ -103,7 +110,7 @@ class SeedVRServer:
             batch_size=sp_size,
             model="seedvr2_ema_7b_sharp_fp16.safetensors",
             preserve_vram=True,
-            debug=False, # Mantém o debug ativo para logs detalhados.
+            debug=true, # 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,