text2image-mcp 2.0.0

package/dist/server.js

@@ -0,0 +1,98 @@
+// src/server.ts
+import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { CallToolRequestSchema, ListToolsRequestSchema, ErrorCode, McpError, } from "@modelcontextprotocol/sdk/types.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { ConfigManager } from "./config.js";
+import { ChatSessionManager } from "./chat-session.js";
+// Tool definitions & handlers
+import { configureGeminiTokenTool, getConfigurationStatusTool, handleConfigureToken, handleGetConfigStatus } from "./tools/config-tools.js";
+import { generateImageTool, handleGenerateImage } from "./tools/generate.js";
+import { editImageTool, handleEditImage } from "./tools/edit.js";
+import { continueEditingTool, handleContinueEditing } from "./tools/continue.js";
+import { getLastImageInfoTool, handleGetLastImageInfo } from "./tools/info.js";
+export class NanoBananaMCP {
+    server;
+    configManager;
+    chatSession = null;
+    lastImagePath = null;
+    constructor() {
+        this.configManager = new ConfigManager();
+        this.server = new Server({ name: "text2image-mcp", version: "2.0.0" }, { capabilities: { tools: {} } });
+        this.setupHandlers();
+    }
+    setupHandlers() {
+        // List tools
+        this.server.setRequestHandler(ListToolsRequestSchema, async () => ({
+            tools: [
+                configureGeminiTokenTool,
+                getConfigurationStatusTool,
+                generateImageTool,
+                editImageTool,
+                continueEditingTool,
+                getLastImageInfoTool,
+            ],
+        }));
+        // Call tools
+        this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
+            try {
+                return await this.routeToolCall(request);
+            }
+            catch (error) {
+                if (error instanceof McpError)
+                    throw error;
+                throw new McpError(ErrorCode.InternalError, `Tool execution failed: ${error instanceof Error ? error.message : String(error)}`);
+            }
+        });
+    }
+    async routeToolCall(request) {
+        const { name } = request.params;
+        // Config tools don't need API key
+        if (name === "configure_gemini_token") {
+            return handleConfigureToken(request, this.configManager);
+        }
+        if (name === "get_configuration_status") {
+            return handleGetConfigStatus(this.configManager);
+        }
+        // All other tools need configured API
+        if (!this.configManager.isConfigured()) {
+            throw new McpError(ErrorCode.InvalidRequest, "Gemini API token not configured. Use configure_gemini_token first or set GEMINI_API_KEY env var.");
+        }
+        const genAI = this.configManager.genAI;
+        // Lazy-init chat session manager
+        if (!this.chatSession) {
+            this.chatSession = new ChatSessionManager(genAI);
+        }
+        switch (name) {
+            case "generate_image": {
+                const { result, savedPath } = await handleGenerateImage(request, genAI, this.chatSession);
+                if (savedPath)
+                    this.lastImagePath = savedPath;
+                return result;
+            }
+            case "edit_image": {
+                const { result, savedPath } = await handleEditImage(request, genAI);
+                if (savedPath) {
+                    this.lastImagePath = savedPath;
+                    this.chatSession.reset(); // New base image, reset chat context
+                }
+                return result;
+            }
+            case "continue_editing": {
+                const { result, savedPath } = await handleContinueEditing(request, genAI, this.chatSession, this.lastImagePath);
+                if (savedPath)
+                    this.lastImagePath = savedPath;
+                return result;
+            }
+            case "get_last_image_info":
+                return handleGetLastImageInfo(this.lastImagePath);
+            default:
+                throw new McpError(ErrorCode.MethodNotFound, `Unknown tool: ${name}`);
+        }
+    }
+    async run() {
+        await this.configManager.load();
+        const transport = new StdioServerTransport();
+        await this.server.connect(transport);
+    }
+}
+//# sourceMappingURL=server.js.map

package/dist/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAAA,gBAAgB;AAChB,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EACL,qBAAqB,EACrB,sBAAsB,EAGtB,SAAS,EACT,QAAQ,GACT,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAEvD,8BAA8B;AAC9B,OAAO,EAAE,wBAAwB,EAAE,0BAA0B,EAAE,oBAAoB,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAC5I,OAAO,EAAE,iBAAiB,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAC7E,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AACjE,OAAO,EAAE,mBAAmB,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AACjF,OAAO,EAAE,oBAAoB,EAAE,sBAAsB,EAAE,MAAM,iBAAiB,CAAC;AAE/E,MAAM,OAAO,aAAa;IAChB,MAAM,CAAS;IACf,aAAa,CAAgB;IAC7B,WAAW,GAA8B,IAAI,CAAC;IAC9C,aAAa,GAAkB,IAAI,CAAC;IAE5C;QACE,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;QACzC,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CACtB,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,OAAO,EAAE,EAC5C,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAChC,CAAC;QACF,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC;IAEO,aAAa;QACnB,aAAa;QACb,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;YACjE,KAAK,EAAE;gBACL,wBAAwB;gBACxB,0BAA0B;gBAC1B,iBAAiB;gBACjB,aAAa;gBACb,mBAAmB;gBACnB,oBAAoB;aACrB;SACF,CAAC,CAAC,CAAC;QAEJ,aAAa;QACb,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAwB,EAA2B,EAAE;YAC/G,IAAI,CAAC;gBACH,OAAO,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;YAC3C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,KAAK,YAAY,QAAQ;oBAAE,MAAM,KAAK,CAAC;gBAC3C,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,0BAA0B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CACnF,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,OAAwB;QAClD,MAAM,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QAEhC,kCAAkC;QAClC,IAAI,IAAI,KAAK,wBAAwB,EAAE,CAAC;YACtC,OAAO,oBAAoB,CAAC,OAAO,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QAC3D,CAAC;QACD,IAAI,IAAI,KAAK,0BAA0B,EAAE,CAAC;YACxC,OAAO,qBAAqB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACnD,CAAC;QAED,sCAAsC;QACtC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,YAAY,EAAE,EAAE,CAAC;YACvC,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,cAAc,EAAE,kGAAkG,CAAC,CAAC;QACnJ,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,KAAM,CAAC;QAExC,iCAAiC;QACjC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,IAAI,CAAC,WAAW,GAAG,IAAI,kBAAkB,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC;QAED,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,gBAAgB,CAAC,CAAC,CAAC;gBACtB,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,mBAAmB,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;gBAC1F,IAAI,SAAS;oBAAE,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;gBAC9C,OAAO,MAAM,CAAC;YAChB,CAAC;YAED,KAAK,YAAY,CAAC,CAAC,CAAC;gBAClB,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,eAAe,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;gBACpE,IAAI,SAAS,EAAE,CAAC;oBACd,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;oBAC/B,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC,CAAC,qCAAqC;gBACjE,CAAC;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC;YAED,KAAK,kBAAkB,CAAC,CAAC,CAAC;gBACxB,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,qBAAqB,CACvD,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,aAAa,CACrD,CAAC;gBACF,IAAI,SAAS;oBAAE,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;gBAC9C,OAAO,MAAM,CAAC;YAChB,CAAC;YAED,KAAK,qBAAqB;gBACxB,OAAO,sBAAsB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAEpD;gBACE,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,cAAc,EAAE,iBAAiB,IAAI,EAAE,CAAC,CAAC;QAC1E,CAAC;IACH,CAAC;IAED,KAAK,CAAC,GAAG;QACP,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;QAChC,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAC7C,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACvC,CAAC;CACF"}

package/dist/tools/config-tools.d.ts

@@ -0,0 +1,7 @@
+import { CallToolRequest, CallToolResult, Tool } from "@modelcontextprotocol/sdk/types.js";
+import { ConfigManager } from "../config.js";
+export declare const configureGeminiTokenTool: Tool;
+export declare const getConfigurationStatusTool: Tool;
+export declare function handleConfigureToken(request: CallToolRequest, configManager: ConfigManager): Promise<CallToolResult>;
+export declare function handleGetConfigStatus(configManager: ConfigManager): Promise<CallToolResult>;
+//# sourceMappingURL=config-tools.d.ts.map

package/dist/tools/config-tools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"config-tools.d.ts","sourceRoot":"","sources":["../../src/tools/config-tools.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,oCAAoC,CAAC;AAC3F,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAE7C,eAAO,MAAM,wBAAwB,EAAE,IAatC,CAAC;AAEF,eAAO,MAAM,0BAA0B,EAAE,IAQxC,CAAC;AAEF,wBAAsB,oBAAoB,CACxC,OAAO,EAAE,eAAe,EACxB,aAAa,EAAE,aAAa,GAC3B,OAAO,CAAC,cAAc,CAAC,CAUzB;AAED,wBAAsB,qBAAqB,CACzC,aAAa,EAAE,aAAa,GAC3B,OAAO,CAAC,cAAc,CAAC,CAczB"}

package/dist/tools/config-tools.js

@@ -0,0 +1,48 @@
+export const configureGeminiTokenTool = {
+    name: "configure_gemini_token",
+    description: "Configure your Gemini API token for nano-banana image generation",
+    inputSchema: {
+        type: "object",
+        properties: {
+            apiKey: {
+                type: "string",
+                description: "Your Gemini API key from Google AI Studio",
+            },
+        },
+        required: ["apiKey"],
+    },
+};
+export const getConfigurationStatusTool = {
+    name: "get_configuration_status",
+    description: "Check if Gemini API token is configured",
+    inputSchema: {
+        type: "object",
+        properties: {},
+        additionalProperties: false,
+    },
+};
+export async function handleConfigureToken(request, configManager) {
+    const { apiKey } = request.params.arguments;
+    await configManager.setApiKey(apiKey);
+    return {
+        content: [{
+                type: "text",
+                text: "Gemini API token configured successfully. You can now use nano-banana image generation features.",
+            }],
+    };
+}
+export async function handleGetConfigStatus(configManager) {
+    const isConfigured = configManager.isConfigured();
+    let statusText;
+    if (isConfigured) {
+        const sourceLabel = configManager.configSource === "environment"
+            ? "Environment variable (GEMINI_API_KEY)"
+            : "Local configuration file (.nano-banana-config.json)";
+        statusText = `Gemini API token is configured and ready.\nSource: ${sourceLabel}`;
+    }
+    else {
+        statusText = `Gemini API token is not configured.\n\nConfiguration options (priority order):\n1. MCP client env: "env": { "GEMINI_API_KEY": "your-key" }\n2. System env: export GEMINI_API_KEY="your-key"\n3. Use configure_gemini_token tool`;
+    }
+    return { content: [{ type: "text", text: statusText }] };
+}
+//# sourceMappingURL=config-tools.js.map

package/dist/tools/config-tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config-tools.js","sourceRoot":"","sources":["../../src/tools/config-tools.ts"],"names":[],"mappings":"AAIA,MAAM,CAAC,MAAM,wBAAwB,GAAS;IAC5C,IAAI,EAAE,wBAAwB;IAC9B,WAAW,EAAE,kEAAkE;IAC/E,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,2CAA2C;aACzD;SACF;QACD,QAAQ,EAAE,CAAC,QAAQ,CAAC;KACrB;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,0BAA0B,GAAS;IAC9C,IAAI,EAAE,0BAA0B;IAChC,WAAW,EAAE,yCAAyC;IACtD,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE,EAAE;QACd,oBAAoB,EAAE,KAAK;KAC5B;CACF,CAAC;AAEF,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,OAAwB,EACxB,aAA4B;IAE5B,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,SAA+B,CAAC;IAClE,MAAM,aAAa,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAEtC,OAAO;QACL,OAAO,EAAE,CAAC;gBACR,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,kGAAkG;aACzG,CAAC;KACH,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,aAA4B;IAE5B,MAAM,YAAY,GAAG,aAAa,CAAC,YAAY,EAAE,CAAC;IAElD,IAAI,UAAkB,CAAC;IACvB,IAAI,YAAY,EAAE,CAAC;QACjB,MAAM,WAAW,GAAG,aAAa,CAAC,YAAY,KAAK,aAAa;YAC9D,CAAC,CAAC,uCAAuC;YACzC,CAAC,CAAC,qDAAqD,CAAC;QAC1D,UAAU,GAAG,sDAAsD,WAAW,EAAE,CAAC;IACnF,CAAC;SAAM,CAAC;QACN,UAAU,GAAG,iOAAiO,CAAC;IACjP,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC;AAC3D,CAAC"}

package/dist/tools/continue.d.ts

@@ -0,0 +1,9 @@
+import { CallToolRequest, CallToolResult, Tool } from "@modelcontextprotocol/sdk/types.js";
+import { GoogleGenAI } from "@google/genai";
+import { ChatSessionManager } from "../chat-session.js";
+export declare const continueEditingTool: Tool;
+export declare function handleContinueEditing(request: CallToolRequest, genAI: GoogleGenAI, chatSession: ChatSessionManager, lastImagePath: string | null): Promise<{
+    result: CallToolResult;
+    savedPath: string | null;
+}>;
+//# sourceMappingURL=continue.d.ts.map

package/dist/tools/continue.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"continue.d.ts","sourceRoot":"","sources":["../../src/tools/continue.ts"],"names":[],"mappings":"AACA,OAAO,EACL,eAAe,EAAE,cAAc,EAAuB,IAAI,EAC3D,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAG5C,OAAO,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AAExD,eAAO,MAAM,mBAAmB,EAAE,IA6BjC,CAAC;AAEF,wBAAsB,qBAAqB,CACzC,OAAO,EAAE,eAAe,EACxB,KAAK,EAAE,WAAW,EAClB,WAAW,EAAE,kBAAkB,EAC/B,aAAa,EAAE,MAAM,GAAG,IAAI,GAC3B,OAAO,CAAC;IAAE,MAAM,EAAE,cAAc,CAAC;IAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAA;CAAE,CAAC,CAmD/D"}

package/dist/tools/continue.js

@@ -0,0 +1,111 @@
+// src/tools/continue.ts
+import { ErrorCode, McpError, } from "@modelcontextprotocol/sdk/types.js";
+import { ASPECT_RATIOS, RESOLUTIONS } from "../types.js";
+import { saveImageFromBase64, readImageAsBase64 } from "../image-utils.js";
+export const continueEditingTool = {
+    name: "continue_editing",
+    description: "Continue editing the LAST generated/edited image using multi-turn conversation. The model remembers previous context. Supports reference images, aspect ratio, and resolution changes.",
+    inputSchema: {
+        type: "object",
+        properties: {
+            prompt: {
+                type: "string",
+                description: "Text describing the changes to make to the last image",
+            },
+            referenceImages: {
+                type: "array",
+                items: { type: "string" },
+                description: "Optional file paths to reference images for style/content guidance",
+            },
+            aspectRatio: {
+                type: "string",
+                enum: [...ASPECT_RATIOS],
+                description: "Change the aspect ratio for this edit",
+            },
+            resolution: {
+                type: "string",
+                enum: [...RESOLUTIONS],
+                description: "Change the resolution for this edit (Pro model only)",
+            },
+        },
+        required: ["prompt"],
+    },
+};
+export async function handleContinueEditing(request, genAI, chatSession, lastImagePath) {
+    if (!lastImagePath) {
+        throw new McpError(ErrorCode.InvalidRequest, "No previous image found. Use generate_image or edit_image first.");
+    }
+    const args = request.params.arguments;
+    // If no active chat session, fall back to re-sending the last image
+    if (!chatSession.hasActiveSession()) {
+        const mainImage = await readImageAsBase64(lastImagePath);
+        const parts = [
+            { inlineData: { data: mainImage.base64, mimeType: mainImage.mimeType } },
+        ];
+        if (args.referenceImages) {
+            for (const refPath of args.referenceImages) {
+                try {
+                    const refImg = await readImageAsBase64(refPath);
+                    parts.push({ inlineData: { data: refImg.base64, mimeType: refImg.mimeType } });
+                }
+                catch { /* skip */ }
+            }
+        }
+        parts.push({ text: args.prompt });
+        const config = { responseModalities: ["TEXT", "IMAGE"] };
+        const imageConfig = {};
+        if (args.aspectRatio)
+            imageConfig.aspectRatio = args.aspectRatio;
+        if (args.resolution)
+            imageConfig.imageSize = args.resolution;
+        if (Object.keys(imageConfig).length > 0)
+            config.imageConfig = imageConfig;
+        const response = await genAI.models.generateContent({
+            model: "gemini-2.5-flash-image",
+            contents: [{ parts }],
+            config,
+        });
+        return processResponse(response, args);
+    }
+    // Active chat session exists — use it for true multi-turn
+    const response = await chatSession.sendMessage(args.prompt);
+    return processResponse(response, args);
+}
+async function processResponse(response, args) {
+    const content = [];
+    let savedPath = null;
+    let textContent = "";
+    if (response.candidates?.[0]?.content?.parts) {
+        for (const part of response.candidates[0].content.parts) {
+            if (part.text && !part.thought)
+                textContent += part.text;
+            if (part.inlineData?.data && !part.thought) {
+                const saved = await saveImageFromBase64(part.inlineData.data, "edited");
+                savedPath = saved.filePath;
+                content.push({
+                    type: "image",
+                    data: part.inlineData.data,
+                    mimeType: part.inlineData.mimeType || "image/png",
+                });
+            }
+        }
+    }
+    let statusText = `Image updated with nano-banana (continue_editing)`;
+    statusText += `\n\nEdit prompt: "${args.prompt}"`;
+    if (args.aspectRatio)
+        statusText += `\nAspect ratio: ${args.aspectRatio}`;
+    if (args.resolution)
+        statusText += `\nResolution: ${args.resolution}`;
+    if (textContent)
+        statusText += `\n\nDescription: ${textContent}`;
+    if (savedPath) {
+        statusText += `\n\nEdited image saved to: ${savedPath}`;
+        statusText += `\nUse continue_editing again for further changes.`;
+    }
+    else {
+        statusText += `\n\nNo edited image generated. Try again.`;
+    }
+    content.unshift({ type: "text", text: statusText });
+    return { result: { content }, savedPath };
+}
+//# sourceMappingURL=continue.js.map

package/dist/tools/continue.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"continue.js","sourceRoot":"","sources":["../../src/tools/continue.ts"],"names":[],"mappings":"AAAA,wBAAwB;AACxB,OAAO,EAC4B,SAAS,EAAE,QAAQ,GACrD,MAAM,oCAAoC,CAAC;AAE5C,OAAO,EAAE,aAAa,EAAE,WAAW,EAA2B,MAAM,aAAa,CAAC;AAClF,OAAO,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAG3E,MAAM,CAAC,MAAM,mBAAmB,GAAS;IACvC,IAAI,EAAE,kBAAkB;IACxB,WAAW,EACT,wLAAwL;IAC1L,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,uDAAuD;aACrE;YACD,eAAe,EAAE;gBACf,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACzB,WAAW,EAAE,oEAAoE;aAClF;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,GAAG,aAAa,CAAC;gBACxB,WAAW,EAAE,uCAAuC;aACrD;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,GAAG,WAAW,CAAC;gBACtB,WAAW,EAAE,sDAAsD;aACpE;SACF;QACD,QAAQ,EAAE,CAAC,QAAQ,CAAC;KACrB;CACF,CAAC;AAEF,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,OAAwB,EACxB,KAAkB,EAClB,WAA+B,EAC/B,aAA4B;IAE5B,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,cAAc,EACxB,kEAAkE,CACnE,CAAC;IACJ,CAAC;IAED,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,SAK3B,CAAC;IAEF,oEAAoE;IACpE,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE,EAAE,CAAC;QACpC,MAAM,SAAS,GAAG,MAAM,iBAAiB,CAAC,aAAa,CAAC,CAAC;QACzD,MAAM,KAAK,GAAU;YACnB,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,CAAC,MAAM,EAAE,QAAQ,EAAE,SAAS,CAAC,QAAQ,EAAE,EAAE;SACzE,CAAC;QAEF,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;gBAC3C,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,OAAO,CAAC,CAAC;oBAChD,KAAK,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;gBACjF,CAAC;gBAAC,MAAM,CAAC,CAAC,UAAU,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QAElC,MAAM,MAAM,GAAQ,EAAE,kBAAkB,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC;QAC9D,MAAM,WAAW,GAAQ,EAAE,CAAC;QAC5B,IAAI,IAAI,CAAC,WAAW;YAAE,WAAW,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;QACjE,IAAI,IAAI,CAAC,UAAU;YAAE,WAAW,CAAC,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC;QAC7D,IAAI,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,GAAG,CAAC;YAAE,MAAM,CAAC,WAAW,GAAG,WAAW,CAAC;QAE1E,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,eAAe,CAAC;YAClD,KAAK,EAAE,wBAAwB;YAC/B,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC;YACrB,MAAM;SACP,CAAC,CAAC;QAEH,OAAO,eAAe,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;IACzC,CAAC;IAED,0DAA0D;IAC1D,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC5D,OAAO,eAAe,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;AACzC,CAAC;AAED,KAAK,UAAU,eAAe,CAC5B,QAAa,EACb,IAAmE;IAEnE,MAAM,OAAO,GAAU,EAAE,CAAC;IAC1B,IAAI,SAAS,GAAkB,IAAI,CAAC;IACpC,IAAI,WAAW,GAAG,EAAE,CAAC;IAErB,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QAC7C,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACxD,IAAI,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO;gBAAE,WAAW,IAAI,IAAI,CAAC,IAAI,CAAC;YACzD,IAAI,IAAI,CAAC,UAAU,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;gBAC3C,MAAM,KAAK,GAAG,MAAM,mBAAmB,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;gBACxE,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC;gBAC3B,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,IAAI;oBAC1B,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,QAAQ,IAAI,WAAW;iBAClD,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,UAAU,GAAG,mDAAmD,CAAC;IACrE,UAAU,IAAI,qBAAqB,IAAI,CAAC,MAAM,GAAG,CAAC;IAClD,IAAI,IAAI,CAAC,WAAW;QAAE,UAAU,IAAI,mBAAmB,IAAI,CAAC,WAAW,EAAE,CAAC;IAC1E,IAAI,IAAI,CAAC,UAAU;QAAE,UAAU,IAAI,iBAAiB,IAAI,CAAC,UAAU,EAAE,CAAC;IACtE,IAAI,WAAW;QAAE,UAAU,IAAI,oBAAoB,WAAW,EAAE,CAAC;IAEjE,IAAI,SAAS,EAAE,CAAC;QACd,UAAU,IAAI,8BAA8B,SAAS,EAAE,CAAC;QACxD,UAAU,IAAI,mDAAmD,CAAC;IACpE,CAAC;SAAM,CAAC;QACN,UAAU,IAAI,2CAA2C,CAAC;IAC5D,CAAC;IAED,OAAO,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;IACpD,OAAO,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE,EAAE,SAAS,EAAE,CAAC;AAC5C,CAAC"}

package/dist/tools/edit.d.ts

@@ -0,0 +1,8 @@
+import { CallToolRequest, CallToolResult, Tool } from "@modelcontextprotocol/sdk/types.js";
+import { GoogleGenAI } from "@google/genai";
+export declare const editImageTool: Tool;
+export declare function handleEditImage(request: CallToolRequest, genAI: GoogleGenAI): Promise<{
+    result: CallToolResult;
+    savedPath: string | null;
+}>;
+//# sourceMappingURL=edit.d.ts.map

package/dist/tools/edit.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"edit.d.ts","sourceRoot":"","sources":["../../src/tools/edit.ts"],"names":[],"mappings":"AACA,OAAO,EACL,eAAe,EAAE,cAAc,EAAE,IAAI,EACtC,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAI5C,eAAO,MAAM,aAAa,EAAE,IAsC3B,CAAC;AAEF,wBAAsB,eAAe,CACnC,OAAO,EAAE,eAAe,EACxB,KAAK,EAAE,WAAW,GACjB,OAAO,CAAC;IAAE,MAAM,EAAE,cAAc,CAAC;IAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAA;CAAE,CAAC,CA6F/D"}

package/dist/tools/edit.js

@@ -0,0 +1,123 @@
+import { MODELS, ASPECT_RATIOS, RESOLUTIONS } from "../types.js";
+import { saveImageFromBase64, readImageAsBase64 } from "../image-utils.js";
+export const editImageTool = {
+    name: "edit_image",
+    description: "Edit a SPECIFIC existing image file with text prompts. Supports optional reference images, model selection, aspect ratios, and resolution.",
+    inputSchema: {
+        type: "object",
+        properties: {
+            imagePath: {
+                type: "string",
+                description: "Full file path to the image to edit",
+            },
+            prompt: {
+                type: "string",
+                description: "Text describing the modifications to make",
+            },
+            referenceImages: {
+                type: "array",
+                items: { type: "string" },
+                description: "Optional file paths to reference images (up to 3 for Flash, up to 14 for Pro)",
+            },
+            model: {
+                type: "string",
+                enum: ["flash", "pro"],
+                description: "Model to use. 'flash' (default) for speed. 'pro' for quality and more reference images.",
+            },
+            aspectRatio: {
+                type: "string",
+                enum: [...ASPECT_RATIOS],
+                description: "Aspect ratio for the output image",
+            },
+            resolution: {
+                type: "string",
+                enum: [...RESOLUTIONS],
+                description: "Image resolution (Pro model only): '1K', '2K', '4K'",
+            },
+        },
+        required: ["imagePath", "prompt"],
+    },
+};
+export async function handleEditImage(request, genAI) {
+    const args = request.params.arguments;
+    let model = args.model || "flash";
+    // Auto-upgrade to Pro if many reference images or Pro-only features
+    if ((args.referenceImages && args.referenceImages.length > 3) ||
+        args.resolution === "2K" || args.resolution === "4K") {
+        model = "pro";
+    }
+    const modelId = MODELS[model];
+    // Build image parts
+    const mainImage = await readImageAsBase64(args.imagePath);
+    const parts = [
+        { inlineData: { data: mainImage.base64, mimeType: mainImage.mimeType } },
+    ];
+    // Add reference images
+    if (args.referenceImages) {
+        for (const refPath of args.referenceImages) {
+            try {
+                const refImg = await readImageAsBase64(refPath);
+                parts.push({ inlineData: { data: refImg.base64, mimeType: refImg.mimeType } });
+            }
+            catch {
+                // Skip unreadable reference images
+            }
+        }
+    }
+    parts.push({ text: args.prompt });
+    // Build config
+    const config = {
+        responseModalities: ["TEXT", "IMAGE"],
+    };
+    const imageConfig = {};
+    if (args.aspectRatio)
+        imageConfig.aspectRatio = args.aspectRatio;
+    if (args.resolution && model === "pro")
+        imageConfig.imageSize = args.resolution;
+    if (Object.keys(imageConfig).length > 0)
+        config.imageConfig = imageConfig;
+    const response = await genAI.models.generateContent({
+        model: modelId,
+        contents: [{ parts }],
+        config,
+    });
+    // Process response
+    const content = [];
+    let savedPath = null;
+    let textContent = "";
+    if (response.candidates?.[0]?.content?.parts) {
+        for (const part of response.candidates[0].content.parts) {
+            if (part.text && !part.thought)
+                textContent += part.text;
+            if (part.inlineData?.data && !part.thought) {
+                const saved = await saveImageFromBase64(part.inlineData.data, "edited");
+                savedPath = saved.filePath;
+                content.push({
+                    type: "image",
+                    data: part.inlineData.data,
+                    mimeType: part.inlineData.mimeType || "image/png",
+                });
+            }
+        }
+    }
+    let statusText = `Image edited with nano-banana (${model === "pro" ? "Gemini 3 Pro" : "Gemini 2.5 Flash"})`;
+    statusText += `\n\nOriginal: ${args.imagePath}`;
+    statusText += `\nEdit prompt: "${args.prompt}"`;
+    if (args.referenceImages?.length) {
+        statusText += `\nReference images: ${args.referenceImages.length}`;
+    }
+    if (args.aspectRatio)
+        statusText += `\nAspect ratio: ${args.aspectRatio}`;
+    if (textContent)
+        statusText += `\n\nDescription: ${textContent}`;
+    if (savedPath) {
+        statusText += `\n\nEdited image saved to: ${savedPath}`;
+        statusText += `\nUse continue_editing to make further changes.`;
+    }
+    else {
+        statusText += `\n\nNo edited image generated. Try again.`;
+    }
+    content.unshift({ type: "text", text: statusText });
+    return { result: { content }, savedPath };
+}
+//# sourceMappingURL=edit.js.map

package/dist/tools/edit.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"edit.js","sourceRoot":"","sources":["../../src/tools/edit.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,MAAM,EAAE,aAAa,EAAE,WAAW,EAAqC,MAAM,aAAa,CAAC;AACpG,OAAO,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAE3E,MAAM,CAAC,MAAM,aAAa,GAAS;IACjC,IAAI,EAAE,YAAY;IAClB,WAAW,EACT,4IAA4I;IAC9I,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,SAAS,EAAE;gBACT,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,qCAAqC;aACnD;YACD,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,2CAA2C;aACzD;YACD,eAAe,EAAE;gBACf,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACzB,WAAW,EAAE,+EAA+E;aAC7F;YACD,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC;gBACtB,WAAW,EAAE,yFAAyF;aACvG;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,GAAG,aAAa,CAAC;gBACxB,WAAW,EAAE,mCAAmC;aACjD;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,GAAG,WAAW,CAAC;gBACtB,WAAW,EAAE,qDAAqD;aACnE;SACF;QACD,QAAQ,EAAE,CAAC,WAAW,EAAE,QAAQ,CAAC;KAClC;CACF,CAAC;AAEF,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,OAAwB,EACxB,KAAkB;IAElB,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,SAO3B,CAAC;IAEF,IAAI,KAAK,GAAa,IAAI,CAAC,KAAK,IAAI,OAAO,CAAC;IAC5C,oEAAoE;IACpE,IAAI,CAAC,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC;QACzD,IAAI,CAAC,UAAU,KAAK,IAAI,IAAI,IAAI,CAAC,UAAU,KAAK,IAAI,EAAE,CAAC;QACzD,KAAK,GAAG,KAAK,CAAC;IAChB,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAE9B,oBAAoB;IACpB,MAAM,SAAS,GAAG,MAAM,iBAAiB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC1D,MAAM,KAAK,GAAU;QACnB,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,CAAC,MAAM,EAAE,QAAQ,EAAE,SAAS,CAAC,QAAQ,EAAE,EAAE;KACzE,CAAC;IAEF,uBAAuB;IACvB,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;QACzB,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YAC3C,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,OAAO,CAAC,CAAC;gBAChD,KAAK,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YACjF,CAAC;YAAC,MAAM,CAAC;gBACP,mCAAmC;YACrC,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;IAElC,eAAe;IACf,MAAM,MAAM,GAAQ;QAClB,kBAAkB,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;KACtC,CAAC;IACF,MAAM,WAAW,GAAQ,EAAE,CAAC;IAC5B,IAAI,IAAI,CAAC,WAAW;QAAE,WAAW,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;IACjE,IAAI,IAAI,CAAC,UAAU,IAAI,KAAK,KAAK,KAAK;QAAE,WAAW,CAAC,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC;IAChF,IAAI,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,GAAG,CAAC;QAAE,MAAM,CAAC,WAAW,GAAG,WAAW,CAAC;IAE1E,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,eAAe,CAAC;QAClD,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC;QACrB,MAAM;KACP,CAAC,CAAC;IAEH,mBAAmB;IACnB,MAAM,OAAO,GAAU,EAAE,CAAC;IAC1B,IAAI,SAAS,GAAkB,IAAI,CAAC;IACpC,IAAI,WAAW,GAAG,EAAE,CAAC;IAErB,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QAC7C,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACxD,IAAI,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO;gBAAE,WAAW,IAAI,IAAI,CAAC,IAAI,CAAC;YACzD,IAAI,IAAI,CAAC,UAAU,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;gBAC3C,MAAM,KAAK,GAAG,MAAM,mBAAmB,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;gBACxE,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC;gBAC3B,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,IAAI;oBAC1B,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,QAAQ,IAAI,WAAW;iBAClD,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,UAAU,GAAG,kCAAkC,KAAK,KAAK,KAAK,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,kBAAkB,GAAG,CAAC;IAC5G,UAAU,IAAI,iBAAiB,IAAI,CAAC,SAAS,EAAE,CAAC;IAChD,UAAU,IAAI,mBAAmB,IAAI,CAAC,MAAM,GAAG,CAAC;IAChD,IAAI,IAAI,CAAC,eAAe,EAAE,MAAM,EAAE,CAAC;QACjC,UAAU,IAAI,uBAAuB,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC;IACrE,CAAC;IACD,IAAI,IAAI,CAAC,WAAW;QAAE,UAAU,IAAI,mBAAmB,IAAI,CAAC,WAAW,EAAE,CAAC;IAC1E,IAAI,WAAW;QAAE,UAAU,IAAI,oBAAoB,WAAW,EAAE,CAAC;IAEjE,IAAI,SAAS,EAAE,CAAC;QACd,UAAU,IAAI,8BAA8B,SAAS,EAAE,CAAC;QACxD,UAAU,IAAI,iDAAiD,CAAC;IAClE,CAAC;SAAM,CAAC;QACN,UAAU,IAAI,2CAA2C,CAAC;IAC5D,CAAC;IAED,OAAO,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;IAEpD,OAAO,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE,EAAE,SAAS,EAAE,CAAC;AAC5C,CAAC"}

package/dist/tools/generate.d.ts

@@ -0,0 +1,9 @@
+import { CallToolRequest, CallToolResult, Tool } from "@modelcontextprotocol/sdk/types.js";
+import { GoogleGenAI } from "@google/genai";
+import { ChatSessionManager } from "../chat-session.js";
+export declare const generateImageTool: Tool;
+export declare function handleGenerateImage(request: CallToolRequest, genAI: GoogleGenAI, chatSession: ChatSessionManager): Promise<{
+    result: CallToolResult;
+    savedPath: string | null;
+}>;
+//# sourceMappingURL=generate.d.ts.map

package/dist/tools/generate.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate.d.ts","sourceRoot":"","sources":["../../src/tools/generate.ts"],"names":[],"mappings":"AACA,OAAO,EACL,eAAe,EAAE,cAAc,EAAE,IAAI,EACtC,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAG5C,OAAO,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AAExD,eAAO,MAAM,iBAAiB,EAAE,IAiC/B,CAAC;AAEF,wBAAsB,mBAAmB,CACvC,OAAO,EAAE,eAAe,EACxB,KAAK,EAAE,WAAW,EAClB,WAAW,EAAE,kBAAkB,GAC9B,OAAO,CAAC;IAAE,MAAM,EAAE,cAAc,CAAC;IAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAA;CAAE,CAAC,CAiF/D"}

package/dist/tools/generate.js

@@ -0,0 +1,106 @@
+import { MODELS, ASPECT_RATIOS, RESOLUTIONS } from "../types.js";
+import { saveImageFromBase64 } from "../image-utils.js";
+export const generateImageTool = {
+    name: "generate_image",
+    description: "Generate a NEW image from a text prompt. Supports model selection, aspect ratios, and resolution control. Use 'pro' model for higher quality, 4K resolution, and Google Search grounding.",
+    inputSchema: {
+        type: "object",
+        properties: {
+            prompt: {
+                type: "string",
+                description: "Text prompt describing the image to create",
+            },
+            model: {
+                type: "string",
+                enum: ["flash", "pro"],
+                description: "Model to use. 'flash' (default) = fast/efficient. 'pro' = higher quality, supports 4K and search grounding.",
+            },
+            aspectRatio: {
+                type: "string",
+                enum: [...ASPECT_RATIOS],
+                description: "Aspect ratio for the output image. Defaults to 1:1.",
+            },
+            resolution: {
+                type: "string",
+                enum: [...RESOLUTIONS],
+                description: "Image resolution (Pro model only). '1K' (default), '2K', or '4K'.",
+            },
+            useGoogleSearch: {
+                type: "boolean",
+                description: "Enable Google Search grounding for real-time data (Pro model only). Useful for weather, news, charts.",
+            },
+        },
+        required: ["prompt"],
+    },
+};
+export async function handleGenerateImage(request, genAI, chatSession) {
+    const args = request.params.arguments;
+    // Auto-select Pro if Pro-only features are requested
+    let model = args.model || "flash";
+    if ((args.resolution === "2K" || args.resolution === "4K" || args.useGoogleSearch) && model === "flash") {
+        model = "pro";
+    }
+    const modelId = MODELS[model];
+    // Build config
+    const config = {
+        responseModalities: ["TEXT", "IMAGE"],
+    };
+    const imageConfig = {};
+    if (args.aspectRatio)
+        imageConfig.aspectRatio = args.aspectRatio;
+    if (args.resolution && model === "pro")
+        imageConfig.imageSize = args.resolution;
+    if (Object.keys(imageConfig).length > 0)
+        config.imageConfig = imageConfig;
+    if (args.useGoogleSearch && model === "pro") {
+        config.tools = [{ googleSearch: {} }];
+    }
+    // Reset chat session (new generation = fresh context)
+    chatSession.reset();
+    const response = await genAI.models.generateContent({
+        model: modelId,
+        contents: args.prompt,
+        config,
+    });
+    // Process response
+    const content = [];
+    let savedPath = null;
+    let textContent = "";
+    if (response.candidates?.[0]?.content?.parts) {
+        for (const part of response.candidates[0].content.parts) {
+            if (part.text && !part.thought) {
+                textContent += part.text;
+            }
+            if (part.inlineData?.data && !part.thought) {
+                const saved = await saveImageFromBase64(part.inlineData.data, "generated");
+                savedPath = saved.filePath;
+                content.push({
+                    type: "image",
+                    data: part.inlineData.data,
+                    mimeType: part.inlineData.mimeType || "image/png",
+                });
+            }
+        }
+    }
+    // Build status text
+    let statusText = `Image generated with nano-banana (${model === "pro" ? "Gemini 3 Pro" : "Gemini 2.5 Flash"})`;
+    statusText += `\n\nPrompt: "${args.prompt}"`;
+    if (args.aspectRatio)
+        statusText += `\nAspect ratio: ${args.aspectRatio}`;
+    if (args.resolution && model === "pro")
+        statusText += `\nResolution: ${args.resolution}`;
+    if (args.useGoogleSearch)
+        statusText += `\nGoogle Search grounding: enabled`;
+    if (textContent)
+        statusText += `\n\nDescription: ${textContent}`;
+    if (savedPath) {
+        statusText += `\n\nImage saved to: ${savedPath}`;
+        statusText += `\n\nUse continue_editing to modify this image.`;
+    }
+    else {
+        statusText += `\n\nNo image was generated. Try running again.`;
+    }
+    content.unshift({ type: "text", text: statusText });
+    return { result: { content }, savedPath };
+}
+//# sourceMappingURL=generate.js.map

package/dist/tools/generate.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate.js","sourceRoot":"","sources":["../../src/tools/generate.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,MAAM,EAAE,aAAa,EAAE,WAAW,EAAqC,MAAM,aAAa,CAAC;AACpG,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAGxD,MAAM,CAAC,MAAM,iBAAiB,GAAS;IACrC,IAAI,EAAE,gBAAgB;IACtB,WAAW,EACT,2LAA2L;IAC7L,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,4CAA4C;aAC1D;YACD,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC;gBACtB,WAAW,EAAE,6GAA6G;aAC3H;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,GAAG,aAAa,CAAC;gBACxB,WAAW,EAAE,qDAAqD;aACnE;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,GAAG,WAAW,CAAC;gBACtB,WAAW,EAAE,mEAAmE;aACjF;YACD,eAAe,EAAE;gBACf,IAAI,EAAE,SAAS;gBACf,WAAW,EAAE,uGAAuG;aACrH;SACF;QACD,QAAQ,EAAE,CAAC,QAAQ,CAAC;KACrB;CACF,CAAC;AAEF,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,OAAwB,EACxB,KAAkB,EAClB,WAA+B;IAE/B,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,SAM3B,CAAC;IAEF,qDAAqD;IACrD,IAAI,KAAK,GAAa,IAAI,CAAC,KAAK,IAAI,OAAO,CAAC;IAC5C,IAAI,CAAC,IAAI,CAAC,UAAU,KAAK,IAAI,IAAI,IAAI,CAAC,UAAU,KAAK,IAAI,IAAI,IAAI,CAAC,eAAe,CAAC,IAAI,KAAK,KAAK,OAAO,EAAE,CAAC;QACxG,KAAK,GAAG,KAAK,CAAC;IAChB,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAE9B,eAAe;IACf,MAAM,MAAM,GAAQ;QAClB,kBAAkB,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;KACtC,CAAC;IAEF,MAAM,WAAW,GAAQ,EAAE,CAAC;IAC5B,IAAI,IAAI,CAAC,WAAW;QAAE,WAAW,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;IACjE,IAAI,IAAI,CAAC,UAAU,IAAI,KAAK,KAAK,KAAK;QAAE,WAAW,CAAC,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC;IAChF,IAAI,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,GAAG,CAAC;QAAE,MAAM,CAAC,WAAW,GAAG,WAAW,CAAC;IAE1E,IAAI,IAAI,CAAC,eAAe,IAAI,KAAK,KAAK,KAAK,EAAE,CAAC;QAC5C,MAAM,CAAC,KAAK,GAAG,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;IACxC,CAAC;IAED,sDAAsD;IACtD,WAAW,CAAC,KAAK,EAAE,CAAC;IAEpB,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,eAAe,CAAC;QAClD,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,IAAI,CAAC,MAAM;QACrB,MAAM;KACP,CAAC,CAAC;IAEH,mBAAmB;IACnB,MAAM,OAAO,GAAU,EAAE,CAAC;IAC1B,IAAI,SAAS,GAAkB,IAAI,CAAC;IACpC,IAAI,WAAW,GAAG,EAAE,CAAC;IAErB,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QAC7C,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACxD,IAAI,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;gBAC/B,WAAW,IAAI,IAAI,CAAC,IAAI,CAAC;YAC3B,CAAC;YACD,IAAI,IAAI,CAAC,UAAU,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;gBAC3C,MAAM,KAAK,GAAG,MAAM,mBAAmB,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;gBAC3E,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC;gBAE3B,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,IAAI;oBAC1B,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,QAAQ,IAAI,WAAW;iBAClD,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,oBAAoB;IACpB,IAAI,UAAU,GAAG,qCAAqC,KAAK,KAAK,KAAK,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,kBAAkB,GAAG,CAAC;IAC/G,UAAU,IAAI,gBAAgB,IAAI,CAAC,MAAM,GAAG,CAAC;IAC7C,IAAI,IAAI,CAAC,WAAW;QAAE,UAAU,IAAI,mBAAmB,IAAI,CAAC,WAAW,EAAE,CAAC;IAC1E,IAAI,IAAI,CAAC,UAAU,IAAI,KAAK,KAAK,KAAK;QAAE,UAAU,IAAI,iBAAiB,IAAI,CAAC,UAAU,EAAE,CAAC;IACzF,IAAI,IAAI,CAAC,eAAe;QAAE,UAAU,IAAI,oCAAoC,CAAC;IAC7E,IAAI,WAAW;QAAE,UAAU,IAAI,oBAAoB,WAAW,EAAE,CAAC;IAEjE,IAAI,SAAS,EAAE,CAAC;QACd,UAAU,IAAI,uBAAuB,SAAS,EAAE,CAAC;QACjD,UAAU,IAAI,gDAAgD,CAAC;IACjE,CAAC;SAAM,CAAC;QACN,UAAU,IAAI,gDAAgD,CAAC;IACjE,CAAC;IAED,OAAO,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;IAEpD,OAAO,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE,EAAE,SAAS,EAAE,CAAC;AAC5C,CAAC"}

package/dist/tools/info.d.ts

@@ -0,0 +1,4 @@
+import { CallToolResult, Tool } from "@modelcontextprotocol/sdk/types.js";
+export declare const getLastImageInfoTool: Tool;
+export declare function handleGetLastImageInfo(lastImagePath: string | null): Promise<CallToolResult>;
+//# sourceMappingURL=info.d.ts.map

package/dist/tools/info.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"info.d.ts","sourceRoot":"","sources":["../../src/tools/info.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,oCAAoC,CAAC;AAG1E,eAAO,MAAM,oBAAoB,EAAE,IAQlC,CAAC;AAEF,wBAAsB,sBAAsB,CAC1C,aAAa,EAAE,MAAM,GAAG,IAAI,GAC3B,OAAO,CAAC,cAAc,CAAC,CA0BzB"}

package/dist/tools/info.js

@@ -0,0 +1,38 @@
+import fs from "fs/promises";
+export const getLastImageInfoTool = {
+    name: "get_last_image_info",
+    description: "Get information about the last generated/edited image (file path, size, etc.)",
+    inputSchema: {
+        type: "object",
+        properties: {},
+        additionalProperties: false,
+    },
+};
+export async function handleGetLastImageInfo(lastImagePath) {
+    if (!lastImagePath) {
+        return {
+            content: [{
+                    type: "text",
+                    text: "No previous image found. Use generate_image or edit_image first.",
+                }],
+        };
+    }
+    try {
+        const stats = await fs.stat(lastImagePath);
+        return {
+            content: [{
+                    type: "text",
+                    text: `Last Image:\n\nPath: ${lastImagePath}\nSize: ${Math.round(stats.size / 1024)} KB\nModified: ${stats.mtime.toLocaleString()}\n\nUse continue_editing to make changes.`,
+                }],
+        };
+    }
+    catch {
+        return {
+            content: [{
+                    type: "text",
+                    text: `Last Image:\n\nPath: ${lastImagePath}\nStatus: File not found (may have been moved or deleted)\n\nUse generate_image to create a new image.`,
+                }],
+        };
+    }
+}
+//# sourceMappingURL=info.js.map

package/dist/tools/info.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"info.js","sourceRoot":"","sources":["../../src/tools/info.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,MAAM,aAAa,CAAC;AAE7B,MAAM,CAAC,MAAM,oBAAoB,GAAS;IACxC,IAAI,EAAE,qBAAqB;IAC3B,WAAW,EAAE,+EAA+E;IAC5F,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE,EAAE;QACd,oBAAoB,EAAE,KAAK;KAC5B;CACF,CAAC;AAEF,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,aAA4B;IAE5B,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,OAAO;YACL,OAAO,EAAE,CAAC;oBACR,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,kEAAkE;iBACzE,CAAC;SACH,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC3C,OAAO;YACL,OAAO,EAAE,CAAC;oBACR,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,wBAAwB,aAAa,WAAW,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,kBAAkB,KAAK,CAAC,KAAK,CAAC,cAAc,EAAE,2CAA2C;iBAC7K,CAAC;SACH,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO;YACL,OAAO,EAAE,CAAC;oBACR,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,wBAAwB,aAAa,wGAAwG;iBACpJ,CAAC;SACH,CAAC;IACJ,CAAC;AACH,CAAC"}