Create startup.sh

startup.sh

@@ -0,0 +1,89 @@
+#!/bin/bash
+
+echo "=== Starting Ollama API Server ==="
+
+# Function to check if Ollama is running
+check_ollama() {
+    curl -s http://localhost:11434/api/version > /dev/null
+    return $?
+}
+
+# Function to start Ollama server
+start_ollama() {
+    echo "Starting Ollama server..."
+    su - ollama -c "ollama serve" &
+    OLLAMA_PID=$!
+    echo "Ollama server started with PID: $OLLAMA_PID"
+    
+    # Wait for Ollama to be ready
+    echo "Waiting for Ollama server to be ready..."
+    for i in {1..30}; do
+        if check_ollama; then
+            echo "✅ Ollama server is ready!"
+            break
+        fi
+        echo "⏳ Waiting for Ollama server... ($i/30)"
+        sleep 2
+    done
+    
+    if ! check_ollama; then
+        echo "❌ Ollama server failed to start properly"
+        exit 1
+    fi
+}
+
+# Function to start FastAPI
+start_fastapi() {
+    echo "Starting FastAPI server..."
+    python app.py &
+    FASTAPI_PID=$!
+    echo "FastAPI server started with PID: $FASTAPI_PID"
+}
+
+# Function to handle shutdown
+cleanup() {
+    echo "Shutting down services..."
+    if [ ! -z "$FASTAPI_PID" ]; then
+        kill $FASTAPI_PID 2>/dev/null
+    fi
+    if [ ! -z "$OLLAMA_PID" ]; then
+        kill $OLLAMA_PID 2>/dev/null
+    fi
+    exit 0
+}
+
+# Set up signal handlers
+trap cleanup SIGTERM SIGINT
+
+# Create necessary directories
+mkdir -p /home/ollama/.ollama/models
+chown -R ollama:ollama /home/ollama/.ollama
+
+# Start Ollama server
+start_ollama
+
+# Start FastAPI server
+start_fastapi
+
+# Keep the script running and monitor processes
+echo "🚀 Both services are running!"
+echo "- Ollama server: http://localhost:11434"
+echo "- FastAPI server: http://localhost:7860"
+echo "- API Documentation: http://localhost:7860/docs"
+
+# Monitor both processes
+while true; do
+    # Check if Ollama is still running
+    if ! kill -0 $OLLAMA_PID 2>/dev/null; then
+        echo "❌ Ollama server stopped unexpectedly"
+        exit 1
+    fi
+    
+    # Check if FastAPI is still running
+    if ! kill -0 $FASTAPI_PID 2>/dev/null; then
+        echo "❌ FastAPI server stopped unexpectedly"
+        exit 1
+    fi
+    
+    sleep 10
+done