Add base image path argument (#6)

- Introduced BASE_IMAGE_PATH argument
- Updated frontend-builder stage
- Updated venv-builder stage
- Updated production stage

Co-authored-by: Haihui.Wang <[email protected]>

Author: HaiHui886

Dockerfile

@@ -5,6 +5,7 @@ ARG POETRY_VERSION=2.1.3
 ARG VERSION_OVERRIDE
 ARG BRANCH_OVERRIDE
 ARG BUILD_CN=false
+ARG BASE_IMAGE_PATH=
 
 ################################ Overview
 
@@ -17,7 +18,7 @@ ARG BUILD_CN=false
 # 5. "prod" - Creates the final production image with the Label Studio, Nginx, and other dependencies.
 
 ################################ Stage: frontend-builder (build frontend assets)
-FROM node:${NODE_VERSION} AS frontend-builder
+FROM ${BASE_IMAGE_PATH}node:${NODE_VERSION} AS frontend-builder
 
 ENV BUILD_NO_SERVER=true \
     BUILD_NO_HASH=true \
@@ -69,7 +70,7 @@ RUN --mount=type=cache,target=${YARN_CACHE_FOLDER},sharing=locked \
     yarn version:libs
 
 ################################ Stage: venv-builder (prepare the virtualenv)
-FROM python:${PYTHON_VERSION}-slim AS venv-builder
+FROM ${BASE_IMAGE_PATH}python:${PYTHON_VERSION}-slim AS venv-builder
 ARG POETRY_VERSION
 
 ENV PYTHONUNBUFFERED=1 \
@@ -141,7 +142,7 @@ ARG BRANCH_OVERRIDE
 RUN --mount=type=bind,source=.git,target=/label-studio/.git \
     VERSION_OVERRIDE=${VERSION_OVERRIDE} BRANCH_OVERRIDE=${BRANCH_OVERRIDE} poetry run python label_studio/core/version.py
 
-FROM python:${PYTHON_VERSION}-slim AS production
+FROM ${BASE_IMAGE_PATH}python:${PYTHON_VERSION}-slim AS production
 
 ENV LS_DIR=/label-studio \
     HOME=/label-studio \