# 🎨 AI Image Gen MCP Server

Multi-model AI image generation MCP Server. Supports Seedream (Doubao), SDXL, FLUX.1 Schnell/Dev. 30x RTX 4090 GPU accelerated.

## Features

- **10 MCP Tools**: generate_image, generate_from_image, list_models, get_model_info, upscale_image, remove_background, batch_generate, get_generation_status, style_transfer, image_variation
- **4 Models**: Seedream 5.0 (API), SDXL (local), FLUX.1 Schnell (local), FLUX.1 Dev (local)
- **8 Art Styles**: photorealistic, anime, oil-painting, watercolor, 3d-render, pixel-art, chinese-ink, cyberpunk
- **GPU Accelerated**: 30x RTX 4090 for local models, Seedream API for cloud generation

## Quick Start

```bash
# Install
pip install -r requirements.txt

# Run (stdio mode - for Claude/WorkBuddy)
python server.py

# Run (HTTP mode - for remote access)
python server.py --transport http --port 8009
```

## Tools

| Tool | Description |
|------|-------------|
| `generate_image` | Text-to-image generation |
| `generate_from_image` | Image-to-image transformation |
| `list_models` | List available models |
| `get_model_info` | Get model details and pricing |
| `upscale_image` | Super-resolution upscaling (2x/4x) |
| `remove_background` | Background removal |
| `batch_generate` | Generate multiple images at once |
| `get_generation_status` | Check task status |
| `style_transfer` | Apply artistic styles |
| `image_variation` | Generate image variations |

## Models

| Model | Provider | Speed | Quality | Cost |
|-------|----------|-------|---------|------|
| Seedream 5.0 | Doubao API | ~17s | High | ¥0.04/image |
| SDXL | Local 4090 | ~5s | High | Free |
| FLUX.1 Schnell | Local 4090 | ~3s | Good | Free |
| FLUX.1 Dev | Local 4090 | ~15s | Best | Free |

## Environment Variables

- `SEEDREAM_API_KEY`: Seedream API key (default: built-in)
- `SEEDREAM_BASE_URL`: Seedream API base URL
- `COMFYUI_URL`: Local ComfyUI endpoint (default: http://localhost:8188)
- `IMAGE_OUTPUT_DIR`: Output directory for generated images

## MCP Config

```json
{
  "mcpServers": {
    "ai-image-gen": {
      "command": "python",
      "args": ["server.py"],
      "cwd": "/path/to/ai-image-gen-mcp"
    }
  }
}
```


## 在线访问（推荐）

无需本地安装，直接在MCP客户端配置：

```json
{
  "mcpServers": {
    "ai-image-gen-mcp": {
      "url": "http://www.mzse.com/image-gen-mcp/"
    }
  }
}
```

## REST API

```bash
curl http://www.mzse.com/image-gen-mcp/
```

## 部署状态

| 项目 | 地址 |
|------|------|
| 域名 | http://www.mzse.com/image-gen-mcp/ |
| GitHub | https://github.com/zhaohongyuziranerran/ai-image-gen-mcp |

## License

MIT