Update app_seedvr.py

app_seedvr.py

@@ -9,7 +9,7 @@ import cv2
 
 # --- INTEGRAÇÃO COM A LÓGICA DO SERVIDOR ---
 try:
-    # Importa a classe SeedVRServer que agora atua como nossa biblioteca de inferência.
+    # Importa a classe SeedVRServer que agora contém toda a lógica de inferência.
     from api.seedvr_server import SeedVRServer
 except ImportError as e:
     print(f"ERRO FATAL: Não foi possível importar o SeedVRServer. Detalhes: {e}")
@@ -18,10 +18,10 @@ except ImportError as e:
 
 # --- INICIALIZAÇÃO ---
 # Cria uma instância única e persistente do servidor.
-# A inicialização (clonar repo, baixar modelos) acontece apenas uma vez, no início.
+# A inicialização pesada (clonar repo, baixar modelos) acontece apenas uma vez, aqui.
 server = SeedVRServer()
 
-# --- FUNÇÕES AUXILIARES ---
+# --- FUNÇÕES AUXILIARES DA UI ---
 
 def _is_video(path: str) -> bool:
     """Verifica se um caminho de arquivo corresponde a um tipo de vídeo."""
@@ -55,7 +55,7 @@ def _extract_first_frame(video_path: str) -> Optional[str]:
 def on_file_upload(file_obj):
     """
     Callback acionado quando o usuário faz o upload de um arquivo.
-    Verifica se o arquivo é um vídeo e sugere um `sp_size` apropriado.
+    Limpa saídas antigas e ajusta o slider de `sp_size` com base no tipo de arquivo.
     """
     if file_obj is None:
         # Limpa os resultados e o log se o arquivo for removido
@@ -63,9 +63,9 @@ def on_file_upload(file_obj):
     
     if _is_video(file_obj.name):
         # Para vídeos, sugere um valor padrão para multi-GPU e torna o slider interativo
-        return gr.update(value=8, interactive=True), None, None, None, gr.update(value=None, visible=False)
+        return gr.update(value=4, interactive=True), None, None, None, gr.update(value=None, visible=False)
     else:
-        # Para imagens, trava o valor em 1
+        # Para imagens, trava o valor em 1, pois não há paralelismo de dados
         return gr.update(value=1, interactive=False), None, None, None, gr.update(value=None, visible=False)
 
 # --- FUNÇÃO PRINCIPAL DE INFERÊNCIA DA UI ---
@@ -93,7 +93,6 @@ def run_inference_ui(
 
     if not input_file_path:
         gr.Warning("Please upload a media file first.")
-        # Reabilita o botão e esconde os componentes de saída
         yield (gr.update(interactive=True, value="Restore Media"), None, None, None, gr.update(visible=False))
         return
 
@@ -104,7 +103,6 @@ def run_inference_ui(
     try:
         # Define um callback que será chamado pelo backend para atualizar o progresso e o log
         def progress_callback_wrapper(step: float, desc: str):
-            """ Wrapper para formatar logs e atualizar o progresso. """
             nonlocal last_log_message
             # Só adiciona ao log se a mensagem for nova, para evitar poluição visual
             if desc != last_log_message:
@@ -114,15 +112,15 @@ def run_inference_ui(
             progress(step, desc=desc)
 
         # 2. Executa a Inferência
-        # Chama o método direto do servidor, passando o nosso callback.
-        video_result_path = server.run_inference_direct(
+        # Chama o método do servidor, passando o nosso callback.
+        video_result_path = server.run_inference(
             file_path=input_file_path,
             seed=42, # Semente fixa conforme solicitado
             res_h=int(resolution),
             res_w=int(resolution), # Largura igual à altura
             sp_size=int(sp_size),
             fps=float(fps) if fps and fps > 0 else None,
-            progress=progress_callback_wrapper, # Passa nossa função de callback
+            progress=progress_callback_wrapper,
         )
         
         progress(1.0, desc="Complete!")
@@ -184,14 +182,14 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="SeedVR Media Res
             with gr.Accordion("Generation Parameters", open=True):
                 resolution_select = gr.Dropdown(
                     label="Resolution",
-                    choices=["480", "560", "720", "960", "1024", "2048"],
+                    choices=["480", "560", "720", "960", "1024"],
                     value="480",
                     info="Sets the output height and width to this value."
                 )
                 
                 sp_size_slider = gr.Slider(
                     label="Frames per Batch (sp_size)",
-                    minimum=1, maximum=16, step=4, value=8,
+                    minimum=1, maximum=16, step=1, value=4,
                     info="For multi-GPU videos. Automatically set to 1 for images."
                 )
 
@@ -241,6 +239,10 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="SeedVR Media Res
     )
 
 if __name__ == "__main__":
+    # Garante que o start_method do multiprocessing seja 'spawn'
+    # É uma boa prática definir isso no ponto de entrada principal.
+    mp.set_start_method('spawn', force=True)
+    
     demo.launch(
         server_name=os.getenv("GRADIO_SERVER_NAME", "0.0.0.0"),
         server_port=int(os.getenv("GRADIO_SERVER_PORT", "7860")),