minor changes

This commit is contained in:
Aryan185
2026-01-02 11:52:05 +00:00
parent a512cb62be
commit 661a30f3bf
+6 -6
View File
@@ -6,13 +6,13 @@ import av
from PIL import Image
from openai import OpenAI
class Sora:
class SoraGen:
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"prompt": ("STRING", {"multiline": True, "default": "A calico cat playing a piano on stage"}),
"openai_api_key": ("STRING", {"multiline": False, "default": ""}),
"api_key": ("STRING", {"multiline": False, "default": ""}),
"model": (["sora-2", "sora-2-pro"], {"default": "sora-2"}),
"size": (["720x1280", "1280x720", "1024x1792", "1792x1024"], {"default": "1280x720"}),
"duration": (["4", "8", "12"], {"default": "4"}),
@@ -23,8 +23,8 @@ class Sora:
RETURN_TYPES = ("IMAGE", "AUDIO")
RETURN_NAMES = ("frames", "audio")
FUNCTION = "gen_video"
CATEGORY = "OpenAI/Video"
FUNCTION = "generate_video"
CATEGORY = "video/generation"
OUTPUT_IS_LIST = (True, False)
def gen_video(self, prompt, openai_api_key, model, size, duration, seed, input_image=None):
@@ -83,5 +83,5 @@ class Sora:
if img_buf:
img_buf.close()
NODE_CLASS_MAPPINGS = {"Sora": Sora}
NODE_DISPLAY_NAME_MAPPINGS = {"Sora": "Sora 2 (OpenAI)"}
NODE_CLASS_MAPPINGS = {"SoraGen": SoraGen}
NODE_DISPLAY_NAME_MAPPINGS = {"SoraGen": "Sora 2 (OpenAI)"}