#!/bin/bash
# 快速测试 Gemini 3 Pro 数学试卷解析

echo "============================================"
echo "  Gemini 3 Pro 数学试卷解析快速测试"
echo "============================================"
echo ""

# 检查是否有试卷文件参数
if [ -z "$1" ]; then
    echo "❌ 错误: 请提供试卷文件路径"
    echo ""
    echo "用法:"
    echo "  ./quick_test_gemini.sh <试卷文件路径> [学科] [年级]"
    echo ""
    echo "示例:"
    echo "  ./quick_test_gemini.sh sample_math.pdf 数学 高一"
    echo "  ./quick_test_gemini.sh physics.pdf 物理"
    echo ""
    exit 1
fi

# 检查文件是否存在
if [ ! -f "$1" ]; then
    echo "❌ 错误: 文件不存在: $1"
    exit 1
fi

# 检查 .env 文件
if [ ! -f ".env" ]; then
    echo "⚠️  警告: 未找到 .env 文件，正在从 .env.example 复制..."
    cp .env.example .env
    echo "✅ 已创建 .env 文件，请编辑并设置 OPENROUTER_API_KEY"
    exit 1
fi

# 检查 API Key
if ! grep -q "OPENROUTER_API_KEY=sk-" .env; then
    echo "❌ 错误: 请在 .env 文件中设置有效的 OPENROUTER_API_KEY"
    echo ""
    echo "步骤:"
    echo "1. 访问 https://openrouter.ai/keys"
    echo "2. 获取 API Key"
    echo "3. 在 .env 文件中设置: OPENROUTER_API_KEY=sk-or-v1-..."
    exit 1
fi

# 设置参数
FILE_PATH="$1"
SUBJECT="${2:-数学}"
GRADE="${3:-}"

echo "📄 试卷文件: $FILE_PATH"
echo "📚 学科: $SUBJECT"
if [ -n "$GRADE" ]; then
    echo "🎓 年级: $GRADE"
fi
echo ""

echo "🚀 开始测试..."
echo ""

# 运行测试脚本
if [ -n "$GRADE" ]; then
    python test_gemini_math.py "$FILE_PATH" "$SUBJECT" "$GRADE"
else
    python test_gemini_math.py "$FILE_PATH" "$SUBJECT"
fi

# 检查结果
if [ $? -eq 0 ]; then
    echo ""
    echo "✅ 测试完成！"
    echo ""
    echo "📊 详细结果已保存到 test_results/ 目录"
    echo ""
    echo "下一步:"
    echo "  - 查看 test_results/ 中的 JSON 文件"
    echo "  - 对比 Claude 和 Gemini 的解析效果"
    echo "  - 在 .env 中设置 AI_MODEL=google/gemini-3-pro-preview 以全局使用"
else
    echo ""
    echo "❌ 测试失败，请检查错误信息"
    exit 1
fi