#!/bin/bash
# Initialize Ollama with default models for chatbot service

echo "🚀 Initializing Ollama with default models..."
echo ""

# Wait for Ollama to be ready
echo "⏳ Waiting for Ollama to start..."
sleep 5

# Check if Ollama is running
if ! docker-compose exec ollama ollama list > /dev/null 2>&1; then
    echo "❌ Ollama is not running. Please start services first:"
    echo "   docker-compose up -d"
    exit 1
fi

echo "✓ Ollama is running"
echo ""

# Pull default model (fast, efficient for chatbots)
echo "📥 Pulling llama3.2:3b (Recommended for chatbots - 2GB)..."
docker-compose exec ollama ollama pull llama3.2:3b

echo ""
echo "✓ Default model installed!"
echo ""

# Optional: Pull additional models
echo "Would you like to install additional models? (y/n)"
read -r response

if [[ "$response" =~ ^[Yy]$ ]]; then
    echo ""
    echo "📥 Pulling phi3:3.8b (Great for customer service - 2.3GB)..."
    docker-compose exec ollama ollama pull phi3:3.8b

    echo ""
    echo "📥 Pulling llama3.2:1b (Fastest, smallest - 1GB)..."
    docker-compose exec ollama ollama pull llama3.2:1b
fi

echo ""
echo "🎉 Ollama setup complete!"
echo ""
echo "Available models:"
docker-compose exec ollama ollama list

echo ""
echo "📝 Next steps:"
echo "1. Access Ollama Web UI: http://localhost:3001"
echo "2. Check AI status: curl http://localhost:8000/api/v1/ai/status"
echo "3. Test chatbot: Visit your frontend or use API"
echo ""
echo "💡 Tip: Your chatbot will automatically use Ollama (no API keys needed!)"
echo "   Add OpenAI/Anthropic keys in .env.docker for better performance."