Update Dockerfile

Dockerfile

@@ -1,59 +1,60 @@
-# Base stage for chat-ui
-FROM ghcr.io/huggingface/chat-ui:latest AS chat_ui_base
-
-# Add text-generation-inference as a separate stage (Optional, if needed)
-FROM ghcr.io/huggingface/text-generation-inference:latest AS text_gen_inference
-
-# Final stage
-FROM ubuntu:20.04 AS final
-
-# Set default values for build arguments (can be overridden during build)
-ARG MODEL_NAME=TheBloke/Mistral-7B-OpenOrca-GPTQ
-ARG TZ=Europe/Paris
-ARG PORT=3000
-
-ENV MODEL_NAME=${MODEL_NAME} \
-    TZ=${TZ} \
-    PORT=${PORT}
-
-# Install necessary packages, including python3-dotenv
-RUN apt-get update && apt-get install -y gnupg curl lsb-release ca-certificates netcat python3-pip && \
-    curl -fsSL https://www.mongodb.org/static/pgp/server-7.0.asc | gpg --dearmor -o /usr/share/keyrings/mongodb-archive-keyring.gpg && \
-    echo "deb [signed-by=/usr/share/keyrings/mongodb-archive-keyring.gpg] https://repo.mongodb.org/apt/ubuntu $(lsb_release -sc)/mongodb-org/7.0 multiverse" | tee /etc/apt/sources.list.d/mongodb-org-7.0.list && \
-    apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends mongodb-org nodejs && \
-    rm -rf /var/lib/apt/lists/* && \
-    python3 -m pip install python-dotenv # Install python-dotenv
-
-# Create user and directories with proper permissions
-RUN useradd -m -u 1000 user && \
-    mkdir /app && chown -R 1000:1000 /app && \
-    mkdir /data && chown -R 1000:1000 /data
-
-# Switch to the non-root "user" user
+FROM ghcr.io/huggingface/chat-ui:latest AS base
+
+FROM ghcr.io/huggingface/text-generation-inference:latest AS final
+
+ARG MODEL_NAME
+ENV MODEL_NAME=${MODEL_NAME}
+
+ENV TZ=Europe/Paris \
+    PORT=3000
+
+# 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-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/*
+
+# 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 mkdir /data
+RUN chown -R 1000:1000 /data
+
+# Switch to the "user" user
 USER user
 
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
 
-# Copy files from chat-ui stage with correct ownership
-COPY --from=chat_ui_base --chown=1000 /app/node_modules /app/node_modules
-COPY --from=chat_ui_base --chown=1000 /app/package.json /app/package.json
-COPY --from=chat_ui_base --chown=1000 /app/build /app/build
-# Assuming .env contains default environment variables for chat-ui
-COPY --from=chat_ui_base --chown=1000 /app/.env /app/.env
-# Copy .env.local from your project root, if it exists
+RUN npm config set prefix /home/user/.local
+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=base --chown=1000 /app/.env /app/.env
 COPY --chown=1000 .env.local /app/.env.local
 
-# Copy entrypoint script and make it executable
 COPY --chown=1000 entrypoint.sh /app/entrypoint.sh
-RUN chmod +x /app/entrypoint.sh
 
-# Set working directory (optional, but good practice)
-WORKDIR /app
-
-# Update system configuration for MongoDB (vm.max_map_count)
-# This needs to be done at runtime, not during the build, so it's moved to entrypoint.sh
-# RUN echo "vm.max_map_count=1677720" >> /etc/sysctl.conf # Not recommended here
+RUN chmod +x /app/entrypoint.sh
 
-# Use the entrypoint script
+# entrypoint
 ENTRYPOINT [ "/app/entrypoint.sh" ]