#!/bin/bash

# Function to check if Ollama is installed
is_ollama_installed() {
    command -v ollama >/dev/null 2>&1
}

# Install Ollama only if not already installed
if is_ollama_installed; then
    echo "✅ Ollama is already installed. Skipping installation..."
else
    echo "🚀 Installing Ollama..."
    sudo apt-get update && sudo apt-get install -y curl
    curl -fsSL https://ollama.com/install.sh | sh
    echo "✅ Ollama installation completed!"
fi

# Start Ollama
echo "🚀 Starting Ollama..."
ollama serve &
sleep 5

# Pull the model if not already present
if ollama list | grep -q "deepseek-r1:1.5b"; then
    echo "✅ Model 'deepseek-r1:1.5b' is already available."
else
    echo "🚀 Pulling model 'deepseek-r1:1.5b'..."
    ollama pull deepseek-r1:1.5b
    echo "✅ Model pulled successfully!"
fi

echo "🚀 Ollama is running!"