# 使用 PyTorch 官方镜像，支持 GPU（可用于 CPU）
FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime

# 设置工作目录
WORKDIR /workspace

# 安装必要的工具
RUN apt-get update && apt-get install -y \
    git \
    ffmpeg \
    libgl1-mesa-glx \
    libglib2.0-0 \
    && rm -rf /var/lib/apt/lists/*

# 克隆 FaceFusion 项目
RUN git clone https://github.com/facefusion/facefusion.git /workspace/facefusion

# 进入项目目录
WORKDIR /workspace/facefusion

# 安装 Python 依赖
RUN pip install --no-cache-dir -r requirements.txt

# 复制应用启动文件
COPY app.py /workspace/facefusion/app.py

# 运行应用
CMD ["python", "app.py"]