Update Dockerfile

Dockerfile

@@ -1,88 +1,14 @@
-ARG MODEL_NAME
-ARG MODEL_PARAMS
-ARG MODEL_PROMPT_TEMPLATE
-ARG APP_COLOR
-ARG APP_NAME
-
-
-FROM node:20 as chatui-builder
-ARG MODEL_NAME
-ARG MODEL_PARAMS
-ARG APP_COLOR
-ARG APP_NAME
-ARG MODEL_PROMPT_TEMPLATE
-
-WORKDIR /app
-
-RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
-    git gettext && \
-    rm -rf /var/lib/apt/lists/*
-
-
-RUN git clone https://github.com/huggingface/chat-ui.git
-
-WORKDIR /app/chat-ui
-
-
-COPY .env.local.template .env.local.template
-
-RUN mkdir defaults
-ADD defaults /defaults
-RUN chmod -R 777 /defaults
-RUN --mount=type=secret,id=MONGODB_URL,mode=0444 \
-    MODEL_NAME="${MODEL_NAME:="$(cat /defaults/MODEL_NAME)"}" && export MODEL_NAME \
-    && MODEL_PARAMS="${MODEL_PARAMS:="$(cat /defaults/MODEL_PARAMS)"}" && export MODEL_PARAMS \
-    && MODEL_PROMPT_TEMPLATE="${MODEL_PROMPT_TEMPLATE:="$(cat /defaults/MODEL_PROMPT_TEMPLATE)"}" && export MODEL_PROMPT_TEMPLATE \
-    && APP_COLOR="${APP_COLOR:="$(cat /defaults/APP_COLOR)"}" && export APP_COLOR \
-    && APP_NAME="${APP_NAME:="$(cat /defaults/APP_NAME)"}" && export APP_NAME \
-    && MONGODB_URL=$(cat /run/secrets/MONGODB_URL > /dev/null | grep '^' || cat /defaults/MONGODB_URL) && export MONGODB_URL && \
-    echo "${MONGODB_URL}" && \
-    envsubst < ".env.local.template" > ".env.local" \ 
-    && rm .env.local.template
-
-
-
-RUN --mount=type=cache,target=/app/.npm \
-    npm set cache /app/.npm && \
-    npm ci
-
-RUN npm run build
+FROM ghcr.io/huggingface/chat-ui:latest AS base
 
-FROM ghcr.io/huggingface/text-generation-inference:latest
+FROM ghcr.io/huggingface/text-generation-inference:latest AS final
 
 ARG MODEL_NAME
-ARG MODEL_PARAMS
-ARG MODEL_PROMPT_TEMPLATE
-ARG APP_COLOR
-ARG APP_NAME
+ENV MODEL_NAME=${MODEL_NAME}
 
 ENV TZ=Europe/Paris \
     PORT=3000
 
-
-
-RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
-    gnupg \
-    curl \
-    gettext && \
-    rm -rf /var/lib/apt/lists/*
-COPY entrypoint.sh.template entrypoint.sh.template
-
-RUN mkdir defaults
-ADD defaults /defaults
-RUN chmod -R 777 /defaults
-
-RUN --mount=type=secret,id=MONGODB_URL,mode=0444 \
-    MODEL_NAME="${MODEL_NAME:="$(cat /defaults/MODEL_NAME)"}" && export MODEL_NAME \
-    && MODEL_PARAMS="${MODEL_PARAMS:="$(cat /defaults/MODEL_PARAMS)"}" && export MODEL_PARAMS \
-    && MODEL_PROMPT_TEMPLATE="${MODEL_PROMPT_TEMPLATE:="$(cat /defaults/MODEL_PROMPT_TEMPLATE)"}" && export MODEL_PROMPT_TEMPLATE \
-    && APP_COLOR="${APP_COLOR:="$(cat /defaults/APP_COLOR)"}" && export APP_COLOR \
-    && APP_NAME="${APP_NAME:="$(cat /defaults/APP_NAME)"}" && export APP_NAME \
-    && MONGODB_URL=$(cat /run/secrets/MONGODB_URL > /dev/null | grep '^' || cat /defaults/MONGODB_URL) && export MONGODB_URL &&  \
-    envsubst < "entrypoint.sh.template" > "entrypoint.sh" \
-    && rm entrypoint.sh.template
-
-
+# mongo installation
 RUN curl -fsSL https://www.mongodb.org/static/pgp/server-7.0.asc | \
     gpg -o /usr/share/keyrings/mongodb-server-7.0.gpg \
    --dearmor
@@ -93,19 +19,20 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-ins
     mongodb-org && \
     rm -rf /var/lib/apt/lists/*
 
-RUN mkdir -p /data/db
-RUN chown -R 1000:1000 /data
-
+# node installation
 RUN curl -fsSL https://deb.nodesource.com/setup_20.x | /bin/bash -
 
 RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
     nodejs && \
     rm -rf /var/lib/apt/lists/*
 
+# image setup
+RUN useradd -m -u 1000 user
+
 RUN mkdir /app
 RUN chown -R 1000:1000 /app
-
-RUN useradd -m -u 1000 user
+RUN mkdir /data
+RUN chown -R 1000:1000 /data
 
 # Switch to the "user" user
 USER user
@@ -114,13 +41,20 @@ ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
 
 RUN npm config set prefix /home/user/.local
-RUN npm install -g pm2
+RUN npm install -g dotenv-cli
+
+
+# copy chat-ui from base image
+COPY --from=base --chown=1000 /app/node_modules /app/node_modules
+COPY --from=base --chown=1000 /app/package.json /app/package.json
+COPY --from=base --chown=1000 /app/build /app/build
 
-COPY --from=chatui-builder --chown=1000 /app/chat-ui/node_modules /app/node_modules
-COPY --from=chatui-builder --chown=1000 /app/chat-ui/package.json /app/package.json
-COPY --from=chatui-builder --chown=1000 /app/chat-ui/build /app/build
+COPY --from=base --chown=1000 /app/.env /app/.env
+COPY --chown=1000 .env.local /app/.env.local
 
-ENTRYPOINT ["/bin/bash"]
-CMD ["entrypoint.sh"]
+COPY --chown=1000 entrypoint.sh /app/entrypoint.sh
 
+RUN chmod +x /app/entrypoint.sh
 
+# entrypoint
+ENTRYPOINT [ "/app/entrypoint.sh" ]