#!/bin/bash
# ============================================================
#  Запуск MLflow Tracking Server
#  Откройте в браузере: http://localhost:5555
#
#  Backend store = file://$(pwd)/mlruns  (тот же каталог, куда скрипты
#  пишут по умолчанию). Поэтому UI видит все эксперименты/запуски
#  без доп. настройки tracking URI в скриптах.
#
#  Альтернатива — SQLite-бэкенд (нужен, только если хотите SQL-запросы
#  к запускам). Тогда раскомментируйте блок SQLITE ниже и закомментируйте
#  FILE, а в скриптах задайте mlflow.set_tracking_uri("http://localhost:5555")
#  (или export MLFLOW_TRACKING_URI=http://localhost:5555).
# ============================================================
set -e
cd "$(dirname "$0")"

PORT=5555

echo "🚀 Запускаем MLflow Tracking Server..."
echo "   UI:    http://localhost:${PORT}"
echo "   Store: file://${PWD}/mlruns"
echo "   Artifacts: file://${PWD}/artifacts"
echo ""
echo "   Нажмите Ctrl+C для остановки"
echo ""

# ─── FILE backend (по умолчанию — единое хранилище со скриптами) ───
mlflow server \
    --backend-store-uri "file://${PWD}/mlruns" \
    --default-artifact-root "file://${PWD}/artifacts" \
    --host 0.0.0.0 \
    --port "${PORT}"

# ─── SQLITE backend (альтернатива — см. комментарий в шапке) ──────
# mlflow server \
#     --backend-store-uri sqlite:///mlflow.db \
#     --default-artifact-root ./artifacts \
#     --host 0.0.0.0 \
#     --port "${PORT}"