#!/bin/bash

# Xinference Conda 环境创建脚本
# 用于创建独立的 Python 环境来运行 Xinference 客户端

set -e

ENV_NAME="xinference"
PYTHON_VERSION="3.10"

echo "========================================="
echo "  创建 Xinference Conda 环境"
echo "========================================="
echo ""

# 检查 conda 是否可用
if ! command -v conda &> /dev/null; then
    echo "❌ 错误: conda 未安装或未在 PATH 中"
    echo "   请先安装 Miniconda 或 Anaconda"
    exit 1
fi

# 检查环境是否已存在
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 "✅ 使用现有环境"
        echo "   激活命令: conda activate $ENV_NAME"
        exit 0
    fi
fi

echo "🔨 创建 conda 环境: ${ENV_NAME} (Python ${PYTHON_VERSION})"
conda create -n $ENV_NAME python=$PYTHON_VERSION -y

echo ""
echo "📦 安装依赖包..."

# 激活环境并安装依赖
eval "$(conda shell.bash hook)"
conda activate $ENV_NAME

# 基础依赖
pip install -U pip

# Xinference 客户端
pip install xinference-client

# 其他可能需要的依赖
pip install numpy
pip install requests

echo ""
echo "✅ 环境创建完成！"
echo ""
echo "========================================="
echo "  环境信息"
echo "========================================="
echo "环境名称: ${ENV_NAME}"
echo "Python 版本: ${PYTHON_VERSION}"
echo ""
echo "🚀 使用方法:"
echo ""
echo "  1. 激活环境:"
echo "     conda activate ${ENV_NAME}"
echo ""
echo "  2. 运行演示:"
echo "     python ecommerce_demo.py"
echo ""
echo "  3. 退出环境:"
echo "     conda deactivate"
echo ""
echo "💡 提示: 将以下别名添加到 ~/.bashrc 或 ~/.zshrc 以快速切换:"
echo "   alias xinference='conda activate ${ENV_NAME}'"
echo ""