Update entrypoint.sh

entrypoint.sh

@@ -1,20 +1,34 @@
 #!/bin/bash
 
-# Make sure `/data/db` directory exists even with persistent storage
-mkdir -p /data/db
-# If app crashed, mongo didn't stop gracefully. Remove all the old *.lock files
-find /data/db -name "*.lock" -type f -exec rm -f {} \; 
-# Start the local Mongo database
-mongod &
+# Start MongoDB in the background
+echo "Starting MongoDB..."
+mongod --dbpath /data/db &
 
-# Start the text-generation-inference process
-text-generation-launcher --model-id ${MODEL_NAME} --num-shard 1 --port 8080 --trust-remote-code &
+# Function to check if a port is open
+check_port() {
+    timeout 3 bash -c "</dev/tcp/127.0.0.1/$1" 2>/dev/null
+}
+
+# Wait for MongoDB to start
+until check_port 27017; do
+  echo "Waiting for MongoDB to start on port 27017..."
+  sleep 5
+done
+
+# Start text-generation-inference server in the background
+echo "Starting text-generation-inference server..."
+text-generation-launcher --model-id ${MODEL_NAME} --num-shard 1 --port 8080 --disable-custom-kernels &
 
 # Wait for text-generation-inference to start
-curl --retry 60 --retry-delay 10 --retry-connrefused http://127.0.0.1:8080/health
+until check_port 8080; do
+  echo "Waiting for text-generation-inference to start on port 8080..."
+  sleep 5
+done
 
 # Start the chat-ui process
-dotenv -e /app/.env -c -- node /app/build/index.js -- --host 0.0.0.0 --port 3000
+echo "Starting Chat UI..."
+dotenv -e /app/.env -c -- node /app/build/index.js -- --host 0.0.0.0 --port ${PORT}
+
 # Wait for any process to exit
 wait -n