KICCO_AI_IMAGE/mcp_server

MCP 服务器 - FLUX风格塑形

基于Model Context Protocol (MCP) 实现的服务器，提供FLUX风格塑形功能。

功能特点

• 利用MCP协议提供标准化的API接口
• 支持FLUX风格塑形功能，将结构图像和风格图像结合生成新图像
• 提供资源、工具和提示模板三种MCP核心组件

安装步骤

1. 安装依赖项

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

2. 获取必要的模型访问权限

FLUX模型需要从Hugging Face下载，部分模型需要授权访问：

• 访问 https://huggingface.co/settings/tokens 创建访问令牌
• 访问以下模型页面申请访问权限：
  • https://huggingface.co/black-forest-labs/FLUX.1-Redux-dev
  • https://huggingface.co/black-forest-labs/FLUX.1-Depth-dev
  • https://huggingface.co/black-forest-labs/FLUX.1-dev
• 创建.env文件并添加你的令牌：

  HUGGING_FACE_HUB_TOKEN=your_token_here
  

使用方法

启动服务器

python mcp_server/flux_starter.py --host 0.0.0.0 --port 8189

可用参数：

• --host: 监听地址（默认：127.0.0.1）
• --port: 监听端口（默认：8189）
• --name: 服务器名称（默认：FLUX-MCP）

客户端演示

服务器启动后，可以使用演示客户端测试功能：

python mcp_server/flux_demo_client.py --structure 图像路径.jpg --style 风格图像路径.jpg

可用参数：

• --host: 服务器地址（默认：127.0.0.1）
• --port: 服务器端口（默认：8189）
• --structure: 结构图像路径（必需）
• --style: 风格图像路径（必需）
• --prompt: 文本提示（默认为空）
• --depth-strength: 深度强度（默认：15.0）
• --style-strength: 风格强度（默认：0.5）
• --output: 输出目录（默认：./output）

MCP资源和工具

本服务器提供以下MCP组件：

资源 (Resources)

• flux://status: 获取FLUX服务状态

工具 (Tools)

• 生成风格化图像: 生成风格化图像
  • 参数：
    • prompt: 文本提示
    • structure_image_base64: 结构图像（Base64编码）
    • style_image_base64: 风格图像（Base64编码）
    • depth_strength: 深度强度（默认：15.0）
    • style_strength: 风格强度（默认：0.5）

提示 (Prompts)

• 风格转移提示: 生成风格转移提示模板
  • 参数：
    • subject: 主题
    • style: 风格

开发信息

• 需要ComfyUI环境作为基础
• 使用mcp Python SDK版本 1.0.0 或更高
• 完全中文注释，便于理解和维护