Sirshendu Ganguly
056ade0661
- Use librosa instead of torchaudio for mediaStorage - Created dedicated toggle handlers for Image Inference, Video Inference, Upscaler, Audio Inference, Accelerator Options, Bytedance Provider Settings, and OpenAI Provider Settings - Fixed Audio Inference widget ordering (use parameters now appear before their corresponding parameters) - All handlers use the same proven pattern from videoInferenceDimensionsHandler - Improved widget disabling/enabling functionality across all nodes
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
from typing import List
|
|
from .utils import runwareUtils as rwUtils
|
|
|
|
|
|
class referenceVideos:
|
|
"""Reference Videos node for configuring reference video inputs"""
|
|
|
|
MAX_VIDEOS = 4
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
optionalInputs = {}
|
|
|
|
for i in range(1, cls.MAX_VIDEOS + 1):
|
|
ordinal = rwUtils.getOrdinal(i)
|
|
optionalInputs[f"Video{i}"] = ("STRING", {
|
|
"tooltip": f"MediaUUID for the {ordinal} reference video from Runware Media Upload node.",
|
|
})
|
|
|
|
return {
|
|
"required": {},
|
|
"optional": optionalInputs
|
|
}
|
|
|
|
DESCRIPTION = "Configure multiple reference video inputs (mediaUUIDs) for video generation."
|
|
FUNCTION = "createReferenceVideos"
|
|
RETURN_TYPES = ("RUNWAREREFERENCEVIDEOS",)
|
|
RETURN_NAMES = ("Reference Videos",)
|
|
CATEGORY = "Runware"
|
|
|
|
def createReferenceVideos(self, **kwargs) -> tuple[List[str]]:
|
|
"""Create list of reference video UUIDs"""
|
|
videoList = []
|
|
|
|
for i in range(1, self.MAX_VIDEOS + 1):
|
|
videoKey = f"Video{i}"
|
|
videoUuid = kwargs.get(videoKey)
|
|
|
|
if videoUuid and videoUuid.strip() != "":
|
|
videoList.append(videoUuid.strip())
|
|
|
|
return (videoList,)
|