Update ollama.sh

ollama.sh

@@ -1,12 +1,30 @@
-#!/bin/sh
-source /app/venv/bin/activate
-# Start Ollama in the background
-# Starting server
-echo "Starting Ollama server"
-ollama serve &
-# Wait for the server to start
-sleep 5
-# Pull the model
-ollama pull all-minilm
-
-exec ollama serve
+#!/bin/bash
+
+# Function to start Ollama server
+start_server() {
+    echo "Starting Ollama server"
+    ollama serve &
+    SERVER_PID=$!
+}
+
+# Function to wait for server readiness
+wait_for_server() {
+    echo "Waiting for server to start..."
+    while ! nc -z localhost 11434; do
+        sleep 1
+    done
+}
+
+# Function to pull model
+pull_model() {
+    echo "Pulling all-minilm model"
+    ollama pull all-minilm
+}
+
+# Main execution
+start_server
+wait_for_server
+pull_model
+
+# Keep container running by waiting for server process
+wait $SERVER_PID