import { CallToolResult } from "@modelcontextprotocol/sdk/types.js"
import { imageDataToBlob } from "../utils"
import z from "zod"
import { RecraftServer } from "../RecraftServer"
import { PARAMETERS } from "../utils/parameters"
import { downloadImage } from "../utils/download"
export const creativeUpscaleTool = {
name: "creative_upscale",
description: "Creative upscale of the input image using Recraft.\n" +
"This operation takes an input image and returns an upscaled image, boosting resolution with a focus on refining small details and faces.\n" +
"This version of upscale is expensive and slower than crisp upscale.\n" +
"Local path or URL to resulting image and its preview will be returned in the response.",
inputSchema: {
type: "object",
properties: {
imageURI: PARAMETERS.imageURI,
},
required: ["imageURI"]
}
}
export const creativeUpscaleHandler = async (server: RecraftServer, args: Record<string, unknown>): Promise<CallToolResult> => {
try {
const { imageURI } = z.object({
imageURI: z.string(),
}).parse(args)
const imageData = await downloadImage(imageURI)
const result = await server.api.imageApi.creativeUpscale({
image: await imageDataToBlob(imageData),
responseFormat: 'url',
expire: server.isLocalResultsStorage,
})
return await server.transformSingleImageOperationToCallToolResult(result.image, 'Creative upscale completed.')
} catch (error) {
return {
content: [
{
type: 'text',
text: `Creative upscale error: ${error}`
}
],
isError: true
}
}
}