Skip to content

GMKR/mcp-imagegen

Repository files navigation

MCP Image Generator

A Model Context Protocol (MCP) server for generating images using Together AI's image generation models. This MCP Server can be run locally or using an SSE endpoint. The MCP Image Generator required a provider, only "Replicate" and "Together" are supported currently. You need to set the TOGETHER_API_KEY or REPLICATE_API_TOKEN environment variables. and set the PROVIDER environment variable to "replicate" or "together"/

SSE Endpoint (Docker environment)

Clone the repository

git clone https://github.com/gmkr/mcp-imagegen.git
cd mcp-imagegen

Build and run Docker container

docker build -f Dockerfile.server -t mcp-imagegen .
docker run -p 3000:3000 mcp-imagegen

Configuring with MCP Client

{
  "mcpServers": {
    "imagegenerator": {
      "url": "http://localhost:3000/sse",
      "env": {
        "PROVIDER": "replicate",
        "REPLICATE_API_TOKEN": "your-replicate-api-token"
      }
    }
  }
}

Adjust the url to the endpoint of the MCP server you want to use. provider can be "replicate" or "together".

Running locally using stdio

Prerequisites

  • Node.js
  • Together AI API key or Replicate API token

Installation

  1. Clone the repository:

    git clone https://github.com/gmkr/mcp-imagegen.git
    cd mcp-imagegen
  2. Install dependencies:

    pnpm install

Configuration

Create a configuration file for your MCP client. Here's an example configuration:

{
  "mcpServers": {
    "imagegenerator": {
      "command": "pnpx",
      "args": [
        "-y",
        "tsx",
        "/path/to/mcp-imagegen/src/index.ts"
      ],
      "env": {
        "PROVIDER": "replicate",
        "REPLICATE_API_TOKEN": "your-replicate-api-token"
      }
    }
  }
}

Replace /path/to/mcp-imagegen with the absolute path to your cloned repository and your-replicate-api-token with your actual Replicate API token.

Usage

The MCP Image Generator provides a tool called generate_image that can be used to generate images based on text prompts.

Tool: generate_image

Generates an image based on the provided prompt.

Parameters:

  • prompt (string): The text prompt to generate an image for
  • width (number, optional): The width of the image to generate (default: 512)
  • height (number, optional): The height of the image to generate (default: 512)
  • numberOfImages (number, optional): The number of images to generate (default: 1)

Environment Variables

  • PROVIDER: The provider to use for image generation (default: "replicate")
  • REPLICATE_API_TOKEN: Your Replicate API token
  • TOGETHER_API_KEY: Your Together AI API key
  • MODEL_NAME: The model to use for image generation (default: "black-forest-labs/flux-schnell")

License

MIT

About

MCP Server for Generating images

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published