import { ComfyNodeDef } from '@/types/apiTypes'

import { app } from '../../scripts/app'

// Adds an upload button to the nodes

app.registerExtension({
  name: 'Comfy.UploadImage',
  beforeRegisterNodeDef(nodeType, nodeData: ComfyNodeDef) {
    if (nodeData?.input?.required?.image?.[1]?.image_upload === true) {
      nodeData.input.required.upload = ['IMAGEUPLOAD']
    }
  }
})