"""
测试AI智能分类功能
"""

import asyncio
import os
from dotenv import load_dotenv
from ai_agent import extract_search_keywords

# 加载环境变量
load_dotenv()


async def test_classification():
    """测试AI分类功能"""
    
    test_cases = [
        "我想做一些美食相关的短视频，主要是家常菜的制作教程",
        "我想拍一些关于大学生活的有趣视频，记录校园日常",
        "我想做一些关于健身的短视频，分享简单的居家锻炼方法",
        "想拍一些旅游vlog，记录旅行中的美好瞬间",
        "我想做游戏解说视频",
        "想做一些汽车评测的内容",
        "我想做短视频，但不知道拍什么"
    ]
    
    print("=" * 80)
    print("AI智能分类测试")
    print("=" * 80)
    
    for i, query in enumerate(test_cases, 1):
        print(f"\n测试 {i}:")
        print(f"用户输入: {query}")
        
        result = extract_search_keywords(query)
        
        if result["success"]:
            print(f"✓ 分类成功")
            print(f"  - 主分类: {result['primary_category']}")
            print(f"  - 分类方法: {result['method']}")
        else:
            print(f"✗ 分类失败")
        
        print("-" * 80)


if __name__ == "__main__":
    asyncio.run(test_classification())