#!/bin/bash
# Скрипт инициализации iiEasy AI-платформы
# Выполняет ребрендинг и загрузку модели Ollama

set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROJECT_DIR="$(cd "$SCRIPT_DIR/.." && pwd)"

echo "=== Инициализация iiEasy AI-платформы ==="
echo ""

cd "$PROJECT_DIR"

# Проверка наличия контейнеров
if ! docker ps --format '{{.Names}}' | grep -q "^open-webui$"; then
    echo "Ошибка: Контейнер open-webui не запущен. Запустите: docker compose up -d"
    exit 1
fi

if ! docker ps --format '{{.Names}}' | grep -q "^ollama$"; then
    echo "Ошибка: Контейнер ollama не запущен. Запустите: docker compose up -d"
    exit 1
fi

# 1. Ребрендинг Open WebUI
echo "1. Применение ребрендинга Open WebUI..."
"$SCRIPT_DIR/rebrand.sh"

echo ""
echo "2. Загрузка модели Ollama (gemma3n:e4b-it-fp16)..."
echo "   Это может занять несколько минут в зависимости от скорости интернета..."
docker exec ollama ollama pull gemma3n:e4b-it-fp16

echo ""
echo "3. Проверка загруженных моделей..."
docker exec ollama ollama list

echo ""
echo "=== Инициализация завершена! ==="
echo ""
echo "Проверьте:"
echo "  - Open WebUI: https://odo.iieasy.ru"
echo "  - Логотип и ребрендинг применены"
echo "  - Модель gemma3n:e4b-it-fp16 доступна в Ollama"