app.py

import gradio as gr
import subprocess
import os
import shutil  # MUDANÇA 1: Importar a biblioteca 'shutil' para cópia de arquivos

def run_inference_app(video_file_path, seed_num):
    # O componente gr.Video passa o caminho do arquivo como uma string
    if video_file_path is None:
        return None, "Por favor, envie um arquivo de vídeo."
    
    # O app roda de dentro do diretório /app/SeedVR
    input_folder = "inputs"
    os.makedirs(input_folder, exist_ok=True)
    
    # MUDANÇA 2: Lógica de cópia de arquivo mais robusta
    # Usamos shutil.copy para copiar o arquivo temporário do Gradio para nossa pasta de entrada.
    # os.path.basename garante que pegamos apenas o nome do arquivo original.
    file_name = os.path.basename(video_file_path)
    input_video_path = os.path.join(input_folder, file_name)
    shutil.copy(video_file_path, input_video_path)
        
    output_folder = "outputs"
    os.makedirs(output_folder, exist_ok=True)
    
    # Comando de inferência. O ambiente conda já está "ativado".
    command = [
        "torchrun", "--nproc-per-node=4",
        "projects/inference_seedvr2_3b.py",
        "--video_path", input_folder,
        "--output_dir", output_folder,
        "--seed", str(seed_num),
        "--res_h", "320",
        "--res_w", "512",
    ]

    log_output = "Iniciando a inferência...\n" + ' '.join(command) + "\n\n"
    
    try:
        process = subprocess.Popen(
            command,
            stdout=subprocess.PIPE,
            stderr=subprocess.STDOUT,
            text=True,
            encoding='utf-8'
        )

        while True:
            line = process.stdout.readline()
            if not line:
                break
            log_output += line
            print(line.strip())
            yield None, log_output

        process.wait()

        if process.returncode != 0:
            raise RuntimeError("O script de inferência falhou. Verifique os logs.")
        
        result_files = [os.path.join(output_folder, f) for f in os.listdir(output_folder) if f.endswith('.mp4')]
        if not result_files:
            return None, log_output + "\n\nERRO: Nenhum vídeo foi gerado."
            
        return result_files, log_output + "\n\nInferência concluída com sucesso!"

    except Exception as e:
        error_message = f"{log_output}\n\nOcorreu um erro: {str(e)}"
        return None, error_message

with gr.Blocks() as demo:
    gr.Markdown("# 🚀 Inferência SeedVR2 com Ambiente Conda")
    gr.Markdown("Este ambiente foi construído com Conda usando um Dockerfile para máxima estabilidade.")

    with gr.Row():
        with gr.Column(scale=1):
            # MUDANÇA 3: Usar gr.Video em vez de gr.File
            video_input = gr.Video(label="Vídeo de Entrada")
            seed_input = gr.Number(label="Seed", value=123)
            run_button = gr.Button("Gerar Vídeo", variant="primary")
        with gr.Column(scale=2):
            gallery_output = gr.Gallery(label="Vídeo de Saída", show_label=True)
            log_display = gr.Textbox(label="Logs de Execução", lines=15, interactive=False, autoscroll=True)
            
    run_button.click(
        fn=run_inference_app,
        inputs=[video_input, seed_input],
        outputs=[gallery_output, log_display]
    )

demo.launch()