#!/usr/bin/env bash
set -e

echo "============================================"
echo " Установка Cosmo на macOS"
echo "============================================"

# --- Python ---
# Предпочитаем Python 3.12 (лучшая совместимость с ML-пакетами)
PYTHON_BIN=""
for py in python3.12 python3.11 python3.13 python3; do
    if command -v "$py" &>/dev/null; then
        PYTHON_BIN="$py"
        break
    fi
done

if [ -z "$PYTHON_BIN" ]; then
    echo "ОШИБКА: Python3 не найден."
    echo "Установи через Homebrew: brew install python@3.12"
    exit 1
fi

PYTHON_VERSION=$($PYTHON_BIN -c "import sys; print(f'{sys.version_info.major}.{sys.version_info.minor}')")
echo "Python: $PYTHON_VERSION ($PYTHON_BIN)"

# --- Homebrew зависимости ---
if command -v brew &>/dev/null; then
    echo "[1/5] Устанавливаю системные зависимости через Homebrew..."
    brew install portaudio ffmpeg 2>/dev/null || true
else
    echo "Homebrew не найден — пропускаю системные зависимости."
    echo "Если будут ошибки с аудио — установи: /bin/bash -c \"\$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)\""
fi

echo "[2/5] Создаю виртуальное окружение..."
rm -rf venv
$PYTHON_BIN -m venv venv
source venv/bin/activate

echo "[3/5] Обновляю pip и устанавливаю зависимости..."
pip install --upgrade pip
pip install -r requirements.txt

echo "[4/5] Устанавливаю faster-whisper..."
# На Mac (Apple Silicon) используем CPU compute type
pip install faster-whisper

echo "[5/5] Устанавливаю openwakeword..."
pip install openwakeword
python -c "import openwakeword; openwakeword.utils.download_models()" 2>/dev/null || true

echo ""
echo "============================================"
echo " Установка завершена!"
echo ""
echo " Следующие шаги:"
echo " 1. Установи и запусти Ollama: https://ollama.com"
echo " 2. Скачай модель: ollama pull qwen2.5:7b"
echo " 3. Запусти Cosmo: bash run_mac.sh"
echo "============================================"