Update start.sh

start.sh

@@ -1,5 +1,24 @@
-#!/bin/sh
-nohup ollama serve > 2>&1 &
-sleep 10
-ollama pull tinyllama
-uvicorn app:app --host 0.0.0.0 --port 7860
+#!/bin/bash
+# Start Ollama in the background
+ollama serve &
+
+# Pull a small model (e.g., TinyLlama) if not present
+if ! ollama list | grep -q "tinyllama"; then
+    ollama pull tinyllama
+fi
+
+# Wait for Ollama to start (up to 30 seconds)
+max_attempts=30
+attempt=0
+while ! curl -s http://localhost:11434/api/tags >/dev/null; do
+    sleep 1
+    attempt=$((attempt + 1))
+    if [ $attempt -eq $max_attempts ]; then
+        echo "Ollama failed to start within 30 seconds. Exiting."
+        exit 1
+    fi
+done
+echo "Ollama is ready."
+
+# Start the FastAPI server
+uvicorn app:app --host 0.0.0.0 --port 7860