File size: 1,912 Bytes
b0c0df0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
# server.py
import base64
import io
from typing import List
from mcp.server.fastmcp import FastMCP
from mcp.types import ImageContent
from PIL import Image
app = FastMCP("demo")
@app.tool(name="image_zoom_in_tool", description="Zoom in on a specific region of an image by cropping it based on a bounding box (bbox) and an optional object label.")
def image_zoom_in_tool(image_path: str, bbox: List[float]):
"""
Zoom in on a specific region of an image by cropping it based on a bounding box (bbox) and an optional object label.
:param image_path: Path to the input image.
:param bbox: Bounding box coordinates in the format [x_min, y_min, x_max, y_max].
:return: Cropped image as a base64-encoded string.
"""
image = Image.open(image_path)
cropped_image = image.crop(bbox)
image_bytes = io.BytesIO()
cropped_image.save(image_bytes, format="PNG")
image_bytes.seek(0)
png = base64.b64encode(image_bytes.getvalue()).decode("utf-8")
return ImageContent(type="image", data=png, mimeType="image/png")
@app.tool(name="weather", description="query weather")
def get_weather(city: str):
weather_data = {"Beijing": {"temp": 25, "condition": "Rainy"}, "Shanghai": {"temp": 28, "condition": "Cloudy"}}
# 返回对应城市的天气信息，如果城市不存在则返回错误信息
result = weather_data.get(city, {"error": "未找到该城市"})
return result
@app.tool(name="get_blank_image", description="get blank image")
def get_blank_image(width: int = 512, height: int = 512):
""" """
image = Image.new("RGB", (width, height), color=(255, 255, 255))
image_bytes = io.BytesIO()
image.save(image_bytes, format="PNG")
image_bytes.seek(0)
png = base64.b64encode(image_bytes.getvalue()).decode("utf-8")
return ImageContent(type="image", data=png, mimeType="image/png")
if __name__ == "__main__":
app.run()
|