Update startup.sh

startup.sh

@@ -1,64 +1,43 @@
 #!/bin/bash
 
-# Exit on any error
-set -e
+echo "=== Starting Ollama Generate API ==="
 
-echo "=== Starting Ollama API Server ==="
+# Set HOME to /tmp to avoid permission issues with /.ollama
+export HOME=/tmp
 
-# Set Ollama environment variables
+# Use /tmp to avoid any permission issues
 export OLLAMA_HOST=0.0.0.0:11434
-export OLLAMA_MODELS=/app/.ollama/models
-export OLLAMA_HOME=/app/.ollama
+export OLLAMA_MODELS=/tmp/ollama/models
+export OLLAMA_HOME=/tmp/ollama
 export OLLAMA_ORIGINS="*"
 
-# Create models directory with proper permissions
-mkdir -p /app/.ollama/models 2>/dev/null || {
-    echo "Warning: Could not create /app/.ollama/models, trying alternative location..."
-    export OLLAMA_MODELS=/tmp/.ollama/models
-    export OLLAMA_HOME=/tmp/.ollama
-    mkdir -p /tmp/.ollama/models
-    echo "Using temporary directory: $OLLAMA_MODELS"
-}
+# Create directories
+mkdir -p /tmp/ollama/models
+mkdir -p /tmp/.ollama
+echo "✅ Created Ollama directories in /tmp"
 
-# Start Ollama server in the background
+# Start Ollama server
 echo "Starting Ollama server..."
 ollama serve &
 OLLAMA_PID=$!
-echo "Ollama server started with PID: $OLLAMA_PID"
 
-# Function to check if Ollama is ready
-check_ollama() {
-    curl -s http://localhost:11434/api/tags > /dev/null 2>&1
-    return $?
-}
-
-# Wait for Ollama server to be ready
-echo "Waiting for Ollama server to be ready..."
-MAX_WAIT=60
-WAIT_COUNT=0
-
-while ! check_ollama; do
-    if [ $WAIT_COUNT -ge $MAX_WAIT ]; then
-        echo "❌ Ollama server failed to start within $MAX_WAIT seconds"
-        kill $OLLAMA_PID 2>/dev/null || true
-        exit 1
-    fi
-    
-    echo "⏳ Waiting for Ollama server... ($((WAIT_COUNT + 1))/$MAX_WAIT)"
-    sleep 1
-    WAIT_COUNT=$((WAIT_COUNT + 1))
-done
-
-echo "✅ Ollama server is ready!"
-
-# Optional: Pull a small model for testing
-echo "Pulling a lightweight model (this may take a few minutes)..."
+# Wait for server to start
+echo "Waiting for Ollama server..."
+sleep 10
+
+# Check if server is running
+if curl -s http://localhost:11434/api/tags > /dev/null; then
+    echo "✅ Ollama server is running!"
+else
+    echo "❌ Ollama server failed to start"
+    kill $OLLAMA_PID 2>/dev/null || true
+    exit 1
+fi
+
+# Pull tiny model in background
+echo "Pulling TinyLlama model..."
 ollama pull tinyllama &
 
-# Start the main application
-echo "Starting main application..."
-python3 app.py
-
-# If we reach here, something went wrong with the main app
-echo "❌ Main application exited unexpectedly"
-kill $OLLAMA_PID 2>/dev/null || true
+# Start FastAPI app
+echo "Starting FastAPI application..."
+python3 app.py