#!/bin/bash

# 文本视频智能体 API 测试脚本
# Bash 脚本用于运行各种测试示例

echo "🚀 文本视频智能体 API 测试脚本"
echo "================================"
echo ""

# 检查是否在正确的目录
if [ ! -f "Cargo.toml" ]; then
    echo "❌ 错误: 请在 text-video-agent-rust-sdk 目录下运行此脚本"
    exit 1
fi

# 显示可用的测试选项
echo "📋 可用的测试选项:"
echo "1. 健康检查测试 (health_check)"
echo "2. LLM 聊天测试 (llm_chat_test)"
echo "3. Midjourney 图像生成测试 (midjourney_image_test)"
echo "4. 文件上传和媒体分析测试 (file_upload_test)"
echo "5. 综合功能测试 (comprehensive_test)"
echo "6. 运行所有测试"
echo "0. 退出"
echo ""

while true; do
    read -p "请选择要运行的测试 (0-6): " choice
    
    case $choice in
        1)
            echo "🔍 运行健康检查测试..."
            cargo run --example health_check
            break
            ;;
        2)
            echo "🤖 运行 LLM 聊天测试..."
            cargo run --example llm_chat_test
            break
            ;;
        3)
            echo "🎨 运行 Midjourney 图像生成测试..."
            cargo run --example midjourney_image_test
            break
            ;;
        4)
            echo "📁 运行文件上传和媒体分析测试..."
            cargo run --example file_upload_test
            break
            ;;
        5)
            echo "🚀 运行综合功能测试..."
            cargo run --example comprehensive_test
            break
            ;;
        6)
            echo "🎯 运行所有测试..."
            echo ""
            
            echo "1/5 健康检查测试"
            cargo run --example health_check
            echo ""
            
            echo "2/5 LLM 聊天测试"
            cargo run --example llm_chat_test
            echo ""
            
            echo "3/5 Midjourney 图像生成测试"
            cargo run --example midjourney_image_test
            echo ""
            
            echo "4/5 文件上传和媒体分析测试"
            cargo run --example file_upload_test
            echo ""
            
            echo "5/5 综合功能测试"
            cargo run --example comprehensive_test
            echo ""
            
            echo "✅ 所有测试完成！"
            break
            ;;
        0)
            echo "👋 退出测试脚本"
            exit 0
            ;;
        *)
            echo "❌ 无效选择，请输入 0-6 之间的数字"
            ;;
    esac
done

echo ""
echo "🎉 测试完成！"
