Skip to content

Commit 4eba316

Browse files
authored
Refactor Pika API node imports and fix unique_id issue. (#8319)
Added unique_id to hidden parameters and corrected description formatting in PikAdditionsNode.
1 parent 592d056 commit 4eba316

File tree

1 file changed

+24
-21
lines changed

1 file changed

+24
-21
lines changed

comfy_api_nodes/nodes_pika.py

Lines changed: 24 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -6,40 +6,42 @@
66
from __future__ import annotations
77

88
import io
9-
from typing import Optional, TypeVar
109
import logging
11-
import torch
10+
from typing import Optional, TypeVar
11+
1212
import numpy as np
13+
import torch
14+
15+
from comfy.comfy_types.node_typing import IO, ComfyNodeABC, InputTypeOptions
16+
from comfy_api.input_impl import VideoFromFile
17+
from comfy_api.input_impl.video_types import VideoCodec, VideoContainer, VideoInput
18+
from comfy_api_nodes.apinode_utils import (
19+
download_url_to_video_output,
20+
tensor_to_bytesio,
21+
)
1322
from comfy_api_nodes.apis import (
14-
PikaBodyGenerate22T2vGenerate22T2vPost,
15-
PikaGenerateResponse,
16-
PikaBodyGenerate22I2vGenerate22I2vPost,
17-
PikaVideoResponse,
18-
PikaBodyGenerate22C2vGenerate22PikascenesPost,
1923
IngredientsMode,
20-
PikaDurationEnum,
21-
PikaResolutionEnum,
22-
PikaBodyGeneratePikaffectsGeneratePikaffectsPost,
24+
PikaBodyGenerate22C2vGenerate22PikascenesPost,
25+
PikaBodyGenerate22I2vGenerate22I2vPost,
26+
PikaBodyGenerate22KeyframeGenerate22PikaframesPost,
27+
PikaBodyGenerate22T2vGenerate22T2vPost,
2328
PikaBodyGeneratePikadditionsGeneratePikadditionsPost,
29+
PikaBodyGeneratePikaffectsGeneratePikaffectsPost,
2430
PikaBodyGeneratePikaswapsGeneratePikaswapsPost,
25-
PikaBodyGenerate22KeyframeGenerate22PikaframesPost,
31+
PikaDurationEnum,
2632
Pikaffect,
33+
PikaGenerateResponse,
34+
PikaResolutionEnum,
35+
PikaVideoResponse,
2736
)
2837
from comfy_api_nodes.apis.client import (
2938
ApiEndpoint,
39+
EmptyRequest,
3040
HttpMethod,
31-
SynchronousOperation,
3241
PollingOperation,
33-
EmptyRequest,
34-
)
35-
from comfy_api_nodes.apinode_utils import (
36-
tensor_to_bytesio,
37-
download_url_to_video_output,
42+
SynchronousOperation,
3843
)
3944
from comfy_api_nodes.mapper_utils import model_field_to_node_input
40-
from comfy_api.input_impl.video_types import VideoInput, VideoContainer, VideoCodec
41-
from comfy_api.input_impl import VideoFromFile
42-
from comfy.comfy_types.node_typing import IO, ComfyNodeABC, InputTypeOptions
4345

4446
R = TypeVar("R")
4547

@@ -204,6 +206,7 @@ def INPUT_TYPES(cls):
204206
"hidden": {
205207
"auth_token": "AUTH_TOKEN_COMFY_ORG",
206208
"comfy_api_key": "API_KEY_COMFY_ORG",
209+
"unique_id": "UNIQUE_ID",
207210
},
208211
}
209212

@@ -457,7 +460,7 @@ def INPUT_TYPES(cls):
457460
},
458461
}
459462

460-
DESCRIPTION = "Add any object or image into your video. Upload a video and specify what youd like to add to create a seamlessly integrated result."
463+
DESCRIPTION = "Add any object or image into your video. Upload a video and specify what you'd like to add to create a seamlessly integrated result."
461464

462465
def api_call(
463466
self,

0 commit comments

Comments
 (0)