# AI海报生成系统

## 概述

AI海报生成系统集成DeepSeek、Kimi、ComfyUI三大AI模型，一键生成Vue组件代码和PSD文件。

**核心功能**:
- 🤖 **AI驱动**: DeepSeek分析 + Kimi文案 + ComfyUI图片
- 🎨 **一键生成**: 输入需求 → Vue代码 + PSD文件
- 📱 **REST API**: 统一接口，支持前端集成
- 🎭 **预定义模板**: lotus.jpg, nku.png, stamp.jpg, background.png
- 🖼️ **自动合成**: 多图层PSD文件生成

## 系统架构

```
前端 → FastAPI(8000) → AI模型集群
              ↓
         Vue代码 + PSD文件
```

## 快速开始

### 环境准备
```bash
# 创建虚拟环境
conda create -n ai_service python=3.11
conda activate ai_service

# 安装依赖
pip install -r requirements.txt

# 配置API密钥
cp .env.example .env
# 编辑 .env 文件，添加:
# DEEPSEEK_API_KEY=your_key
# MOONSHOT_API_KEY=your_key
```

### 启动服务
```bash
cd scripts
python run_pipeline.py  # 选择: 2 (API服务器)
```

### 验证服务
```bash
curl http://localhost:8000/health
```

## API使用

### 生成海报
```bash
curl -X POST http://localhost:8000/api/generate-poster \
  -H "Content-Type: application/json" \
  -d '{"user_input": "端午节海报，传统风格"}'
```

### 下载文件
```bash
curl "http://localhost:8000/api/download/psd?session_id=SESSION_ID" -o poster.psd
curl "http://localhost:8000/api/download/vue?session_id=SESSION_ID" -o poster.vue
```

## 项目结构

```
ai_service/
├── scripts/                 # 核心脚本
│   ├── run_pipeline.py      # API服务器
│   ├── utils.py            # 工具函数
│   ├── generate_layout.py  # Vue代码生成
│   ├── generate_text.py    # 文案生成
│   ├── flux_con.py         # 图片生成
│   └── export_psd_from_json.py # PSD合成
├── configs/                # 配置文件
│   ├── vue_templates.yaml  # Vue模板
│   └── font.yaml          # 字体配置
├── outputs/               # 输出目录
└── workflows/             # ComfyUI工作流
```

## 核心模块

| 模块 | 功能 | AI模型 |
|------|------|--------|
| `utils.py` | 用户输入分析 | DeepSeek |
| `flux_con.py` | 图片生成 | ComfyUI |
| `generate_text.py` | 文案生成 | Kimi |
| `generate_layout.py` | Vue代码生成 | DeepSeek |
| `export_psd_from_json.py` | PSD合成 | - |

## 使用示例

### JavaScript调用
```javascript
const response = await fetch('http://localhost:8000/api/generate-poster', {
  method: 'POST',
  headers: { 'Content-Type': 'application/json' },
  body: JSON.stringify({ user_input: "春节海报，红色背景" })
});

const result = await response.json();
console.log('Vue代码:', result.data.vue_code);
```

### Python调用
```python
import requests

response = requests.post('http://localhost:8000/api/generate-poster', 
                        json={"user_input": "端午节海报，传统风格"})
result = response.json()

if result["status"] == "success":
    print("生成成功!")
    print("Vue代码:", result["data"]["vue_code"])
```

### 本地测试
```bash
cd scripts
python run_pipeline.py  # 选择: 1 (本地测试)
# 输入: "春节海报，红色背景，现代风格"
```

## 部署

### Docker部署
```dockerfile
FROM python:3.11
WORKDIR /app
COPY . .
RUN pip install -r requirements.txt
EXPOSE 8000
CMD ["uvicorn", "scripts.run_pipeline:app", "--host", "0.0.0.0", "--port", "8000"]
```

### PM2部署
```bash
pm2 start "uvicorn scripts.run_pipeline:app --host 0.0.0.0 --port 8000" --name poster-api
```

### Nginx反向代理
```nginx
server {
    listen 80;
    location / {
        proxy_pass http://localhost:8000;
    }
}
```

## 故障排查

### 常见问题
```bash
# 检查API密钥
echo $DEEPSEEK_API_KEY

# 测试ComfyUI连接
curl http://101.201.50.90:8188/system_stats

# 检查端口占用
netstat -tulpn | grep 8000
```

## 更新日志

### v2.0.0 (2025-07-03)
- ✅ 重构代码结构，移除冗余代码
- ✅ 新增预定义Vue模板系统
- ✅ 优化工具函数模块化
- ✅ 简化API接口设计

### v1.0.0 (2025-01-02)
- ✅ 集成DeepSeek + Kimi + ComfyUI
- ✅ 统一API接口设计
- ✅ 支持Vue组件和PSD文件生成