AI Image Generator Server

一个基于Model Context Protocol (MCP)的AI图片生成服务器，使用ModelScope FLUX模型自动生成高质量的网页图片。

✨ 功能特点

• 🎨 智能图片生成 - 支持8种网页图片类型（横幅、产品展示、团队照片等）
• 🌐 MCP协议兼容 - 完全符合MCP标准，可集成到支持MCP的客户端
• 🔧 灵活配置 - 支持自定义尺寸、提示词和图片类型
• 🚀 即开即用 - 简单配置即可在Cursor等编辑器中使用
• 🎯 高质量输出 - 基于FLUX模型生成专业级图片

🛠️ 支持的图片类型

类型	描述	用途
hero_banner	网站主横幅的大图	首页横幅、着陆页
product_showcase	电商产品展示图片	产品页面、商品展示
team_photo	专业团队成员照片	关于我们、团队介绍
blog_featured	博客文章特色图片	文章封面、内容配图
service_icon	服务或功能图标	服务介绍、功能展示
background_pattern	背景图案和纹理	页面背景、装饰元素
call_to_action	行动号召区域图片	CTA按钮、引导区域
testimonial_bg	客户评价背景图片	客户证言、评价展示

🚀 快速开始

1. 克隆项目

git clone https://github.com/你的用户名/ai-image-generator-mcp.git
cd ai-image-generator-mcp

2. 安装依赖

python -m venv .venv
source .venv/bin/activate  # Linux/Mac
# 或 .venv\Scripts\activate  # Windows
pip install -r requirements.txt

3. 配置API密钥

获取ModelScope API密钥并设置环境变量：

export MODELSCOPE_API_KEY="你的API密钥"

4. 测试服务器

python test_mcp.py

5. 配置MCP客户端

将以下配置添加到你的MCP客户端配置文件中：

Cursor配置 (~/.cursor/mcp_servers.json)

{
  "mcpServers": {
    "ai-image-generator": {
      "command": "python",
      "args": ["/path/to/your/project/mcp_server.py"],
      "cwd": "/path/to/your/project",
      "env": {
        "MODELSCOPE_API_KEY": "你的API密钥"
      }
    }
  }
}

Claude Desktop配置

在Claude Desktop的设置中添加相同的配置。

🔧 MCP工具说明

1. generate-web-image

根据网页内容和图片类型生成图片

{
  "name": "generate-web-image",
  "arguments": {
    "type": "hero_banner",
    "description": "现代科技公司的主页横幅，展示创新理念",
    "customPrompt": "可选的自定义英文提示词"
  }
}

2. generate-image

生成自定义尺寸的图片

{
  "name": "generate-image",
  "arguments": {
    "prompt": "A futuristic cityscape with flying cars",
    "width": 1920,
    "height": 1080
  }
}

3. generate-custom-web-image

使用完全自定义的提示词生成网页图片

{
  "name": "generate-custom-web-image",
  "arguments": {
    "prompt": "Professional business team in modern office"
  }
}

4. list-image-types

查看所有支持的图片类型

{
  "name": "list-image-types",
  "arguments": {
    "random_string": "test"
  }
}

📁 项目结构

ai-image-generator-mcp/
├── mcp_server.py           # 主MCP服务器文件
├── mcp_config.json         # MCP客户端配置示例
├── image_generator.py      # 图片生成核心逻辑
├── prompt_builder.py       # 提示词构建工具
├── test_mcp.py            # 测试脚本
├── requirements.txt        # Python依赖
├── README.md              # 项目说明
└── MCP_USAGE.md          # 详细使用指南

🔑 API密钥获取

1. 访问 ModelScope
2. 注册并登录账号
3. 绑定阿里云账号
4. 在API管理页面获取密钥

🛠️ 开发和扩展

添加新的图片类型

在 prompt_builder.py 中的 build_web_prompt 函数中添加新的类型映射：

type_styles = {
    "your_new_type": "description of the new image type style",
    # ... existing types
}

自定义提示词模板

修改 prompt_builder.py 中的提示词生成逻辑来适应你的需求。

添加新的AI模型

在 image_generator.py 中修改API调用逻辑来支持其他图片生成模型。

🐛 故障排除

常见问题

1. API密钥错误

   • 确保API密钥正确
   • 确认已绑定阿里云账号

2. 导入错误

   • 确保在正确的虚拟环境中
   • 检查所有依赖是否已安装

3. 网络错误

   • 检查网络连接
   • 确认ModelScope服务状态

调试模式

设置环境变量启用详细日志：

export PYTHONPATH=.
export LOG_LEVEL=DEBUG
python test_mcp.py

📄 许可证

MIT License - 详见 LICENSE 文件

🤝 贡献

欢迎提交Issue和Pull Request！

📞 支持

如有问题，请在GitHub Issues中提出。

---

⭐ 如果这个项目对你有帮助，请给个Star！