#!/bin/bash
# Tool_OCR - Conda 環境設置腳本 (macOS Apple Silicon)

set -e  # 遇到錯誤立即退出

echo "==================================="
echo "Tool_OCR - 環境設置"
echo "==================================="

# 檢查 Conda 是否已安裝
if command -v conda &> /dev/null; then
    echo "✓ Conda 已安裝: $(conda --version)"
else
    echo "📦 開始安裝 Miniconda..."

    # 下載 Miniconda for Apple Silicon
    MINICONDA_URL="https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-arm64.sh"
    INSTALLER="/tmp/miniconda_installer.sh"

    echo "下載 Miniconda..."
    curl -o "$INSTALLER" "$MINICONDA_URL"

    echo "安裝 Miniconda (默認安裝到 ~/miniconda3)..."
    bash "$INSTALLER" -b -p "$HOME/miniconda3"

    # 初始化 Conda
    echo "初始化 Conda..."
    "$HOME/miniconda3/bin/conda" init zsh bash

    # 清理安裝檔案
    rm "$INSTALLER"

    echo "✓ Miniconda 安裝完成!"
    echo ""
    echo "⚠️  請執行以下命令以載入 Conda:"
    echo "    source ~/.zshrc  (如果使用 zsh)"
    echo "    source ~/.bash_profile  (如果使用 bash)"
    echo ""
    echo "然後重新執行此腳本繼續設置。"
    exit 0
fi

# 檢查是否在 base 環境
CURRENT_ENV=$(conda info --envs | grep '*' | awk '{print $1}')
echo "當前 Conda 環境: $CURRENT_ENV"

# 創建 tool_ocr 環境
ENV_NAME="tool_ocr"
if conda env list | grep -q "^$ENV_NAME "; then
    echo "✓ 環境 '$ENV_NAME' 已存在"
    read -p "是否重新創建? (y/N): " -n 1 -r
    echo
    if [[ $REPLY =~ ^[Yy]$ ]]; then
        echo "移除現有環境..."
        conda env remove -n "$ENV_NAME" -y
    else
        echo "使用現有環境"
        exit 0
    fi
fi

echo "📦 創建 Conda 環境: $ENV_NAME (Python 3.10)..."
conda create -n "$ENV_NAME" python=3.10 -y

echo ""
echo "✅ Conda 環境設置完成!"
echo ""
echo "下一步:"
echo "  1. 啟動環境: conda activate $ENV_NAME"
echo "  2. 安裝依賴: pip install -r requirements.txt"
echo "  3. 下載 PaddleOCR 模型"
echo ""