first commit
This commit is contained in:
@@ -0,0 +1,221 @@
|
|||||||
|
# 抖音数据AI分析系统
|
||||||
|
|
||||||
|
一个基于AI Agent的智能抖音数据分析系统,支持自然语言交互、自主工具调用和提示词驱动。
|
||||||
|
|
||||||
|
## ✨ 核心特性
|
||||||
|
|
||||||
|
- 🤖 **AI Agent智能代理** - 自然语言交互,自动选择工具完成任务
|
||||||
|
- 📊 **数据抓取** - 支持搜索和创作指导两种数据源
|
||||||
|
- 🧠 **AI分析** - 基于阿里云百炼大模型的深度分析
|
||||||
|
- 🎨 **提示词驱动** - 修改提示词即可控制Agent行为
|
||||||
|
- 🔧 **易于扩展** - 轻松添加新工具和功能
|
||||||
|
|
||||||
|
## 🚀 快速开始
|
||||||
|
|
||||||
|
### 1. 安装依赖
|
||||||
|
|
||||||
|
```bash
|
||||||
|
pip install -r requirements.txt
|
||||||
|
```
|
||||||
|
|
||||||
|
### 2. 配置API Key
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 创建 .env 文件
|
||||||
|
echo "DASHSCOPE_API_KEY=your_api_key_here" > .env
|
||||||
|
```
|
||||||
|
|
||||||
|
获取API Key: https://dashscope.console.aliyun.com/
|
||||||
|
|
||||||
|
### 3. 测试运行
|
||||||
|
|
||||||
|
```bash
|
||||||
|
python test_agent.py
|
||||||
|
```
|
||||||
|
|
||||||
|
## 💡 使用示例
|
||||||
|
|
||||||
|
### 方式1:API接口(推荐)
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 启动服务
|
||||||
|
python api.py
|
||||||
|
|
||||||
|
# 调用Agent
|
||||||
|
curl -X POST "http://localhost:8001/api/agent" \
|
||||||
|
-H "Content-Type: application/json" \
|
||||||
|
-d '{"query": "帮我分析一下游戏类视频的热门趋势"}'
|
||||||
|
```
|
||||||
|
|
||||||
|
### 方式2:命令行
|
||||||
|
|
||||||
|
```bash
|
||||||
|
python ai_agent.py --query "帮我分析一下游戏类视频的热门趋势"
|
||||||
|
```
|
||||||
|
|
||||||
|
### 方式3:Python代码
|
||||||
|
|
||||||
|
```python
|
||||||
|
from ai_agent import create_agent
|
||||||
|
import asyncio
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
agent = create_agent()
|
||||||
|
result = await agent.run("帮我分析一下游戏类视频的热门趋势")
|
||||||
|
print(result["final_answer"])
|
||||||
|
|
||||||
|
asyncio.run(main())
|
||||||
|
```
|
||||||
|
|
||||||
|
## 📖 文档
|
||||||
|
|
||||||
|
- **[START_HERE.md](START_HERE.md)** ⭐ - 新手入门指南
|
||||||
|
- **[SUMMARY.md](SUMMARY.md)** - 功能总结和核心概念
|
||||||
|
- **[README_AGENT.md](README_AGENT.md)** - AI Agent完整文档
|
||||||
|
- [README_AI.md](README_AI.md) - AI分析功能文档
|
||||||
|
- [PROJECT_OVERVIEW.md](PROJECT_OVERVIEW.md) - 项目总览
|
||||||
|
- [AGENT_QUICKSTART.md](AGENT_QUICKSTART.md) - Agent快速开始
|
||||||
|
|
||||||
|
## 🎯 主要功能
|
||||||
|
|
||||||
|
### 1. AI Agent智能代理
|
||||||
|
|
||||||
|
用自然语言描述需求,Agent自动完成:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
"帮我分析一下游戏类视频的热门趋势"
|
||||||
|
"我想做美食相关的内容,给我一些建议"
|
||||||
|
"王者荣耀的视频现在什么内容最火?"
|
||||||
|
```
|
||||||
|
|
||||||
|
### 2. 数据抓取
|
||||||
|
|
||||||
|
- 搜索抖音视频
|
||||||
|
- 获取创作指导数据
|
||||||
|
- 支持分类筛选
|
||||||
|
|
||||||
|
### 3. AI分析
|
||||||
|
|
||||||
|
- 内容趋势分析
|
||||||
|
- 互动数据分析
|
||||||
|
- 创作者分析
|
||||||
|
- 热词分析
|
||||||
|
- 实用建议
|
||||||
|
|
||||||
|
## 🎨 自定义
|
||||||
|
|
||||||
|
### 修改Agent行为
|
||||||
|
|
||||||
|
编辑 `prompts/agent_prompt.md`:
|
||||||
|
|
||||||
|
```markdown
|
||||||
|
## 工作流程
|
||||||
|
|
||||||
|
**场景A:用户提到明确的分类**
|
||||||
|
- 使用 `get_creative_guidance` 获取数据
|
||||||
|
- 使用 `analyze_video_data` 分析
|
||||||
|
- 总结并给出建议
|
||||||
|
```
|
||||||
|
|
||||||
|
### 添加新工具
|
||||||
|
|
||||||
|
在 `ai_agent.py` 中:
|
||||||
|
|
||||||
|
```python
|
||||||
|
# 1. 定义工具函数
|
||||||
|
async def your_tool(param: str) -> Dict:
|
||||||
|
return {"success": True, "data": ...}
|
||||||
|
|
||||||
|
# 2. 注册工具
|
||||||
|
agent.register_tool(
|
||||||
|
name="your_tool",
|
||||||
|
func=your_tool,
|
||||||
|
description="工具描述",
|
||||||
|
parameters={...}
|
||||||
|
)
|
||||||
|
```
|
||||||
|
|
||||||
|
## 🏗️ 项目结构
|
||||||
|
|
||||||
|
```
|
||||||
|
.
|
||||||
|
├── ai_agent.py # AI Agent核心模块 ⭐
|
||||||
|
├── ai_analyzer.py # AI分析模块
|
||||||
|
├── api.py # FastAPI服务
|
||||||
|
├── prompts/
|
||||||
|
│ ├── agent_prompt.md # Agent提示词 ⭐
|
||||||
|
│ └── analyze_prompt.md # 分析提示词
|
||||||
|
├── douyin_data_soupce/ # 数据源模块
|
||||||
|
└── 文档/
|
||||||
|
```
|
||||||
|
|
||||||
|
## 🔧 技术栈
|
||||||
|
|
||||||
|
- **AI模型**: 阿里云百炼(DashScope)
|
||||||
|
- **Web框架**: FastAPI
|
||||||
|
- **爬虫**: Playwright
|
||||||
|
- **语言**: Python 3.8+
|
||||||
|
|
||||||
|
## 📊 API接口
|
||||||
|
|
||||||
|
- `POST /api/agent` - AI智能代理(推荐)
|
||||||
|
- `POST /api/search` - 搜索抖音视频
|
||||||
|
- `POST /api/creative-guidance` - 获取创作指导
|
||||||
|
- `POST /api/analyze` - AI分析数据
|
||||||
|
- `POST /api/analyze-file` - 分析JSON文件
|
||||||
|
|
||||||
|
访问 http://localhost:8001/docs 查看完整API文档
|
||||||
|
|
||||||
|
## 🎓 工作原理
|
||||||
|
|
||||||
|
```
|
||||||
|
用户输入
|
||||||
|
↓
|
||||||
|
AI理解意图
|
||||||
|
↓
|
||||||
|
选择工具
|
||||||
|
↓
|
||||||
|
执行工具
|
||||||
|
↓
|
||||||
|
分析结果
|
||||||
|
↓
|
||||||
|
生成答案
|
||||||
|
```
|
||||||
|
|
||||||
|
## 💡 使用场景
|
||||||
|
|
||||||
|
- ✅ 分析抖音内容趋势
|
||||||
|
- ✅ 获取创作建议
|
||||||
|
- ✅ 研究竞品策略
|
||||||
|
- ✅ 发现热门话题
|
||||||
|
- ✅ 优化内容方向
|
||||||
|
|
||||||
|
## 🆘 常见问题
|
||||||
|
|
||||||
|
### Q: 如何获取API Key?
|
||||||
|
A: 访问 https://dashscope.console.aliyun.com/ 注册并创建
|
||||||
|
|
||||||
|
### Q: 如何自定义Agent行为?
|
||||||
|
A: 编辑 `prompts/agent_prompt.md` 文件
|
||||||
|
|
||||||
|
### Q: 如何添加新功能?
|
||||||
|
A: 在 `ai_agent.py` 中定义并注册新工具
|
||||||
|
|
||||||
|
### Q: 支持哪些分类?
|
||||||
|
A: 全部、美食、旅行、泛生活、汽车、科技、游戏、二次元
|
||||||
|
|
||||||
|
## 📝 License
|
||||||
|
|
||||||
|
MIT
|
||||||
|
|
||||||
|
## 🤝 贡献
|
||||||
|
|
||||||
|
欢迎提交Issue和Pull Request!
|
||||||
|
|
||||||
|
## 📧 联系
|
||||||
|
|
||||||
|
如有问题,请查看文档或提交Issue。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
**开始使用**: 查看 [START_HERE.md](START_HERE.md) 📖
|
||||||
Reference in New Issue
Block a user