adjusted input parameters of ui.PreviewUI3D

This commit is contained in:
bigcat88 2025-07-24 22:10:35 +03:00
parent f569823738
commit c3d9243915
No known key found for this signature in database
GPG Key ID: 1F0BF0EC3CF22721
2 changed files with 6 additions and 5 deletions

View File

@ -475,11 +475,12 @@ class PreviewVideo(_UIOutput):
class PreviewUI3D(_UIOutput):
def __init__(self, values: list[SavedResult | dict], **kwargs):
self.values = values
def __init__(self, model_file, camera_info, **kwargs):
self.model_file = model_file
self.camera_info = camera_info
def as_dict(self):
return {"result": self.values}
return {"result": [self.model_file, self.camera_info]}
class PreviewText(_UIOutput):

View File

@ -148,7 +148,7 @@ class Preview3D(io.ComfyNode):
@classmethod
def execute(cls, model_file, camera_info=None):
return io.NodeOutput(ui=ui.PreviewUI3D([model_file, camera_info], cls=cls))
return io.NodeOutput(ui=ui.PreviewUI3D(model_file, camera_info, cls=cls))
class Preview3DAnimation(io.ComfyNode):
@ -169,7 +169,7 @@ class Preview3DAnimation(io.ComfyNode):
@classmethod
def execute(cls, model_file, camera_info=None):
return io.NodeOutput(ui=ui.PreviewUI3D([model_file, camera_info], cls=cls))
return io.NodeOutput(ui=ui.PreviewUI3D(model_file, camera_info, cls=cls))
NODES_LIST = [