图像生成服务器

基于Model Context Protocol (MCP)和SANA SPRINT API的文本到图像生成服务器,支持STDIO通信协议和FastMCP库。

  • by devbrother2024
  • Python
  • 本地部署
  • 更新于
在 GitHub 查看 devbrother2024/mcp-generate-image

使用说明

映像生成MCP服务器

使用Model Context Protocol(MCP)创建SANA SPRINT图像的服务器。

功能

  • 基于文本提示生成图像(利用SANA SPRINT API)
  • STDIO通信协议支持
  • 使用FastMCP库

安装和运行

首选参数

# uv를 사용하여 가상환경 생성
uv venv

# 가상환경 활성화
# Windows
.venv\Scripts\activate
# macOS/Linux
source .venv/bin/activate

# 의존성 파일(requirements.txt 또는 pyproject.toml)로부터 패키지 동기화
uv sync

# 또는 개발 모드로 현재 패키지 설치
uv pip install -e .

运行服务器

# 직접 실행
python server.py

# 또는 fastmcp CLI 사용
fastmcp run server.py

作为Docker运行

# Docker 이미지 빌드
docker build -t generate-image-mcp .

# Docker 컨테이너 실행
docker run -it --rm generate-image-mcp

工具列表

  1. generate_image -基于文本提示创建图像。
    • 参数:
      • prompt:用于生成图像的文本说明(必需)

测试方法

  1. 服务器启动

    python server.py
    
  2. 运行客户端(在其他终端上)

    python client_example.py
    

使用的技术

  • FastMCP:用于实现Model Context Protocol的Python SDK
  • Gradio Client:用于调用SANA SPRINT API的客户端库

添加木业图像

添加服务器用于测试的木业映像的步骤 mock_images 在目录中保存图像文件(JPG,PNG)。