|
6 | 6 | from __future__ import annotations |
7 | 7 |
|
8 | 8 | import io |
9 | | -from typing import Optional, TypeVar |
10 | 9 | import logging |
11 | | -import torch |
| 10 | +from typing import Optional, TypeVar |
| 11 | + |
12 | 12 | 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 | +) |
13 | 22 | from comfy_api_nodes.apis import ( |
14 | | - PikaBodyGenerate22T2vGenerate22T2vPost, |
15 | | - PikaGenerateResponse, |
16 | | - PikaBodyGenerate22I2vGenerate22I2vPost, |
17 | | - PikaVideoResponse, |
18 | | - PikaBodyGenerate22C2vGenerate22PikascenesPost, |
19 | 23 | IngredientsMode, |
20 | | - PikaDurationEnum, |
21 | | - PikaResolutionEnum, |
22 | | - PikaBodyGeneratePikaffectsGeneratePikaffectsPost, |
| 24 | + PikaBodyGenerate22C2vGenerate22PikascenesPost, |
| 25 | + PikaBodyGenerate22I2vGenerate22I2vPost, |
| 26 | + PikaBodyGenerate22KeyframeGenerate22PikaframesPost, |
| 27 | + PikaBodyGenerate22T2vGenerate22T2vPost, |
23 | 28 | PikaBodyGeneratePikadditionsGeneratePikadditionsPost, |
| 29 | + PikaBodyGeneratePikaffectsGeneratePikaffectsPost, |
24 | 30 | PikaBodyGeneratePikaswapsGeneratePikaswapsPost, |
25 | | - PikaBodyGenerate22KeyframeGenerate22PikaframesPost, |
| 31 | + PikaDurationEnum, |
26 | 32 | Pikaffect, |
| 33 | + PikaGenerateResponse, |
| 34 | + PikaResolutionEnum, |
| 35 | + PikaVideoResponse, |
27 | 36 | ) |
28 | 37 | from comfy_api_nodes.apis.client import ( |
29 | 38 | ApiEndpoint, |
| 39 | + EmptyRequest, |
30 | 40 | HttpMethod, |
31 | | - SynchronousOperation, |
32 | 41 | PollingOperation, |
33 | | - EmptyRequest, |
34 | | -) |
35 | | -from comfy_api_nodes.apinode_utils import ( |
36 | | - tensor_to_bytesio, |
37 | | - download_url_to_video_output, |
| 42 | + SynchronousOperation, |
38 | 43 | ) |
39 | 44 | 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 |
43 | 45 |
|
44 | 46 | R = TypeVar("R") |
45 | 47 |
|
@@ -204,6 +206,7 @@ def INPUT_TYPES(cls): |
204 | 206 | "hidden": { |
205 | 207 | "auth_token": "AUTH_TOKEN_COMFY_ORG", |
206 | 208 | "comfy_api_key": "API_KEY_COMFY_ORG", |
| 209 | + "unique_id": "UNIQUE_ID", |
207 | 210 | }, |
208 | 211 | } |
209 | 212 |
|
@@ -457,7 +460,7 @@ def INPUT_TYPES(cls): |
457 | 460 | }, |
458 | 461 | } |
459 | 462 |
|
460 | | - 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." |
| 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." |
461 | 464 |
|
462 | 465 | def api_call( |
463 | 466 | self, |
|
0 commit comments