#!/bin/bash

get_state() {
    docker ps --format "json" | jq -r "select(.Names == \"$1\") | .State"
}

if [ "$(get_state ai-ollama)" != 'running' ]; then
    echo
    echo "Запуск Ollama"
    echo
    docker run \
        --pull=always \
        --rm \
        --detach \
        --volume ./.data/ollama:/root/.ollama \
        --publish 11434:11434 \
        --name ai-ollama \
        ollama/ollama
fi

if [ "$(get_state ai-webui)" != 'running' ]; then
    echo
    echo "Запуск Open WebUI"
    echo
    docker run \
        --pull=always \
        --rm \
        --detach \
        --volume ./.data/webui:/app/backend/data \
        --publish 9999:8080 \
        --add-host=host.docker.internal:host-gateway \
        --name ai-webui \
        ghcr.io/open-webui/open-webui:main
fi


if [ "$(get_state ai-ollama)" == 'running' ]; then
    echo "* Ollama доступен по адресу: localhost:11434"
fi

if [ "$(get_state ai-webui)" == 'running' ]; then
    echo "* Open WebUI доступен по адресу: http://localhost:9999/"
fi

echo "* Для остановки контейнеров выполните ./down"