# Dockerfile (VERSÃO FINAL COM OS DOIS PROJETOS)

FROM nvidia/cuda:12.1.1-devel-ubuntu22.04

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

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

ENV PATH /opt/conda/bin:$PATH

COPY . /app
WORKDIR /app

RUN yes | conda tos accept
RUN conda update -n base -c defaults conda
RUN conda env create -f environment.yml && conda clean --all -y

# Instala flash_attn para ambos os projetos
# V-INT usa uma versão mais nova, então instalamos essa.
RUN conda run -n seedvr env FLASH_ATTENTION_SKIP_CUDA_BUILD=TRUE \
    pip install "flash_attn==2.6.3" --no-build-isolation --no-cache-dir

# Clona os repositórios dos projetos
RUN git clone https://github.com/ByteDance-Seed/SeedVR.git
RUN git clone https://github.com/ByteDance-Seed/VINCIE.git

WORKDIR /app
CMD ["conda", "run", "-n", "seedvr", "env", "PYTHONUNBUFFERED=1", "python", "app.py", "--share"]