prestart.ps1

Write-Output "Running preload.ps1"

# Function to check if Ollama is installed
function Check-OllamaInstalled {
    if (Get-Command ollama -ErrorAction SilentlyContinue) {
        Write-Output "Ollama is already installed"
        return $true
    } else {
        Write-Output "Ollama is not installed"
        return $false
    }
}

# Function to check if llava:v1.6 & mxbai-embed-large are installed
function Check-ModelInstalled {
    $llavaInstalled = ollama list | Select-String "llava:v1.6"
    $mxbaiInstalled = ollama list | Select-String "mxbai-embed-large"
    if ($llavaInstalled -and $mxbaiInstalled) {
        Write-Output "All required models are already installed"
        return $true
    } else {
        Write-Output "One or more required models are not installed"
        return $false
    }
}

# Function to install Ollama on Windows
function Install-OllamaWindows {
    Invoke-WebRequest -Uri https://ollama.com/download/ollama-windows-amd64.exe -OutFile "C:\Windows\System32\ollama.exe"
}

# Function to add Ollama as a startup service on Windows
function Setup-ServiceWindows {
    New-Service -Name "Ollama" -Binary "C:\Windows\System32\ollama.exe serve" -Description "Ollama Service" -StartupType Automatic
    Start-Service -Name "Ollama"
}

# Main script
if (-not (Check-OllamaInstalled)) {
    Install-OllamaWindows
    Setup-ServiceWindows
}

if (-not (Check-ModelInstalled)) {
    ollama pull llava:v1.6
    ollama pull mxbai-embed-large
}