diff --git a/comfy_api/v3/ui.py b/comfy_api/v3/ui.py index 644bc4ca3..4094812e0 100644 --- a/comfy_api/v3/ui.py +++ b/comfy_api/v3/ui.py @@ -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): diff --git a/comfy_extras/v3/nodes_load_3d.py b/comfy_extras/v3/nodes_load_3d.py index af12b497a..0068da40f 100644 --- a/comfy_extras/v3/nodes_load_3d.py +++ b/comfy_extras/v3/nodes_load_3d.py @@ -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 = [