mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-01-31 13:29:55 +00:00
14 lines
381 B
TypeScript
14 lines
381 B
TypeScript
import { app } from '../../scripts/app'
|
|
import { ComfyNodeDef } from '@/types/apiTypes'
|
|
|
|
// 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']
|
|
}
|
|
}
|
|
})
|