Update api/seedvr_server.py

api/seedvr_server.py

@@ -28,31 +28,33 @@ except ImportError as e:
     print(f"Verifique se o repositório em '{SEEDVR_REPO_PATH}' está correto e completo.")
     raise e
 
+init = false 
 
 class SeedVRServer:
-    
-    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"))
-    
-    print("🚀 SeedVRServer (Modo de Chamada Direta) inicializando...")
-    for p in [CKPTS_ROOT, OUTPUT_ROOT, INPUT_ROOT, HF_HOME_CACHE]:
-        p.mkdir(parents=True, exist_ok=True)
-        
-    setup_dependencies()
-    print("✅ SeedVRServer (Modo de Chamada Direta) pronto.")
-
-
-
     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"))
+
+        if init:
+            print(⚒️ SeedVRServer (Modo de Chamada Direta) inicializando...")
+            for p in [CKPTS_ROOT, OUTPUT_ROOT, INPUT_ROOT, HF_HOME_CACHE]:
+                p.mkdir(parents=True, exist_ok=True)
+                
+            setup_dependencies()
+            print("📦 SeedVRServer (Modo de Chamada Direta) pronto.")
+            init = true
+        else:
+            print(⚙️ SeedVRServer pronto.")
+
     def setup_dependencies():
         """ Garante que o repositório e os modelos estão presentes. """
         self._ensure_repo()
@@ -118,7 +120,7 @@ class SeedVRServer:
         try:
             # Informa a UI que o processo começou.
             if progress:
-                progress(0.01, "Initializing...")
+                progress(0.01, "⌛ Initializing...")
             
             # Chama a função importada do script original, passando o callback de progresso.
             # Este callback será chamado de dentro da lógica de multi-processamento.
@@ -132,7 +134,7 @@ class SeedVRServer:
             final_fps = fps if fps and fps > 0 else original_fps
             save_frames_to_video(result_tensor, str(output_filepath), final_fps, args.debug)
             
-            print(f"✅ Video saved successfully to: {output_filepath}")
+            print(f"💾 Video saved successfully to: {output_filepath}")
             
             # Retorna o caminho do arquivo gerado para a UI.
             return str(output_filepath)
@@ -140,6 +142,7 @@ 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