#!/bin/bash
cd "$(dirname "$0")"

echo "=========================================="
echo " IVUS Annotation Tool (Mac/Linux)"
echo "=========================================="

# 1. 仮想環境(venv)の存在チェック
if [ ! -d "venv" ]; then
    echo ""
    echo "[初回セットアップ] 環境構築を開始します..."
    echo "※これには数分かかる場合があります。閉じずにお待ちください。"
    echo ""

    # 仮想環境の作成
    echo "1/3: 仮想環境を作成中..."
    python3 -m venv venv
    if [ $? -ne 0 ]; then
        echo "[ERROR] Python3が見つかりません。インストールしてください。"
        read -p "Press Enter to exit..."
        exit 1
    fi

    # 仮想環境の有効化
    source venv/bin/activate

    # パッケージのインストール
    echo "2/3: 必要なライブラリをインストール中..."
    pip install --upgrade pip
    pip install -r requirements.txt
    if [ $? -ne 0 ]; then
        echo "[ERROR] ライブラリのインストールに失敗しました。"
        read -p "Press Enter to exit..."
        exit 1
    fi

    echo "3/3: セットアップ完了！"
    echo ""
else
    # 2回目以降
    source venv/bin/activate
fi

# 2. アプリケーションの起動
echo "アプリケーションを起動しています..."
streamlit run app.py