#!/usr/bin/env bash
# 推論サーバーの単体起動スクリプト

set -e

echo "=== Face Detection Inference Server ==="
echo ""

# 環境変数の読み込み
if [ -f ".env" ]; then
    echo "環境変数を読み込み中..."
    export $(cat .env | grep -v '^#' | xargs)
else
    echo "警告: .env ファイルが見つかりません"
fi

# 仮想環境の確認とアクティベート
if [ ! -d ".venv" ]; then
    echo "エラー: .venv が見つかりません"
    echo "仮想環境を作成してください: python -m venv .venv"
    exit 1
fi

source .venv/bin/activate

# モデルの確認
MODEL_PATH="models/yolov8n-face-lindevs.pt"
if [ ! -f "$MODEL_PATH" ]; then
    echo "警告: モデルファイルが見つかりません: $MODEL_PATH"
    echo "最初のリクエスト時にエラーになる可能性があります"
    echo ""
fi

# GPU情報の表示
echo "=== GPU情報 ==="
python -c "
import torch
if torch.cuda.is_available():
    print(f'GPU検出: {torch.cuda.get_device_name(0)}')
    print(f'ROCm version: {torch.version.hip if hasattr(torch.version, \"hip\") else \"N/A\"}')
else:
    print('GPU未検出（CPUモードで動作します）')
" 2>/dev/null || echo "PyTorchが見つかりません"
echo ""

# サーバー起動
echo "=== サーバーを起動中 ==="
echo "URL: http://127.0.0.1:8181"
echo "終了するには Ctrl+C を押してください"
echo ""

python server/main.py