Update app_seedvr.py

app_seedvr.py

@@ -6,19 +6,16 @@ from pathlib import Path
 from typing import Optional
 import gradio as gr
 import cv2
+import multiprocessing as mp # <--- LINHA ADICIONADA AQUI
 
 # --- INTEGRAÇÃO COM A LÓGICA DO SERVIDOR ---
 try:
-    # 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}")
-    # A aplicação não pode rodar sem a lógica do servidor.
     raise
 
 # --- INICIALIZAÇÃO ---
-# Cria uma instância única e persistente do servidor.
-# A inicialização pesada (clonar repo, baixar modelos) acontece apenas uma vez, aqui.
 server = SeedVRServer()
 
 # --- FUNÇÕES AUXILIARES DA UI ---
@@ -43,8 +40,6 @@ def _extract_first_frame(video_path: str) -> Optional[str]:
         if not success:
             print(f"Erro: Não foi possível ler o primeiro frame de {video_path}")
             return None
-        
-        # Salva o frame no mesmo diretório do vídeo, com extensão .jpg
         image_path = Path(video_path).with_suffix(".jpg")
         cv2.imwrite(str(image_path), image)
         return str(image_path)
@@ -53,19 +48,13 @@ def _extract_first_frame(video_path: str) -> Optional[str]:
         return None
 
 def on_file_upload(file_obj):
-    """
-    Callback acionado quando o usuário faz o upload de um arquivo.
-    Limpa saídas antigas e ajusta o slider de `sp_size` com base no tipo de arquivo.
-    """
+    """Callback acionado quando o usuário faz o upload de um arquivo."""
     if file_obj is None:
-        # Limpa os resultados e o log se o arquivo for removido
         return gr.update(value=1), None, None, None, gr.update(value=None, visible=False)
     
     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=4, interactive=True), None, None, None, gr.update(value=None, visible=False)
     else:
-        # 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 ---
@@ -78,11 +67,8 @@ def run_inference_ui(
     progress=gr.Progress(track_tqdm=True)
 ):
     """
-    A função de callback principal do Gradio. Usa geradores (`yield`)
-    para permitir atualizações da UI em tempo real durante a tarefa de longa duração.
+    A função de callback principal do Gradio.
     """
-    # 1. Estado Inicial e Validação
-    # No início, desabilita o botão, limpa resultados anteriores e mostra a janela de log.
     yield (
         gr.update(interactive=False, value="Processing... 🚀"),
         gr.update(value=None, visible=False),
@@ -101,23 +87,18 @@ def run_inference_ui(
     was_input_video = _is_video(input_file_path)
 
     try:
-        # Define um callback que será chamado pelo backend para atualizar o progresso e o log
         def progress_callback_wrapper(step: float, desc: str):
             nonlocal last_log_message
-            # Só adiciona ao log se a mensagem for nova, para evitar poluição visual
             if desc != last_log_message:
                 log_buffer.append(f"{desc}\n")
                 last_log_message = desc
-            # Atualiza o objeto de progresso do Gradio
             progress(step, desc=desc)
 
-        # 2. Executa a Inferência
-        # 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
+            seed=42,
             res_h=int(resolution),
-            res_w=int(resolution), # Largura igual à altura
+            res_w=int(resolution),
             sp_size=int(sp_size),
             fps=float(fps) if fps and fps > 0 else None,
             progress=progress_callback_wrapper,
@@ -126,17 +107,15 @@ def run_inference_ui(
         progress(1.0, desc="Complete!")
         log_buffer.append("✅ Inference complete! Processing final output...\n")
         
-        # 3. Processa e Exibe os Resultados
         final_image, final_video = None, None
         if was_input_video:
             final_video = video_result_path
             log_buffer.append("✅ Video result is ready.\n")
-        else: # Se a entrada foi uma imagem
+        else:
             final_image = _extract_first_frame(video_result_path)
-            final_video = video_result_path # Também disponibiliza o vídeo de 1 frame
+            final_video = video_result_path
             log_buffer.append("✅ Image result extracted from video.\n")
         
-        # Yield final para mostrar os resultados e reabilitar o botão
         yield (
             gr.update(interactive=True, value="Restore Media"),
             gr.update(value=final_image, visible=final_image is not None),
@@ -152,7 +131,6 @@ def run_inference_ui(
         import traceback
         traceback.print_exc()
         
-        # Yield para estado de erro: reabilita o botão e mostra o log com o erro
         yield (
             gr.update(interactive=True, value="Restore Media"),
             None, None, None,
@@ -160,9 +138,8 @@ def run_inference_ui(
         )
 
 # --- LAYOUT DA INTERFACE GRÁFICA (GRADIO) ---
-
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="SeedVR Media Restoration") as demo:
-    # Cabeçalho
+    # ... (O layout da UI permanece exatamente o mesmo)
     gr.Markdown(
         """
         <div style='text-align: center; margin-bottom: 20px;'>
@@ -171,13 +148,10 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="SeedVR Media Res
         </div>
         """
     )
-
     with gr.Row():
-        # --- Coluna da Esquerda: Entradas e Controles ---
         with gr.Column(scale=1):
             gr.Markdown("### 1. Upload Media")
             input_media = gr.File(label="Input File (Video or Image)", type="filepath", interactive=True)
-            
             gr.Markdown("### 2. Configure Settings")
             with gr.Accordion("Generation Parameters", open=True):
                 resolution_select = gr.Dropdown(
@@ -186,34 +160,22 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="SeedVR Media Res
                     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=1, value=4,
                     info="For multi-GPU videos. Automatically set to 1 for images."
                 )
-
                 fps_out = gr.Number(label="Output FPS (for Videos)", value=24, precision=0, info="Set to 0 to use the original FPS.")
-
             run_button = gr.Button("Restore Media", variant="primary", icon="✨")
-
-        # --- Coluna da Direita: Resultados ---
         with gr.Column(scale=2):
             gr.Markdown("### 3. Results")
-            
-            # Janela de Log
             log_window = gr.Textbox(
-                label="Inference Log 📝",
-                lines=8, max_lines=15,
-                interactive=False, visible=False, autoscroll=True
+                label="Inference Log 📝", lines=8, max_lines=15,
+                interactive=False, visible=False, autoscroll=True,
             )
-
-            # Componentes de saída (começam invisíveis)
             output_image = gr.Image(label="Image Result", show_download_button=True, type="filepath", visible=False)
             output_video = gr.Video(label="Video Result", visible=False)
             output_download = gr.File(label="Download Full Result (Video)", visible=False)
-
-    # --- Rodapé ---
     gr.Markdown(
         """
         ---
@@ -222,16 +184,12 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="SeedVR Media Res
         """
     )
     
-    # --- Lógica de Eventos da UI ---
-    
-    # Ao fazer upload de um arquivo, ajusta o slider `sp_size` e limpa saídas antigas.
     input_media.upload(
         fn=on_file_upload,
         inputs=[input_media],
         outputs=[sp_size_slider, output_image, output_video, output_download, log_window]
     )
     
-    # Ao clicar no botão, executa a função de inferência principal.
     run_button.click(
         fn=run_inference_ui,
         inputs=[input_media, resolution_select, sp_size_slider, fps_out],
@@ -239,8 +197,9 @@ 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.
+    # Garante que o start_method do multiprocessing seja 'spawn', que é mais seguro
+    # e evita problemas de estado compartilhado entre processos.
+    # É uma boa prática definir isso no ponto de entrada principal da aplicação.
     mp.set_start_method('spawn', force=True)
     
     demo.launch(