Update entrypoint.sh

entrypoint.sh

@@ -1,22 +1,31 @@
 #!/bin/bash
+set -euo pipefail
 
-# 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 Mongo ---
+mkdir -p "$HOME/data/db" "$HOME/logs"
+mongod --dbpath "$HOME/data/db" \
+       --bind_ip 127.0.0.1 \
+       --logpath "$HOME/logs/mongod.log" \
+       --fork
 
-# Start the text-generation-inference process
-text-generation-launcher --model-id ${MODEL_NAME} --num-shard 1 --port 8080 --trust-remote-code &
+# --- Start the Python proxy with dotenv (DO NOT source .env in bash) ---
+# Loads /app/.env first, then overrides with /app/.env.local
+npx dotenv -e /app/.env -e /app/.env.local -c -- \
+  sh -lc 'python3 -m uvicorn proxy:app --host 127.0.0.1 --port 8000' &
 
-# Wait for text-generation-inference to start
-curl --retry 60 --retry-delay 10 --retry-connrefused http://127.0.0.1:8080/health
+# --- Wait for proxy health ---
+for i in {1..40}; do
+  if curl -fsS http://127.0.0.1:8000/healthz >/dev/null; then
+    echo "[entrypoint] Proxy is up"
+    break
+  fi
+  echo "[entrypoint] Waiting for proxy..."
+  sleep 0.5
+done
 
-# Start the chat-ui process
-dotenv -e /app/.env -c -- node /app/build/index.js -- --host 0.0.0.0 --port 3000
-# Wait for any process to exit
-wait -n
+# --- Start Chat-UI under dotenv as well ---
+npx dotenv -e /app/.env -e /app/.env.local -c -- \
+  node /app/build/index.js -- --host 0.0.0.0 --port 3000
 
-# Exit with status of process that exited first
+wait -n
 exit $?