Update Dockerfile

Dockerfile

@@ -1,33 +1,28 @@
-FROM ghcr.io/huggingface/chat-ui:latest AS base
+# Base stage for chat-ui
+FROM ghcr.io/huggingface/chat-ui:latest AS chat_ui_base
+
+# Add text-generation-inference as a separate stage
+FROM ghcr.io/huggingface/text-generation-inference:latest AS text_gen_inference
+
+# Final stage
+FROM ubuntu:20.04 AS final
 
-# Base configurations
 ARG MODEL_NAME
 ENV MODEL_NAME=${MODEL_NAME}
 ENV TZ=Europe/Paris \
     PORT=3000
 
-# Install MongoDB
-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
-
+# Update and install dependencies
 RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
-    mongodb-org && \
+    mongodb-org \
+    nodejs \
+    curl && \
     rm -rf /var/lib/apt/lists/*
 
-# Install Node.js
-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 && chown -R 1000:1000 /app
-RUN mkdir /data && chown -R 1000:1000 /data
+# Create user and directories
+RUN useradd -m -u 1000 user && \
+    mkdir /app && chown -R 1000:1000 /app && \
+    mkdir /data && chown -R 1000:1000 /data
 
 # Switch to the "user" user
 USER user
@@ -35,19 +30,16 @@ USER user
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
 
-RUN npm config set prefix /home/user/.local
-RUN npm install -g dotenv-cli
-
-# Copy chat-ui dependencies
-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 files from chat-ui stage
+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
+COPY --from=chat_ui_base --chown=1000 /app/.env /app/.env
 COPY --chown=1000 .env.local /app/.env.local
 
-# Adjust entrypoint for HF Space
+# Copy entrypoint script
 COPY --chown=1000 entrypoint.sh /app/entrypoint.sh
 RUN chmod +x /app/entrypoint.sh
 
+# Use the entrypoint script
 ENTRYPOINT [ "/app/entrypoint.sh" ]