#!/bin/bash

# Alternative AI Model Setup using Ollama
# This provides similar functionality to Docker Model Runner

echo "=== Ollama AI Model Setup ==="
echo ""

# Check if Ollama is installed
if command -v ollama &> /dev/null; then
    echo "Ollama is already installed."
    ollama --version
else
    echo "Installing Ollama..."
    
    # Install Ollama
    curl -fsSL https://ollama.com/install.sh | sh
    
    echo "Ollama installed successfully!"
fi

echo ""
echo "=== Starting Ollama Service ==="
# Start Ollama service
ollama serve &

echo "Waiting for Ollama to start..."
sleep 5

echo ""
echo "=== Available Commands ==="
echo "1. List available models: ollama list"
echo "2. Pull a model: ollama pull llama2"
echo "3. Run a model: ollama run llama2"
echo "4. Pull Hugging Face models: ollama pull huggingface/model-name"
echo ""
echo "=== Popular Models to Try ==="
echo "  ollama pull llama2          # Meta's Llama 2"
echo "  ollama pull codellama       # Code-focused Llama"
echo "  ollama pull mistral         # Mistral 7B"
echo "  ollama pull phi             # Microsoft's Phi-3"
echo "  ollama pull gemma           # Google's Gemma"
echo ""
echo "=== Docker Integration ==="
echo "You can also run Ollama in Docker:"
echo "  docker run -d -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama"
echo "  docker exec -it ollama ollama pull llama2"
echo "  docker exec -it ollama ollama run llama2"