Update Dockerfile

Dockerfile

@@ -1,45 +1,59 @@
 # Base stage for chat-ui
 FROM ghcr.io/huggingface/chat-ui:latest AS chat_ui_base
 
-# Add text-generation-inference as a separate stage
+# 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
 
-ARG MODEL_NAME
-ENV MODEL_NAME=${MODEL_NAME}
-ENV TZ=Europe/Paris \
-    PORT=3000
+# 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
 
-# Install MongoDB, netcat, and other dependencies
-RUN apt-get update && apt-get install -y gnupg curl lsb-release ca-certificates netcat && \
+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 curl && \
-    rm -rf /var/lib/apt/lists/*
+    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
+# 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 "user" user
+# Switch to the non-root "user" user
 USER user
 
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
 
-# Copy files from chat-ui stage
+# 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
 COPY --chown=1000 .env.local /app/.env.local
 
-# Copy entrypoint script
+# 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
+
 # Use the entrypoint script
-ENTRYPOINT [ "/app/entrypoint.sh" ]
+ENTRYPOINT [ "/app/entrypoint.sh" ]