Dockerfile

# Dockerfile (VERSÃO FINAL COM WRAPPER SCRIPT)

# 1. COMEÇAR COM A BASE CORRETA
FROM nvidia/cuda:12.1.1-devel-ubuntu22.04

# 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/*

# 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

# 4. ADICIONAR CONDA AO PATH
ENV PATH /opt/conda/bin:$PATH

# 5. COPIAR TODOS OS ARQUIVOS DO REPOSITÓRIO PARA DENTRO DA IMAGEM
COPY . /app
WORKDIR /app

# 6. ACEITAR OS TERMOS DE SERVIÇO
RUN yes | conda tos accept

# 7. ATUALIZAR O CONDA
RUN conda update -n base -c defaults conda

# 8. CRIAR O AMBIENTE CONDA A PARTIR DO environment.yml
RUN conda env create -f environment.yml && conda clean --all -y

# 9. INSTALAR FLASH_ATTN DENTRO DO AMBIENTE
RUN conda run -n seedvr pip install "flash_attn==2.5.9.post1" --no-build-isolation

# 10. BAIXAR OS VÍDEOS DE EXEMPLO DENTRO DA PASTA CORRETA
WORKDIR /app/SeedVR
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

# 11. BAIXAR O MODELO (joguei pra perto do final pra aproveitar o cache se mudar o código)
RUN huggingface-cli download ByteDance-Seed/SeedVR2-3B --local-dir ckpts --local-dir-use-symlinks False

# 12. TORNAR O NOSSO SCRIPT DE INICIALIZAÇÃO EXECUTÁVEL
# Voltamos para o diretório raiz do app para encontrar o run.sh
WORKDIR /app
RUN chmod +x run.sh

# 13. DEFINIR O PONTO DE ENTRADA FINAL E À PROVA DE FALHAS
# O WORKDIR final deve ser onde o app.py está
WORKDIR /app/SeedVR
CMD ["/app/run.sh"]