基于Flux Schnell模型的MCP图像生成服务器。
- 提供基于MCP协议的图像生成API
- 支持自定义图片尺寸(宽度和高度)
- 支持设置随机种子以复现特定生成结果
- 支持异步流式响应
- 提供HTTP接口调用Hugging Face的模型服务
- Python >= 3.10
- 依赖包:
- httpx >= 0.28.1
- mcp[cli] >= 1.3.0
- 创建并激活 Python 虚拟环境
uv venv && source .venv/bin/activate # Unix/macOS
# 或
.venv\Scripts\activate # Windows
- 安装开发依赖
uv sync # 以可编辑模式安装项目
- 启用调试
mcp dev main.py
或者
npx -y @modelcontextprotocol/inspector uv run main.py
- 调用图像生成工具:
# 示例代码
async def test_main():
img_url = await image_generation(
prompt="your prompt here",
image_width=512, # 可选,默认512
image_height=512, # 可选,默认512
seed=3 # 可选,默认3
)
print(img_url)
prompt
(str): 图像生成提示词image_width
(int, optional): 生成图片宽度,默认512image_height
(int, optional): 生成图片高度,默认512seed
(int, optional): 随机种子,默认3
春天来了,大地苏醒,万物复苏。花儿竞相开放,嫩绿的叶子在微风中轻轻摇曳。空气中弥漫着泥土的芬芳和花儿的香气。小鸟在枝头欢快地歌唱,蝴蝶在花丛中翩翩起舞。阳光洒在大地上,温暖而明亮。春天的生机勃勃,让人心旷神怡。
这个示例展示了使用服务生成的图片效果。您可以在demo目录中找到完整的网页展示代码。
生成的图片URL可以直接用于:
- 网页图片展示
- 社交媒体分享
- 应用程序界面