Update app.py

app.py

@@ -1,90 +1,55 @@
-import gradio as gr
-import subprocess
-import os
-import shutil  # MUDANÇA 1: Importar a biblioteca 'shutil' para cópia de arquivos
+# Dockerfile (VERSÃO FINAL COM VÍDEOS DE EXEMPLO)
 
-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",
-    ]
+# 1. COMEÇAR COM A BASE CORRETA
+FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
 
-    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'
-        )
+# 2. INSTALAR DEPENDÊNCIAS DO SISTEMA
+ENV DEBIAN_FRONTEND=noninteractive
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    wget \
+    git \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
 
-        while True:
-            line = process.stdout.readline()
-            if not line:
-                break
-            log_output += line
-            print(line.strip())
-            yield None, log_output
+# 3. INSTALAR O MINICONDA
+RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh && \
+    /bin/bash ~/miniconda.sh -b -p /opt/conda && \
+    rm ~/miniconda.sh
 
-        process.wait()
+# 4. ADICIONAR CONDA AO PATH
+ENV PATH /opt/conda/bin:$PATH
 
-        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!"
+# 5. ACEITAR OS TERMOS DE SERVIÇO
+RUN yes | conda tos accept
 
-    except Exception as e:
-        error_message = f"{log_output}\n\nOcorreu um erro: {str(e)}"
-        return None, error_message
+# 6. ATUALIZAR O CONDA
+RUN conda update -n base -c defaults conda
 
-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.")
+# 7. CRIAR O AMBIENTE CONDA
+COPY environment.yml .
+RUN conda env create -f environment.yml && conda clean --all -y
 
-    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]
-    )
+# 8. INSTALAR FLASH_ATTN DENTRO DO AMBIENTE
+RUN conda run -n seedvr pip install "flash_attn==2.5.9.post1" --no-build-isolation
 
-demo.launch()
+# 9. DEFINIR O AMBIENTE CONDA COMO PADRÃO
+SHELL ["conda", "run", "-n", "seedvr", "/bin/bash", "-c"]
+
+# 10. PREPARAR O APLICATIVO
+WORKDIR /app
+RUN git clone https://github.com/bytedance-seed/SeedVR.git
+WORKDIR /app/SeedVR
+
+# 11. BAIXAR O MODELO DURANTE A CONSTRUÇÃO
+RUN huggingface-cli download ByteDance-Seed/SeedVR2-3B --local-dir ckpts --local-dir-use-symlinks False
+
+# 12. !!! NOVO !!! BAIXAR OS VÍDEOS DE EXEMPLO
+RUN wget -O 01.mp4 https://huggingface.co/datasets/Iceclear/SeedVR_VideoDemos/resolve/main/seedvr_videos_crf23/aigc1k/23_1_lq.mp4 && \
+    wget -O 02.mp4 https://huggingface.co/datasets/Iceclear/SeedVR_VideoDemos/resolve/main/seedvr_videos_crf23/aigc1k/28_1_lq.mp4 && \
+    wget -O 03.mp4 https://huggingface.co/datasets/Iceclear/SeedVR_VideoDemos/resolve/main/seedvr_videos_crf23/aigc1k/2_1_lq.mp4
+
+# 13. COPIAR O CÓDIGO DO NOSSO APP
+COPY app.py .
+
+# 14. DEFINIR O COMANDO DE EXECUÇÃO
+CMD ["python", "app.py"]