add build image script

Author: yihuiwen

dockerfiles/Dockerfile_deploy

@@ -1,10 +1,11 @@
-FROM lightx2v/lightx2v:25111101-cu128 AS base
+ARG BASE_TAG=26011201-cu128
+FROM lightx2v/lightx2v:${BASE_TAG} AS base
 
 RUN mkdir /workspace/LightX2V
 WORKDIR /workspace/LightX2V
 ENV PYTHONPATH=/workspace/LightX2V
 
-RUN pip install diffusers==0.37.1
+RUN pip install diffusers==0.37.1 librosa
 
 COPY tools tools
 COPY assets assets

dockerfiles/build_deploy.sh

@@ -0,0 +1,17 @@
+#!/bin/bash
+set -euo pipefail
+
+SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
+REPO_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
+DOCKERFILE="$SCRIPT_DIR/Dockerfile_deploy"
+
+AUTO_BASE_TAG=$(grep -oP 'ARG BASE_TAG=\K\S+' "$DOCKERFILE" | head -1)
+BASE_TAG="${1:-$AUTO_BASE_TAG}"
+REGISTRY="${2:-registry.cn-sh-01.sensecore.cn/devsft-ccr-2/lightx2v}"
+DATE_TAG=$(date +%y%m%d)
+GIT_HASH=$(git -C "$REPO_ROOT" rev-parse --short HEAD)
+IMAGE_TAG="${REGISTRY}:server-${BASE_TAG}-${DATE_TAG}-${GIT_HASH}"
+
+echo "Base image tag: $BASE_TAG"
+echo "Building image:  $IMAGE_TAG"
+docker buildx build --platform linux/amd64 -f "$DOCKERFILE" -t "$IMAGE_TAG" --build-arg BASE_TAG="$BASE_TAG" "$REPO_ROOT"