mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-07-27 16:26:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #8883 from bigcat88/v3/io/uploadtype

Browse Source 
[V3] make generic upload parameters for io.Combo.Input
This commit is contained in:
Jedrzej Kosinski 2025-07-13 22:11:43 -07:00 committed by GitHub
commit 8f7e27352e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
comfy_api/v3/io.py
View File

@ -28,6 +28,14 @@ class FolderType(str, Enum):
output = "output" output = "output"
temp = "temp" temp = "temp"
class UploadType(str, Enum):
image = "image_upload"
audio = "audio_upload"
video = "video_upload"
model = "file_upload"
class RemoteOptions: class RemoteOptions:
def __init__(self, route: str, refresh_button: bool, control_after_refresh: Literal["first", "last"]="first", def __init__(self, route: str, refresh_button: bool, control_after_refresh: Literal["first", "last"]="first",
timeout: int=None, max_retries: int=None, refresh: int=None): timeout: int=None, max_retries: int=None, refresh: int=None):
@ -394,16 +402,15 @@ class Combo(ComfyType):
Type = str Type = str
def __init__(self, id: str, options: list[str]=None, display_name: str=None, optional=False, tooltip: str=None, lazy: bool=None, def __init__(self, id: str, options: list[str]=None, display_name: str=None, optional=False, tooltip: str=None, lazy: bool=None,
default: str=None, control_after_generate: bool=None, default: str=None, control_after_generate: bool=None,
image_upload: bool=None, image_folder: FolderType=None, content_types: list[Literal["image", "video", "audio", "model"]]=None, upload: UploadType=None, image_folder: FolderType=None,
remote: RemoteOptions=None, remote: RemoteOptions=None,
socketless: bool=None): socketless: bool=None):
super().__init__(id, display_name, optional, tooltip, lazy, default, socketless) super().__init__(id, display_name, optional, tooltip, lazy, default, socketless)
self.multiselect = False self.multiselect = False
self.options = options self.options = options
self.control_after_generate = control_after_generate self.control_after_generate = control_after_generate
self.image_upload = image_upload self.upload = upload
self.image_folder = image_folder self.image_folder = image_folder
self.content_types = content_types
self.remote = remote self.remote = remote
self.default: str self.default: str
@ -412,9 +419,8 @@ class Combo(ComfyType):
"multiselect": self.multiselect, "multiselect": self.multiselect,
"options": self.options, "options": self.options,
"control_after_generate": self.control_after_generate, "control_after_generate": self.control_after_generate,
"image_upload": self.image_upload, **({self.upload.value: True} if self.upload is not None else {}),
"image_folder": self.image_folder.value if self.image_folder else None, "image_folder": self.image_folder.value if self.image_folder else None,
"content_types": self.content_types if self.content_types else None,
"remote": self.remote.as_dict() if self.remote else None, "remote": self.remote.as_dict() if self.remote else None,
}) })

6
comfy_extras/v3/nodes_images.py
View File

@ -100,9 +100,8 @@ class LoadImage_V3(io.ComfyNodeV3):
inputs=[ inputs=[
io.Combo.Input( io.Combo.Input(
"image", "image",
image_upload=True, upload=io.UploadType.image,
image_folder=io.FolderType.input, image_folder=io.FolderType.input,
content_types=["image"],
options=cls.get_files_options(), options=cls.get_files_options(),
), ),
], ],
@ -192,9 +191,8 @@ class LoadImageOutput_V3(io.ComfyNodeV3):
inputs=[ inputs=[
io.Combo.Input( io.Combo.Input(
"image", "image",
image_upload=True, upload=io.UploadType.image,
image_folder=io.FolderType.output, image_folder=io.FolderType.output,
content_types=["image"],
remote=io.RemoteOptions( remote=io.RemoteOptions(
route="/internal/files/output", route="/internal/files/output",
refresh_button=True, refresh_button=True,