Hugging Face's logo

Spaces:
lengfeng1360
/
AIstudioProxyAPI
Runtime error

App Files Community
AIstudioProxyAPI
File size: 4,652 Bytes 
927965d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
 
#!/bin/bash

# AI Studio Proxy API 一键安装脚本 (macOS/Linux)
# 使用 Poetry 进行依赖管理

set -e

# 颜色定义
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# 日志函数
log_info() {
    echo -e "${BLUE}[INFO]${NC} $1"
}

log_success() {
    echo -e "${GREEN}[SUCCESS]${NC} $1"
}

log_warning() {
    echo -e "${YELLOW}[WARNING]${NC} $1"
}

log_error() {
    echo -e "${RED}[ERROR]${NC} $1"
}

# 检查命令是否存在
command_exists() {
    command -v "$1" >/dev/null 2>&1
}

# 检查 Python 版本
check_python() {
    log_info "检查 Python 版本..."
    
    if command_exists python3; then
        PYTHON_CMD="python3"
    elif command_exists python; then
        PYTHON_CMD="python"
    else
        log_error "未找到 Python。请先安装 Python 3.9+"
        exit 1
    fi
    
    PYTHON_VERSION=$($PYTHON_CMD --version 2>&1 | cut -d' ' -f2)
    PYTHON_MAJOR=$(echo $PYTHON_VERSION | cut -d'.' -f1)
    PYTHON_MINOR=$(echo $PYTHON_VERSION | cut -d'.' -f2)
    
    if [ "$PYTHON_MAJOR" -lt 3 ] || ([ "$PYTHON_MAJOR" -eq 3 ] && [ "$PYTHON_MINOR" -lt 9 ]); then
        log_error "Python 版本过低: $PYTHON_VERSION。需要 Python 3.9+"
        exit 1
    fi
    
    log_success "Python 版本: $PYTHON_VERSION ✓"
}

# 安装 Poetry
install_poetry() {
    if command_exists poetry; then
        log_success "Poetry 已安装 ✓"
        return
    fi
    
    log_info "安装 Poetry..."
    curl -sSL https://install.python-poetry.org | $PYTHON_CMD -
    
    # 添加 Poetry 到 PATH
    export PATH="$HOME/.local/bin:$PATH"
    
    if command_exists poetry; then
        log_success "Poetry 安装成功 ✓"
    else
        log_error "Poetry 安装失败。请手动安装 Poetry"
        exit 1
    fi
}

# 克隆项目
clone_project() {
    log_info "克隆项目..."
    
    if [ -d "AIstudioProxyAPI" ]; then
        log_warning "项目目录已存在，跳过克隆"
        cd AIstudioProxyAPI
    else
        git clone https://github.com/CJackHwang/AIstudioProxyAPI.git
        cd AIstudioProxyAPI
        log_success "项目克隆成功 ✓"
    fi
}

# 安装依赖
install_dependencies() {
    log_info "安装项目依赖..."
    poetry install
    log_success "依赖安装成功 ✓"
}

# 下载 Camoufox
download_camoufox() {
    log_info "下载 Camoufox 浏览器..."
    poetry run camoufox fetch
    log_success "Camoufox 下载成功 ✓"
}

# 安装 Playwright 依赖
install_playwright_deps() {
    log_info "安装 Playwright 依赖..."
    poetry run playwright install-deps firefox || {
        log_warning "Playwright 依赖安装失败，但不影响主要功能"
    }
}

# 创建配置文件
create_config() {
    log_info "创建配置文件..."
    
    if [ ! -f ".env" ] && [ -f ".env.example" ]; then
        cp .env.example .env
        log_success "配置文件创建成功 ✓"
        log_info "请编辑 .env 文件进行个性化配置"
    else
        log_warning "配置文件已存在或模板不存在"
    fi
}

# 验证安装
verify_installation() {
    log_info "验证安装..."
    
    # 检查 Poetry 环境
    poetry env info >/dev/null 2>&1 || {
        log_error "Poetry 环境验证失败"
        exit 1
    }
    
    # 检查关键依赖
    poetry run python -c "import fastapi, playwright, camoufox" || {
        log_error "关键依赖验证失败"
        exit 1
    }
    
    log_success "安装验证成功 ✓"
}

# 显示后续步骤
show_next_steps() {
    echo
    log_success "🎉 安装完成！"
    echo
    echo "后续步骤："
    echo "1. 进入项目目录: cd AIstudioProxyAPI"
    echo "2. 激活虚拟环境: poetry env activate"
    echo "3. 配置环境变量: nano .env"
    echo "4. 首次认证设置: python launch_camoufox.py --debug"
    echo "5. 日常运行: python launch_camoufox.py --headless"
    echo
    echo "详细文档："
    echo "- 环境配置: docs/environment-configuration.md"
    echo "- 认证设置: docs/authentication-setup.md"
    echo "- 日常使用: docs/daily-usage.md"
    echo
}

# 主函数
main() {
    echo "🚀 AI Studio Proxy API 一键安装脚本"
    echo "使用 Poetry 进行现代化依赖管理"
    echo

    check_python
    install_poetry
    clone_project
    install_dependencies
    download_camoufox
    install_playwright_deps
    create_config
    verify_installation
    show_next_steps
}

# 运行主函数
main "$@"