Update entrypoint.sh

entrypoint.sh

@@ -1,19 +1,52 @@
 #!/bin/bash
 
+# Function to handle shutdown gracefully
+cleanup() {
+    echo "Shutting down Ollama server..."
+    pkill -f "ollama serve"
+    exit 0
+}
+
+
+# Set up signal handlers for graceful shutdown
+trap cleanup SIGTERM SIGINT
+
 # Start Ollama server in the background
+echo "Starting Ollama server..."
 ollama serve &
+OLLAMA_PID=$!
+
+# Wait for the server to be ready with timeout
+echo "Waiting for Ollama server to start..."
+TIMEOUT=60
+COUNTER=0
 
-# Wait for the server to be ready
 while ! nc -z localhost 7860; do
-    echo "Waiting for Ollama server to start..."
+    if [ $COUNTER -ge $TIMEOUT ]; then
+        echo "ERROR: Ollama server failed to start within $TIMEOUT seconds"
+        exit 1
+    fi
+    echo "Waiting for Ollama server to start... ($COUNTER/$TIMEOUT)"
     sleep 1
+    ((COUNTER++))
 done
 
-# Pull the model
+echo "Ollama server is ready!"
+
+# Pull the model with error handling
 echo "Pulling the model..."
-ollama pull gemma3:27b-it-qat
-# ollama pull smollm:1.7b-instruct-v0.2-q6_K
-# hf.co/unsloth/phi-4-GGUF:Q8_0
+if ! ollama pull gemma3:270m; then
+    echo "ERROR: Failed to pull model smollm2:135m"
+    exit 1
+fi
+
+if ! ollama pull nomic-embed-text; then
+    echo "ERROR: Failed to pull model gemma3:1b"
+    exit 1
+fi
+
+echo "Model pulled successfully!"
 
-# Keep the container running
-wait
+# Keep the container running and wait for the ollama process
+echo "Container is ready. Ollama server is running on port 7860."
+wait $OLLAMA_PID