#!/bin/bash

# Скрипт для обновления базы данных и переобучения моделей

set -e  # Остановка при ошибке

echo "======================================"
echo "Начало обновления моделей"
echo "======================================"
echo ""

# Активация виртуального окружения
source .venv/bin/activate

# 1. Парсинг про матчей
echo "======================================"
echo "1. Парсинг про матчей..."
echo "======================================"
python start/parse_pro_matches.py
echo ""

# 2. Парсинг деталей про матчей
echo "======================================"
echo "2. Парсинг деталей про матчей..."
echo "======================================"
python start/parse_pro_details_match.py
echo ""

# 3. Построение датасета
echo "======================================"
echo "3. Построение датасета из БД..."
echo "======================================"
python educationML/build_dataset_pro.py
echo ""

# 4. Обучение модели Long-Format (Heroes + Positions)
echo "======================================"
echo "4. Обучение модели Long-Format..."
echo "======================================"
python educationML/train_model_pro.py
echo ""

# 5. Обучение модели Bag of Heroes
echo "======================================"
echo "5. Обучение модели Bag of Heroes..."
echo "======================================"
python educationML/train_model_bag_of_heroes.py
echo ""

# 6. Построение датасета с игроками
echo "======================================"
echo "6. Построение датасета с игроками..."
echo "======================================"
python educationML/build_dataset_with_players.py
echo ""

# 7. Обучение модели With Players
echo "======================================"
echo "7. Обучение модели With Players..."
echo "======================================"
python educationML/train_model_with_players.py
echo ""

# 8. Обучение мета-модели (Stacking)
echo "======================================"
echo "8. Обучение мета-модели (Stacking)..."
echo "======================================"
python educationML/train_model_stacking.py
echo ""

echo "======================================"
echo "✅ Все модели успешно обновлены!"
echo "======================================"