Update Dockerfile

Dockerfile

@@ -1,58 +1,37 @@
-# 1. 指定包含 Python 3.10 的基础镜像
-FROM python:3.10-slim
+# ============================================================
+# pyVideoTrans WebUI Dockerfile
+#
+# CPU:  docker build -t pyvideotrans-webui .
+# GPU:  docker build --build-arg USE_CUDA=true -t pyvideotrans-webui:gpu .
+# ============================================================
+
+ARG USE_CUDA=false
+
+FROM python:3.10-slim AS cpu-base
+FROM nvidia/cuda:12.8.0-cudnn-runtime-ubuntu22.04 AS gpu-base
+
+FROM  AS final-base
 
-# 2. [新增] 从官方镜像中直接复制最新版的 uv 到当前容器 (最快、最稳定的安装方式)
 COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/
 
-# 3. 设置 Gradio 需要的默认环境变量
 ENV GRADIO_SERVER_NAME="0.0.0.0"
 ENV GRADIO_SERVER_PORT=7860
 ENV DEBIAN_FRONTEND=noninteractive
+ENV PYTHONUNBUFFERED=1
 ENV FONTCONFIG_PATH=/etc/fonts
 
-# 4. 设置工作目录
 WORKDIR /app
 
 
 
-# 5. 安装下载工具、FFmpeg 7 以及必要的系统动态库(GUI和音频库)
-RUN apt-get update && \
-    apt-get install -y \
-    fontconfig \
-    fonts-noto-cjk  \
-    fonts-liberation \
-    fonts-dejavu \
-        wget \
-        xz-utils \
-        libglib2.0-0 \
-        libgl1 \
-        libsm6 \
-        libxext6 \
-        libxrender-dev \
-        libxkbcommon-x11-0 \
-        libdbus-1-3 \
-		rubberband-cli \
-		libsndfile1-dev \
-        libsndfile1 && \
-    wget https://johnvansickle.com/ffmpeg/releases/ffmpeg-release-amd64-static.tar.xz && \
-    tar -Jxf ffmpeg-release-amd64-static.tar.xz && \
-    cp ffmpeg-*-static/ffmpeg /usr/local/bin/ && \
-    cp ffmpeg-*-static/ffprobe /usr/local/bin/ && \
-    rm -rf ffmpeg-* && \
-    apt-get clean && rm -rf /var/lib/apt/lists/*
-# 6. [新增] 优先复制配置文件 (这利用了 Docker 缓存，以后改代码不改依赖时能秒构建)
-COPY pyproject.toml ./
-# 如果你的项目根目录下生成了 uv.lock 文件，请将下面这行的注释取消掉：
-# COPY uv.lock ./
-
-# 7. [修改] 使用 uv 读取 pyproject.toml 并安装依赖到系统环境中
-RUN uv pip install --system -r pyproject.toml --all-extras &&  uv pip install --system torch==2.7.1 torchaudio==2.7.1  --index-url https://download.pytorch.org/whl/cu128 &&  uv pip install --system nvidia-cublas-cu12 nvidia-cudnn-cu12
-
-# 8. 复制你项目里的所有其他代码文件
-COPY . .
-
-# 9. 暴露端口
+RUN apt-get update && apt-get install -y --no-install-recommends fontconfig fonts-noto-cjk  fonts-liberation fonts-dejavu wget xz-utils git libglib2.0-0 libgl1 libsm6 libxext6 libxrender-dev libxkbcommon-x11-0 libdbus-1-3 libsndfile1 python3-dev rubberband-cli libsndfile1-dev && wget -q https://johnvansickle.com/ffmpeg/releases/ffmpeg-release-amd64-static.tar.xz && tar -Jxf ffmpeg-release-amd64-static.tar.xz && cp ffmpeg-*-static/ffmpeg /usr/local/bin/ && cp ffmpeg-*-static/ffprobe /usr/local/bin/ && rm -rf ffmpeg-* && apt-get clean && rm -rf /var/lib/apt/lists/*
+
+RUN git clone -b dev https://github.com/jianchang512/pyvideotrans.git .
+
+RUN if [ "" = "true" ]; then echo ">>> CUDA" && uv pip install --system -r pyproject.toml --all-extras && uv pip install --system torch==2.7.1 torchaudio==2.7.1 --index-url https://download.pytorch.org/whl/cu128 && uv pip install --system nvidia-cublas-cu12 nvidia-cudnn-cu12; else echo ">>> CPU" && uv pip install --system -r pyproject.toml --all-extras; fi
+
+RUN rm -rf /root/.cache/uv /tmp/*
+
 EXPOSE 7860
 
-# 10. 启动你的主程序
 CMD ["python", "webui.py"]