import type { ToolDefinition } from "gui-chat-protocol";

export const TOOL_NAME = "generateImage";
// Endpoint contract is shared with editImages (see
// `editImages/definition.ts#ImageEndpoints`).

export interface ImageToolData {
  imageData: string;
  prompt?: string;
}

export interface GenerateImageArgs {
  prompt: string;
}

const toolDefinition: ToolDefinition = {
  type: "function",
  name: TOOL_NAME,
  description:
    "Generate an image based on the prompt and display it on the screen. Be descriptive and specify the concrete details of the images in the prompt. Each call generates one image.",
  prompt: `When the user asks you to generate, draw, or create an image, use the ${TOOL_NAME} API. You may also offer to generate an image when a visual would clearly enhance the conversation, but do not generate images unsolicited during casual discussion.`,
  parameters: {
    type: "object",
    properties: {
      prompt: {
        type: "string",
        description: "A detailed prompt describing the image to generate",
      },
    },
    required: ["prompt"],
  },
};

export default toolDefinition;