import { z, IntegrationDefinition } from '@botpress/sdk'

const sizeDescription =
  'Size of the image to generate. Can be "1024x1024", "1792x1024", or "1024x1792". Defaults to 1024x1024.'
const qualityDescription = 'Quality of the image to generate. Can be "standard" or "hd". Defaults to "standard".'
const modelDescription = 'Model to use for image generation. Defaults to "dall-e-3".'

export default new IntegrationDefinition({
  name: 'dalle',
  version: '0.3.8',
  icon: 'icon.svg',
  title: 'DALL-E (Deprecated)',
  description: 'Integrate DALL-E to generate images directly within your chatbot conversations.',
  readme: 'hub.md',
  configuration: {
    schema: z.object({
      apiKey: z.string().title('OpenAI API Key').describe('An API Key for the OpenAI DALL-E service'),
    }),
  },
  channels: {},
  actions: {
    generateImage: {
      title: 'generate Image',
      description: 'Generate an image',
      input: {
        schema: z.object({
          prompt: z.string().describe('Prompt for image generation.').title('Prompt'),
          size: z.string().optional().describe(sizeDescription).title('Size'),
          quality: z.string().optional().describe(qualityDescription).title('Quality'),
          model: z.string().optional().describe(modelDescription).title('Model'),
          user: z.string().optional().describe('User ID').title('User'),
        }),
      },
      output: {
        schema: z.object({
          url: z.string().url().describe('The url of the generated image').title('Url'),
          createdDate: z.string().describe('The date the image was created').title('Created Date'),
        }),
      },
    },
  },
  attributes: {
    category: 'AI Models',
    repo: 'botpress',
  },
})