Update api/seedvr_server.py

api/seedvr_server.py

@@ -28,60 +28,61 @@ except ImportError as e:
     print(f"Verifique se o repositório em '{SEEDVR_REPO_PATH}' está correto e completo.")
     raise e
 
-INIT = False 
+INIT = False
 
 class SeedVRServer:
     def __init__(self, **kwargs):
         """
         Inicializa o servidor, define os caminhos e prepara o ambiente.
         """
-    
-        SEEDVR_ROOT = SEEDVR_REPO_PATH
-        CKPTS_ROOT = Path("/data/seedvr_models_fp16") 
-        OUTPUT_ROOT = Path(os.getenv("OUTPUT_ROOT", "/app/outputs"))
-        INPUT_ROOT = Path(os.getenv("INPUT_ROOT", "/app/inputs"))
-        HF_HOME_CACHE = Path(os.getenv("HF_HOME", "/data/.cache/huggingface"))
-        REPO_URL = os.getenv("SEEDVR_GIT_URL", "https://github.com/numz/ComfyUI-SeedVR2_VideoUpscaler")
-        NUM_GPUS_TOTAL = 4#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 = int(os.getenv("NUM_GPUS", "4"))
+
 
         if INIT:
-            print("⚒️ SeedVRServer (Modo de Chamada Direta) inicializando...")
-            for p in [CKPTS_ROOT, OUTPUT_ROOT, INPUT_ROOT, HF_HOME_CACHE]:
+            print("🚀 SeedVRServer ja inicializando...")
+        else:
+            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)
-                
-            setup_dependencies()
+        
+            self.setup_dependencies()
             print("📦 SeedVRServer (Modo de Chamada Direta) pronto.")
-            init = True
-        else:
-            print("⚙️ SeedVRServer pronto.")
+            INIT = True
+            
 
-    def setup_dependencies():
+    def setup_dependencies(self):
         """ Garante que o repositório e os modelos estão presentes. """
         self._ensure_repo()
         self._ensure_model()
 
     def _ensure_repo(self) -> None:
         """ Clona o repositório do SeedVR se ele não existir. """
-        if not (SEEDVR_ROOT / ".git").exists():
-            print(f"[SeedVRServer] Clonando repositório para {SEEDVR_ROOT}...")
+        if not (self.SEEDVR_ROOT / ".git").exists():
+            print(f"[SeedVRServer] Clonando repositório para {self.SEEDVR_ROOT}...")
             # Usamos subprocess.run aqui porque é uma tarefa de inicialização única.
-            subprocess.run(["git", "clone", "--depth", "1", REPO_URL, str(SEEDVR_ROOT)], check=True)
+            subprocess.run(["git", "clone", "--depth", "1", self.REPO_URL, str(self.SEEDVR_ROOT)], check=True)
         else:
             print("[SeedVRServer] Repositório SeedVR já existe.")
 
     def _ensure_model(self) -> None:
         """ Baixa os checkpoints do Hugging Face se não existirem localmente. """
-        print(f"[SeedVRServer] Verificando checkpoints (FP16) em {CKPTS_ROOT}...")
+        print(f"[SeedVRServer] Verificando checkpoints (FP16) em {self.CKPTS_ROOT}...")
         model_files = {
             "seedvr2_ema_3b_fp16.safetensors": "MonsterMMORPG/SeedVR2_SECourses",
             "ema_vae_fp16.safetensors": "MonsterMMORPG/SeedVR2_SECourses",
         }
         for filename, repo_id in model_files.items():
-            if not (CKPTS_ROOT / filename).exists():
+            if not (self.CKPTS_ROOT / filename).exists():
                 print(f"Baixando {filename} de {repo_id}...")
                 hf_hub_download(
-                    repo_id=repo_id, filename=filename, local_dir=str(CKPTS_ROOT),
-                    cache_dir=str(HF_HOME_CACHE), token=os.getenv("HF_TOKEN")
+                    repo_id=repo_id, filename=filename, local_dir=str(self.CKPTS_ROOT),
+                    cache_dir=str(self.HF_HOME_CACHE), token=os.getenv("HF_TOKEN")
                 )
         print("[SeedVRServer] Checkpoints (FP16) estão no local correto.")
         
@@ -95,7 +96,7 @@ class SeedVRServer:
         Executa a inferência diretamente no mesmo processo e retorna o caminho do arquivo de saída.
         """
         # Cria um diretório de saída único para salvar o resultado.
-        out_dir = OUTPUT_ROOT / f"run_{int(time.time())}_{Path(file_path).stem}"
+        out_dir = self.OUTPUT_ROOT / f"run_{int(time.time())}_{Path(file_path).stem}"
         out_dir.mkdir(parents=True, exist_ok=True)
         output_filepath = out_dir / f"result_{Path(file_path).stem}.mp4"
 
@@ -104,14 +105,14 @@ class SeedVRServer:
         args = SimpleNamespace(
             video_path=file_path,
             output=str(output_filepath),
-            model_dir=str(CKPTS_ROOT),
+            model_dir=str(self.CKPTS_ROOT),
             seed=seed,
             resolution=res_h, # O script do SeedVR usa a altura (lado menor) como referência.
             batch_size=sp_size,
-            model="seedvr2_ema_7b_sharp_fp16.safetensors",
-            preserve_vram=True,
+            model="seedvr2_ema_7b_fp16.safetensors",
+            preserve_vram=False,
             debug=True, # Mantém o debug ativo para logs detalhados.
-            cuda_device=",".join(map(str, range(NUM_GPUS_TOTAL))),
+            cuda_device=",".join(map(str, range(self.NUM_GPUS_TOTAL))),
             skip_first_frames=0,
             load_cap=0,
             output_format='video' # Garante que sempre gere vídeo
@@ -142,7 +143,6 @@ class SeedVRServer:
         except Exception as e:
             print(f"❌ Error during direct inference execution: {e}")
             import traceback
-            init = False
             traceback.print_exc()
             # Propaga o erro para a UI do Gradio, que o exibirá de forma amigável.
             raise