69 lines
2.5 KiB
Python
69 lines
2.5 KiB
Python
|
|
|
|
class RunwareOpenAIProviderSettings:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"quality": (["auto", "high", "medium", "low"], {
|
|
"default": "auto",
|
|
"tooltip": "Quality setting for OpenAI DALL-E models.",
|
|
}),
|
|
"useBackground": (["enable", "disabled"], {
|
|
"default": "disabled",
|
|
"tooltip": "Enable to include background parameter in API request.",
|
|
}),
|
|
"background": (["transparent", "opaque"], {
|
|
"default": "opaque",
|
|
"tooltip": "Background setting for OpenAI image generation. Only used when 'Use Background' is enabled.",
|
|
}),
|
|
"useStyle": (["enable", "disabled"], {
|
|
"default": "disabled",
|
|
"tooltip": "Enable to include style parameter in API request.",
|
|
}),
|
|
"style": (["vivid", "natural"], {
|
|
"default": "vivid",
|
|
"tooltip": "Style setting for OpenAI DALL-E models. Only used when 'Use Style' is enabled.",
|
|
}),
|
|
}
|
|
}
|
|
|
|
DESCRIPTION = "Configure OpenAI-specific settings for image generation with quality, background, and style options."
|
|
FUNCTION = "create_openai_settings"
|
|
RETURN_TYPES = ("RUNWAREPROVIDERSETTINGS",)
|
|
RETURN_NAMES = ("Provider Settings",)
|
|
CATEGORY = "Runware"
|
|
|
|
def create_openai_settings(self, **kwargs):
|
|
quality = kwargs.get("quality", "auto")
|
|
useBackground = kwargs.get("useBackground", "disabled")
|
|
background = kwargs.get("background", "opaque")
|
|
useStyle = kwargs.get("useStyle", "disabled")
|
|
style = kwargs.get("style", "vivid")
|
|
|
|
# Create OpenAI settings dictionary
|
|
openai_settings = {
|
|
"quality": quality,
|
|
}
|
|
|
|
# Add background only if enabled
|
|
if useBackground == "enable":
|
|
openai_settings["background"] = background
|
|
|
|
# Add style only if enabled
|
|
if useStyle == "enable":
|
|
openai_settings["style"] = style
|
|
|
|
print(f"[OpenAI Provider Settings] Created settings: {openai_settings}")
|
|
return (openai_settings,)
|
|
|
|
|
|
# Node class mappings for ComfyUI registration
|
|
NODE_CLASS_MAPPINGS = {
|
|
"RunwareOpenAIProviderSettings": RunwareOpenAIProviderSettings,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"RunwareOpenAIProviderSettings": "Runware OpenAI Provider Settings",
|
|
}
|