Comfy-Org/ComfyUI_frontend
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI_frontend.git synced 2026-02-07 00:20:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add support for image_folder field in node image input spec (#2509)

Browse Source
This commit is contained in:
bymyself
2025-02-10 22:12:03 -07:00
committed by GitHub
parent a93f57eeb2
commit e4f8d4b8d0
3 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/extensions/core/uploadImage.ts
View File

@@ -1,4 +1,4 @@
import { ComfyNodeDef } from '@/types/apiTypes'
import { ComfyNodeDef, InputSpec } from '@/types/apiTypes'
import { app } from '../../scripts/app'
@@ -7,8 +7,16 @@ import { app } from '../../scripts/app'
app.registerExtension({
name: 'Comfy.UploadImage',
beforeRegisterNodeDef(nodeType, nodeData: ComfyNodeDef) {
if (nodeData?.input?.required?.image?.[1]?.image_upload === true) {
nodeData.input.required.upload = ['IMAGEUPLOAD']
// Check if there is a required input named 'image' in the nodeData
const imageInputSpec: InputSpec | undefined =
nodeData?.input?.required?.image
// Get the config from the image input spec if it exists
const config = imageInputSpec?.[1] ?? {}
const { image_upload = false, image_folder = 'input' } = config
if (image_upload && nodeData?.input?.required) {
nodeData.input.required.upload = ['IMAGEUPLOAD', { image_folder }]
}
}
})