Update Dockerfile

Dockerfile

@@ -1,113 +1,43 @@
-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 rm entrypoint.sh
-
-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
-
+FROM ghcr.io/huggingface/chat-ui:latest AS base
 
-
-RUN --mount=type=cache,target=/app/.npm \
-    npm set cache /app/.npm && \
-    npm ci
-
-RUN npm run build
-
-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 \
-    wget \ 
-    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
-
-RUN wget http://archive.ubuntu.com/ubuntu/pool/main/o/openssl/libssl1.1_1.1.1f-1ubuntu2_amd64.deb
-RUN dpkg -i libssl1.1_1.1.1f-1ubuntu2_amd64.deb
 
-RUN curl -fsSL https://pgp.mongodb.com/server-6.0.asc | \
-    gpg -o /usr/share/keyrings/mongodb-server-6.0.gpg \
-    --dearmor
+# 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
 
-RUN echo "deb [ arch=amd64,arm64 signed-by=/usr/share/keyrings/mongodb-server-6.0.gpg ] https://repo.mongodb.org/apt/ubuntu focal/mongodb-org/6.0 multiverse" | tee /etc/apt/sources.list.d/mongodb-org-6.0.list
+RUN echo "deb [ arch=amd64,arm64 signed-by=/usr/share/keyrings/mongodb-server-7.0.gpg ] https://repo.mongodb.org/apt/ubuntu jammy/mongodb-org/7.0 multiverse" | tee /etc/apt/sources.list.d/mongodb-org-7.0.list
 
 RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
     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 mkdir /app
 RUN chown -R 1000:1000 /app
 
 RUN useradd -m -u 1000 user
+
+RUN chown -R 1000:1000 /app
+
+RUN mkdir /data
+RUN chown -R 1000:1000 /data
+
 # Switch to the "user" user
 USER user
 
@@ -117,11 +47,14 @@ ENV HOME=/home/user \
 RUN npm config set prefix /home/user/.local
 RUN npm install -g pm2
 
-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
 
-ENTRYPOINT ["/bin/bash"]
-CMD ["entrypoint.sh"]
+# 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 --chown=1000 entrypoint.sh /app/entrypoint.sh
 
+RUN chmod +x /app/entrypoint.sh
 
+# entrypoint
+ENTRYPOINT [ "/app/entrypoint.sh" ]