Sirshendu Ganguly
f85b8f809d
- Convert useSteps, useCFGScale, useScheduler, useSeed to BOOLEAN in ImageInference - Convert useFps to BOOLEAN in VideoInference - Reorder parameters so 'use' toggles appear above their corresponding parameters - Fix VideoInference providerSettings input type from DICT to RUNWAREPROVIDERSETTINGS - Update provider settings handling logic to match ImageInference format - Improve UX with native boolean toggles and logical parameter ordering
81 lines
3.3 KiB
Python
81 lines
3.3 KiB
Python
from .utils import runwareUtils as rwUtils
|
|
|
|
|
|
class RunwarePixverseProviderSettings:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"effect": (["none", "boom drop", "zoom in", "zoom out", "pan left", "pan right", "tilt up", "tilt down"], {
|
|
"default": "none",
|
|
"tooltip": "Camera effect for Pixverse videos.",
|
|
}),
|
|
"cameraMovement": (["none", "static", "dynamic", "smooth"], {
|
|
"default": "none",
|
|
"tooltip": "Camera movement style for Pixverse videos.",
|
|
}),
|
|
"style": (["anime", "realistic", "cinematic", "artistic", "vintage", "modern"], {
|
|
"default": "realistic",
|
|
"tooltip": "Visual style for Pixverse videos.",
|
|
}),
|
|
"motionMode": (["normal", "slow", "fast", "smooth", "dynamic"], {
|
|
"default": "normal",
|
|
"tooltip": "Motion mode for Pixverse videos.",
|
|
}),
|
|
"soundEffectSwitch": ("BOOLEAN", {
|
|
"default": False,
|
|
"tooltip": "Enable or disable sound effects for Pixverse videos.",
|
|
"label_on": "Enabled",
|
|
"label_off": "Disabled",
|
|
}),
|
|
"soundEffectContent": ("STRING", {
|
|
"multiline": True,
|
|
"placeholder": "Sound effect description or content",
|
|
"tooltip": "Sound effect content for Pixverse videos.",
|
|
}),
|
|
}
|
|
}
|
|
|
|
DESCRIPTION = "Configure Pixverse-specific settings for video generation with effect, style, motion, and sound options."
|
|
FUNCTION = "create_pixverse_settings"
|
|
RETURN_TYPES = ("RUNWAREPROVIDERSETTINGS",)
|
|
RETURN_NAMES = ("Provider Settings",)
|
|
CATEGORY = "Runware"
|
|
|
|
def create_pixverse_settings(self, **kwargs):
|
|
effect = kwargs.get("effect", "none")
|
|
cameraMovement = kwargs.get("cameraMovement", "none")
|
|
style = kwargs.get("style", "realistic")
|
|
motionMode = kwargs.get("motionMode", "normal")
|
|
soundEffectSwitch = kwargs.get("soundEffectSwitch", False)
|
|
soundEffectContent = kwargs.get("soundEffectContent", "")
|
|
|
|
# Create Pixverse settings dictionary
|
|
pixverse_settings = {
|
|
"effect": effect,
|
|
"cameraMovement": cameraMovement,
|
|
"style": style,
|
|
"motionMode": motionMode,
|
|
"soundEffectSwitch": soundEffectSwitch,
|
|
}
|
|
|
|
# Only add soundEffectContent if it's not empty
|
|
if soundEffectContent and soundEffectContent.strip():
|
|
pixverse_settings["soundEffectContent"] = soundEffectContent
|
|
|
|
# Remove "none" values to keep the settings clean
|
|
pixverse_settings = {k: v for k, v in pixverse_settings.items() if v != "none" and v is not None}
|
|
|
|
print(f"[Pixverse Provider Settings] Created settings: {pixverse_settings}")
|
|
return (pixverse_settings,)
|
|
|
|
|
|
# Node class mappings for ComfyUI registration
|
|
NODE_CLASS_MAPPINGS = {
|
|
"RunwarePixverseProviderSettings": RunwarePixverseProviderSettings,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"RunwarePixverseProviderSettings": "Runware Pixverse Provider Settings",
|
|
}
|