mulmocast 0.1.7 → 1.1.0

package/assets/templates/akira_comic.json

@@ -4,7 +4,7 @@
   "systemPrompt": "Generate a script for a presentation of the given topic. Another AI will generate images for each beat based on the image prompt of that beat. Mention the reference in one of beats, if it exists. Use the JSON below as a template.",
   "presentationStyle": {
     "$mulmocast": {
-      "version": "1.0",
+      "version": "1.1",
       "credit": "closing"
     },
     "canvasSize": {

package/assets/templates/ani.json

@@ -4,7 +4,7 @@
   "systemPrompt": "Generate a script for a presentation of the given topic. 言葉づかいは少しツンデレにして。Another AI will generate comic for each beat based on the image prompt of that beat. You don't need to specify the style of the image, just describe the scene. Mention the reference in one of beats, if it exists. Use the JSON below as a template. Create appropriate amount of beats, and make sure the beats are coherent and flow well.",
   "presentationStyle": {
     "$mulmocast": {
-      "version": "1.0",
+      "version": "1.1",
       "credit": "closing"
     },
     "movieParams": {

package/assets/templates/ani_ja.json

@@ -4,7 +4,7 @@
   "systemPrompt": "Generate a Japanese script for a presentation of the given topic. 言葉づかいは少しツンデレにして。Another AI will generate comic for each beat based on the image prompt of that beat. You don't need to specify the style of the image, just describe the scene. Mention the reference in one of beats, if it exists. Use the JSON below as a template. Create appropriate amount of beats, and make sure the beats are coherent and flow well.",
   "presentationStyle": {
     "$mulmocast": {
-      "version": "1.0",
+      "version": "1.1",
       "credit": "closing"
     },
     "movieParams": {
@@ -23,9 +23,8 @@
       "height": 1536
     },
     "speechParams": {
-      "provider": "nijivoice",
       "speakers": {
-        "Presenter": { "voiceId": "9d9ed276-49ee-443a-bc19-26e6136d05f0" }
+        "Presenter": { "provider": "nijivoice", "voiceId": "9d9ed276-49ee-443a-bc19-26e6136d05f0" }
       }
     },
     "imageParams": {

package/assets/templates/characters.json

@@ -3,7 +3,7 @@
   "description": "Template for story with multiple characters.",
   "presentationStyle": {
     "$mulmocast": {
-      "version": "1.0",
+      "version": "1.1",
       "credit": "closing"
     },
     "canvasSize": {

package/assets/templates/children_book.json

@@ -4,7 +4,7 @@
   "systemPrompt": "Please generate a script for a children book on the topic provided by the user. Each page (=beat) must haven an image prompt appropriate for the text.",
   "presentationStyle": {
     "$mulmocast": {
-      "version": "1.0",
+      "version": "1.1",
       "credit": "closing"
     },
     "canvasSize": {

package/assets/templates/comic_strips.json

@@ -4,7 +4,7 @@
   "systemPrompt": "Generate a script for a presentation of the given topic. Another AI will generate comic strips for each beat based on the text description of that beat. Mention the reference in one of beats, if it exists. Use the JSON below as a template.",
   "presentationStyle": {
     "$mulmocast": {
-      "version": "1.0",
+      "version": "1.1",
       "credit": "closing"
     },
     "canvasSize": {

package/assets/templates/drslump_comic.json

@@ -4,7 +4,7 @@
   "systemPrompt": "Generate a script for a presentation of the given topic. Another AI will generate images for each beat based on the image prompt of that beat. Mention the reference in one of beats, if it exists. Use the JSON below as a template.",
   "presentationStyle": {
     "$mulmocast": {
-      "version": "1.0",
+      "version": "1.1",
       "credit": "closing"
     },
     "canvasSize": {

package/assets/templates/ghibli_comic.json

@@ -4,7 +4,7 @@
   "systemPrompt": "Generate a script for a presentation of the given topic. Another AI will generate comic strips for each beat based on the text description of that beat. Mention the reference in one of beats, if it exists. Use the JSON below as a template.",
   "presentationStyle": {
     "$mulmocast": {
-      "version": "1.0",
+      "version": "1.1",
       "credit": "closing"
     },
     "canvasSize": {

package/assets/templates/ghibli_image_only.json

@@ -4,7 +4,7 @@
   "systemPrompt": "Generate a script for a presentation of the given topic. Another AI will generate an image for each beat based on the text description of that beat. Use the JSON below as a template.",
   "presentationStyle": {
     "$mulmocast": {
-      "version": "1.0",
+      "version": "1.1",
       "credit": "closing"
     },
     "canvasSize": {

package/assets/templates/ghibli_shorts.json

@@ -4,7 +4,7 @@
   "systemPrompt": "Generate a Japanese script for a Youtube shorts of the given topic. Another AI will generate comic strips for each beat based on the text description of that beat. Mention the reference in one of beats, if it exists. Use the JSON below as a template.",
   "presentationStyle": {
     "$mulmocast": {
-      "version": "1.0",
+      "version": "1.1",
       "credit": "closing"
     },
     "canvasSize": {
@@ -12,9 +12,8 @@
       "height": 1536
     },
     "speechParams": {
-      "provider": "nijivoice",
       "speakers": {
-        "Presenter": { "voiceId": "3708ad43-cace-486c-a4ca-8fe41186e20c", "speechOptions": { "speed": 1.5 } }
+        "Presenter": { "provider": "nijivoice", "voiceId": "3708ad43-cace-486c-a4ca-8fe41186e20c", "speechOptions": { "speed": 1.5 } }
       }
     },
     "imageParams": {

package/assets/templates/ghost_comic.json

@@ -4,7 +4,7 @@
   "systemPrompt": "Generate a script for a presentation of the given topic. Another AI will generate images for each beat based on the image prompt of that beat. Mention the reference in one of beats, if it exists. Use the JSON below as a template.",
   "presentationStyle": {
     "$mulmocast": {
-      "version": "1.0",
+      "version": "1.1",
       "credit": "closing"
     },
     "canvasSize": {

package/assets/templates/onepiece_comic.json

@@ -4,7 +4,7 @@
   "systemPrompt": "Generate a script for a presentation of the given topic. Another AI will generate images for each beat based on the image prompt of that beat. Mention the reference in one of beats, if it exists. Use the JSON below as a template.",
   "presentationStyle": {
     "$mulmocast": {
-      "version": "1.0",
+      "version": "1.1",
       "credit": "closing"
     },
     "canvasSize": {

package/assets/templates/portrait_movie.json

@@ -4,7 +4,7 @@
   "systemPrompt": "Generate a script for a presentation of the given topic. Another AI will generate images for each beat based on the image prompt of that beat. Movie prompts must be written in English. Mention the reference in one of beats, if it exists. Use the JSON below as a template.",
   "presentationStyle": {
     "$mulmocast": {
-      "version": "1.0",
+      "version": "1.1",
       "credit": "closing"
     },
     "canvasSize": {

package/assets/templates/realistic_movie.json

@@ -4,7 +4,7 @@
   "systemPrompt": "Generate a script for a presentation of the given topic. Another AI will generate images for each beat based on the image prompt of that beat. Movie prompts must be written in English. Mention the reference in one of beats, if it exists. Use the JSON below as a template.",
   "presentationStyle": {
     "$mulmocast": {
-      "version": "1.0",
+      "version": "1.1",
       "credit": "closing"
     },
     "canvasSize": {

package/assets/templates/sensei_and_taro.json

@@ -4,7 +4,7 @@
   "systemPrompt": "この件について、内容全てを高校生にも分かるように、太郎くん(Student)と先生(Teacher)の会話、という形の台本をArtifactとして作って。ただし要点はしっかりと押さえて。以下に別のトピックに関するサンプルを貼り付けます。このJSONフォーマットに従って。",
   "presentationStyle": {
     "$mulmocast": {
-      "version": "1.0",
+      "version": "1.1",
       "credit": "closing"
     },
     "canvasSize": {
@@ -15,11 +15,10 @@
       "style": "<style>Ghibli style. Student (Taro) is a young teenager with a dark short hair with glasses. Teacher is a middle-aged man with grey hair and moustache.</style>"
     },
     "speechParams": {
-      "provider": "nijivoice",
       "speakers": {
-        "Announcer": { "displayName": { "ja": "アナウンサー" }, "voiceId": "3708ad43-cace-486c-a4ca-8fe41186e20c" },
-        "Student": { "displayName": { "ja": "太郎" }, "voiceId": "a7619e48-bf6a-4f9f-843f-40485651257f" },
-        "Teacher": { "displayName": { "ja": "先生" }, "voiceId": "bc06c63f-fef6-43b6-92f7-67f919bd5dae" }
+        "Announcer": { "provider": "nijivoice", "displayName": { "ja": "アナウンサー" }, "voiceId": "3708ad43-cace-486c-a4ca-8fe41186e20c" },
+        "Student": { "provider": "nijivoice", "displayName": { "ja": "太郎" }, "voiceId": "a7619e48-bf6a-4f9f-843f-40485651257f" },
+        "Teacher": { "provider": "nijivoice", "displayName": { "ja": "先生" }, "voiceId": "bc06c63f-fef6-43b6-92f7-67f919bd5dae" }
       }
     }
   },

package/assets/templates/shorts.json

@@ -4,7 +4,7 @@
   "systemPrompt": "Generate a script for a Youtube shorts of the given topic. The first beat should be a hook, which describes the topic. Another AI will generate images for each beat based on the image prompt of that beat. Movie prompts must be written in English.",
   "presentationStyle": {
     "$mulmocast": {
-      "version": "1.0"
+      "version": "1.1"
     },
     "canvasSize": {
       "width": 720,

package/assets/templates/trailer.json

@@ -4,7 +4,7 @@
   "systemPrompt": "Generate a script for a movie trailer of the given story. Another AI will generate images for each beat based on the image prompt of that beat. Movie prompts must be written in English.",
   "presentationStyle": {
     "$mulmocast": {
-      "version": "1.0"
+      "version": "1.1"
     },
     "canvasSize": {
       "width": 1280,

package/lib/actions/audio.js

@@ -9,6 +9,7 @@ import ttsGoogleAgent from "../agents/tts_google_agent.js";
 import ttsElevenlabsAgent from "../agents/tts_elevenlabs_agent.js";
 import { fileWriteAgent } from "@graphai/vanilla_node_agents";
 import { MulmoPresentationStyleMethods } from "../methods/index.js";
+import { text2SpeechProviderSchema, } from "../types/index.js";
 import { fileCacheAgentFilter } from "../utils/filters.js";
 import { getAudioArtifactFilePath, getAudioFilePath, getOutputStudioFilePath, resolveDirPath, defaultBGMPath, mkdir, writingMessage } from "../utils/file.js";
 import { text2hash, localizedText, settings2GraphAIConfig } from "../utils/utils.js";
@@ -30,12 +31,10 @@ const getAudioPath = (context, beat, audioFile) => {
     return audioFile;
 };
 const getAudioParam = (presentationStyle, beat) => {
-    const voiceId = MulmoPresentationStyleMethods.getVoiceId(presentationStyle, beat);
-    // Use speaker-specific provider if available, otherwise fall back to script-level provider
-    const provider = MulmoPresentationStyleMethods.getTTSProvider(presentationStyle, beat);
-    const speechOptions = MulmoPresentationStyleMethods.getSpeechOptions(presentationStyle, beat);
-    const model = MulmoPresentationStyleMethods.getTTSModel(presentationStyle, beat);
-    return { voiceId, provider, speechOptions, model };
+    const speaker = MulmoPresentationStyleMethods.getSpeaker(presentationStyle, beat);
+    const speechOptions = { ...speaker.speechOptions, ...beat.speechOptions };
+    const provider = text2SpeechProviderSchema.parse(speaker.provider);
+    return { voiceId: speaker.voiceId, provider, speechOptions, model: speaker.model };
 };
 export const getBeatAudioPath = (text, context, beat, lang) => {
     const audioDirPath = MulmoStudioContextMethods.getAudioDirPath(context);
@@ -183,7 +182,7 @@ export const audioFilePath = (context) => {
 const getConcurrency = (context) => {
     // Check if any speaker uses nijivoice or elevenlabs (providers that require concurrency = 1)
     const hasLimitedConcurrencyProvider = Object.values(context.presentationStyle.speechParams.speakers).some((speaker) => {
-        const provider = (speaker.provider ?? context.presentationStyle.speechParams.provider);
+        const provider = text2SpeechProviderSchema.parse(speaker.provider);
         return provider2TTSAgent[provider].hasLimitedConcurrency;
     });
     return hasLimitedConcurrencyProvider ? 1 : 8;

package/lib/actions/image_agents.d.ts

@@ -1,4 +1,4 @@
-import { MulmoStudioContext, MulmoBeat, MulmoCanvasDimension } from "../types/index.js";
+import { MulmoStudioContext, MulmoBeat, MulmoCanvasDimension, MulmoImageParams } from "../types/index.js";
 export declare const imagePreprocessAgent: (namedInputs: {
     context: MulmoStudioContext;
     beat: MulmoBeat;
@@ -12,32 +12,15 @@ export declare const imagePreprocessAgent: (namedInputs: {
 } | {
     imagePath: string | undefined;
     referenceImageForMovie: string | undefined;
-    imageParams: {
-        provider: string;
-        model?: string | undefined;
-        style?: string | undefined;
-        moderation?: string | undefined;
-        images?: Record<string, {
-            type: "image";
-            source: {
-                url: string;
-                kind: "url";
-            } | {
-                kind: "base64";
-                data: string;
-            } | {
-                text: string;
-                kind: "text";
-            } | {
-                path: string;
-                kind: "path";
-            };
-        } | {
-            type: "imagePrompt";
-            prompt: string;
-        }> | undefined;
-    };
+    imageParams: MulmoImageParams;
     movieFile: string | undefined;
+    soundEffectFile?: string;
+    soundEffectPrompt?: string;
+    soundEffectModel?: string;
+    soundEffectAgentInfo?: {
+        agentName: string;
+        defaultModel: string;
+    };
     htmlPrompt?: undefined;
     htmlPath?: undefined;
     htmlImageSystemPrompt?: undefined;
@@ -59,32 +42,15 @@ export declare const imagePreprocessAgent: (namedInputs: {
             } | undefined;
         };
     };
-    imageParams: {
-        provider: string;
-        model?: string | undefined;
-        style?: string | undefined;
-        moderation?: string | undefined;
-        images?: Record<string, {
-            type: "image";
-            source: {
-                url: string;
-                kind: "url";
-            } | {
-                kind: "base64";
-                data: string;
-            } | {
-                text: string;
-                kind: "text";
-            } | {
-                path: string;
-                kind: "path";
-            };
-        } | {
-            type: "imagePrompt";
-            prompt: string;
-        }> | undefined;
-    };
+    imageParams: MulmoImageParams;
     movieFile: string | undefined;
+    soundEffectFile?: string;
+    soundEffectPrompt?: string;
+    soundEffectModel?: string;
+    soundEffectAgentInfo?: {
+        agentName: string;
+        defaultModel: string;
+    };
     htmlPrompt?: undefined;
     htmlPath?: undefined;
     htmlImageSystemPrompt?: undefined;
@@ -109,32 +75,15 @@ export declare const imagePreprocessAgent: (namedInputs: {
             } | undefined;
         };
     };
-    imageParams: {
-        provider: string;
-        model?: string | undefined;
-        style?: string | undefined;
-        moderation?: string | undefined;
-        images?: Record<string, {
-            type: "image";
-            source: {
-                url: string;
-                kind: "url";
-            } | {
-                kind: "base64";
-                data: string;
-            } | {
-                text: string;
-                kind: "text";
-            } | {
-                path: string;
-                kind: "path";
-            };
-        } | {
-            type: "imagePrompt";
-            prompt: string;
-        }> | undefined;
-    };
+    imageParams: MulmoImageParams;
     movieFile: string | undefined;
+    soundEffectFile?: string;
+    soundEffectPrompt?: string;
+    soundEffectModel?: string;
+    soundEffectAgentInfo?: {
+        agentName: string;
+        defaultModel: string;
+    };
     htmlPrompt?: undefined;
     htmlPath?: undefined;
     htmlImageSystemPrompt?: undefined;

package/lib/actions/image_agents.js

@@ -1,5 +1,5 @@
 import { MulmoPresentationStyleMethods, MulmoStudioContextMethods, MulmoBeatMethods } from "../methods/index.js";
-import { getBeatPngImagePath, getBeatMoviePath } from "../utils/file.js";
+import { getBeatPngImagePath, getBeatMoviePaths } from "../utils/file.js";
 import { imagePrompt, htmlImageSystemPrompt } from "../utils/prompt.js";
 import { renderHTMLToImage } from "../utils/markdown.js";
 import { GraphAILogger } from "graphai";
@@ -18,10 +18,18 @@ export const imagePreprocessAgent = async (namedInputs) => {
         return { imagePath, htmlPrompt, htmlPath, htmlImageSystemPrompt: htmlImageSystemPrompt(context.presentationStyle.canvasSize) };
     }
     const imageAgentInfo = MulmoPresentationStyleMethods.getImageAgentInfo(context.presentationStyle, beat);
+    const moviePaths = getBeatMoviePaths(context, index);
     const returnValue = {
         imageParams: imageAgentInfo.imageParams,
-        movieFile: beat.moviePrompt ? getBeatMoviePath(context, index) : undefined,
+        movieFile: beat.moviePrompt ? moviePaths.movieFile : undefined,
     };
+    if (beat.soundEffectPrompt) {
+        returnValue.soundEffectAgentInfo = MulmoPresentationStyleMethods.getSoundEffectAgentInfo(context.presentationStyle, beat);
+        returnValue.soundEffectModel =
+            beat.soundEffectParams?.model ?? context.presentationStyle.soundEffectParams?.model ?? returnValue.soundEffectAgentInfo.defaultModel;
+        returnValue.soundEffectFile = moviePaths.soundEffectFile;
+        returnValue.soundEffectPrompt = beat.soundEffectPrompt;
+    }
     if (beat.image) {
         const plugin = MulmoBeatMethods.getPlugin(beat);
         const pluginPath = plugin.path({ beat, context, imagePath, ...htmlStyle(context, beat) });
@@ -29,7 +37,7 @@ export const imagePreprocessAgent = async (namedInputs) => {
         return { ...returnValue, imagePath: pluginPath, referenceImageForMovie: pluginPath };
     }
     const movieAgentInfo = MulmoPresentationStyleMethods.getMovieAgentInfo(context.presentationStyle, beat);
-    GraphAILogger.log(`movieParams: ${index}`, movieAgentInfo.movieParams, beat.moviePrompt);
+    GraphAILogger.log(`movieParams: ${index}`, movieAgentInfo.movieParams, returnValue.soundEffectAgentInfo, "\n", beat.moviePrompt, beat.soundEffectPrompt);
     if (beat.moviePrompt && !beat.imagePrompt) {
         return { ...returnValue, imagePath, imageFromMovie: true, movieAgentInfo }; // no image prompt, only movie prompt
     }

package/lib/actions/images.js

@@ -6,7 +6,7 @@ import * as vanilla from "@graphai/vanilla";
 import { openAIAgent } from "@graphai/openai_agent";
 import { anthropicAgent } from "@graphai/anthropic_agent";
 import { fileWriteAgent } from "@graphai/vanilla_node_agents";
-import { imageGoogleAgent, imageOpenaiAgent, movieGoogleAgent, movieReplicateAgent, mediaMockAgent } from "../agents/index.js";
+import { imageGoogleAgent, imageOpenaiAgent, movieGoogleAgent, movieReplicateAgent, mediaMockAgent, soundEffectReplicateAgent } from "../agents/index.js";
 import { MulmoPresentationStyleMethods, MulmoStudioContextMethods } from "../methods/index.js";
 import { getOutputStudioFilePath, mkdir } from "../utils/file.js";
 import { fileCacheAgentFilter } from "../utils/filters.js";
@@ -23,10 +23,14 @@ const movieAgents = {
     movieGoogleAgent,
     movieReplicateAgent,
 };
+const soundEffectAgents = {
+    soundEffectReplicateAgent,
+};
 const defaultAgents = {
     ...vanillaAgents,
     ...imageAgents,
     ...movieAgents,
+    ...soundEffectAgents,
     mediaMockAgent,
     fileWriteAgent,
     openAIAgent,
@@ -167,6 +171,9 @@ const beat_graph_data = {
         },
         audioChecker: {
             agent: async (namedInputs) => {
+                if (namedInputs.soundEffectFile) {
+                    return { hasMovieAudio: true };
+                }
                 const sourceFile = namedInputs.movieFile || namedInputs.imageFile;
                 if (!sourceFile) {
                     return { hasMovieAudio: false };
@@ -175,22 +182,47 @@ const beat_graph_data = {
                 return { hasMovieAudio: hasAudio };
             },
             inputs: {
-                onComplete: [":movieGenerator", ":htmlImageGenerator"], // to wait for movieGenerator and htmlImageGenerator to finish
+                onComplete: [":movieGenerator", ":htmlImageGenerator", ":soundEffectGenerator"], // to wait for movieGenerator and htmlImageGenerator to finish
                 movieFile: ":preprocessor.movieFile",
                 imageFile: ":preprocessor.imagePath",
+                soundEffectFile: ":preprocessor.soundEffectFile",
             },
         },
+        soundEffectGenerator: {
+            if: ":preprocessor.soundEffectPrompt",
+            agent: ":preprocessor.soundEffectAgentInfo.agentName",
+            inputs: {
+                onComplete: [":movieGenerator"], // to wait for movieGenerator to finish
+                prompt: ":preprocessor.soundEffectPrompt",
+                movieFile: ":preprocessor.movieFile",
+                soundEffectFile: ":preprocessor.soundEffectFile",
+                params: {
+                    model: ":preprocessor.soundEffectModel",
+                    duration: ":beat.duration",
+                },
+                cache: {
+                    force: [":context.force"],
+                    file: ":preprocessor.soundEffectFile",
+                    index: ":__mapIndex",
+                    sessionType: "soundEffect",
+                    mulmoContext: ":context",
+                },
+            },
+            defaultValue: {},
+        },
         output: {
             agent: "copyAgent",
             inputs: {
-                onComplete: [":imageFromMovie", ":htmlImageGenerator", ":audioChecker"], // to wait for imageFromMovie to finish
+                onComplete: [":imageFromMovie", ":htmlImageGenerator", ":audioChecker", ":soundEffectGenerator"], // to wait for imageFromMovie to finish
                 imageFile: ":preprocessor.imagePath",
                 movieFile: ":preprocessor.movieFile",
+                soundEffectFile: ":preprocessor.soundEffectFile",
                 hasMovieAudio: ":audioChecker.hasMovieAudio",
             },
             output: {
                 imageFile: ".imageFile",
                 movieFile: ".movieFile",
+                soundEffectFile: ".soundEffectFile",
                 hasMovieAudio: ".hasMovieAudio",
             },
             isResult: true,
@@ -284,7 +316,7 @@ export const graphOption = async (context, settings) => {
             {
                 name: "fileCacheAgentFilter",
                 agent: fileCacheAgentFilter,
-                nodeIds: ["imageGenerator", "movieGenerator", "htmlImageAgent"],
+                nodeIds: ["imageGenerator", "movieGenerator", "htmlImageAgent", "soundEffectGenerator"],
             },
         ],
         taskManager: new TaskManager(MulmoPresentationStyleMethods.getConcurrency(context.presentationStyle)),

package/lib/actions/movie.js

@@ -162,7 +162,7 @@ const createVideo = async (audioArtifactFilePath, outputVideoPath, context) => {
             beatTimestamps.push(timestamp);
             return timestamp; // Skip voice-over beats.
         }
-        const sourceFile = studioBeat.movieFile ?? studioBeat.imageFile;
+        const sourceFile = studioBeat.soundEffectFile ?? studioBeat.movieFile ?? studioBeat.imageFile;
         assert(!!sourceFile, `studioBeat.imageFile or studioBeat.movieFile is not set: index=${index}`);
         assert(!!studioBeat.duration, `studioBeat.duration is not set: index=${index}`);
         const extraPadding = (() => {

package/lib/agents/index.d.ts

@@ -10,8 +10,9 @@ import ttsElevenlabsAgent from "./tts_elevenlabs_agent.js";
 import ttsNijivoiceAgent from "./tts_nijivoice_agent.js";
 import ttsOpenaiAgent from "./tts_openai_agent.js";
 import validateSchemaAgent from "./validate_schema_agent.js";
+import soundEffectReplicateAgent from "./sound_effect_replicate_agent.js";
 import { browserlessAgent } from "@graphai/browserless_agent";
 import { textInputAgent } from "@graphai/input_agents";
 import { openAIAgent } from "@graphai/openai_agent";
 import { fileWriteAgent } from "@graphai/vanilla_node_agents";
-export { openAIAgent, fileWriteAgent, browserlessAgent, textInputAgent, addBGMAgent, combineAudioFilesAgent, imageGoogleAgent, imageOpenaiAgent, tavilySearchAgent, movieGoogleAgent, movieReplicateAgent, mediaMockAgent, ttsElevenlabsAgent, ttsNijivoiceAgent, ttsOpenaiAgent, validateSchemaAgent, };
+export { openAIAgent, fileWriteAgent, browserlessAgent, textInputAgent, addBGMAgent, combineAudioFilesAgent, imageGoogleAgent, imageOpenaiAgent, tavilySearchAgent, movieGoogleAgent, movieReplicateAgent, mediaMockAgent, ttsElevenlabsAgent, ttsNijivoiceAgent, ttsOpenaiAgent, validateSchemaAgent, soundEffectReplicateAgent, };

package/lib/agents/index.js

@@ -10,9 +10,10 @@ import ttsElevenlabsAgent from "./tts_elevenlabs_agent.js";
 import ttsNijivoiceAgent from "./tts_nijivoice_agent.js";
 import ttsOpenaiAgent from "./tts_openai_agent.js";
 import validateSchemaAgent from "./validate_schema_agent.js";
+import soundEffectReplicateAgent from "./sound_effect_replicate_agent.js";
 import { browserlessAgent } from "@graphai/browserless_agent";
 import { textInputAgent } from "@graphai/input_agents";
 import { openAIAgent } from "@graphai/openai_agent";
 // import * as vanilla from "@graphai/vanilla";
 import { fileWriteAgent } from "@graphai/vanilla_node_agents";
-export { openAIAgent, fileWriteAgent, browserlessAgent, textInputAgent, addBGMAgent, combineAudioFilesAgent, imageGoogleAgent, imageOpenaiAgent, tavilySearchAgent, movieGoogleAgent, movieReplicateAgent, mediaMockAgent, ttsElevenlabsAgent, ttsNijivoiceAgent, ttsOpenaiAgent, validateSchemaAgent, };
+export { openAIAgent, fileWriteAgent, browserlessAgent, textInputAgent, addBGMAgent, combineAudioFilesAgent, imageGoogleAgent, imageOpenaiAgent, tavilySearchAgent, movieGoogleAgent, movieReplicateAgent, mediaMockAgent, ttsElevenlabsAgent, ttsNijivoiceAgent, ttsOpenaiAgent, validateSchemaAgent, soundEffectReplicateAgent, };

package/lib/agents/movie_replicate_agent.js

@@ -11,6 +11,7 @@ async function generateMovie(model, apiKey, prompt, imagePath, aspectRatio, dura
         duration,
         image: undefined,
         start_image: undefined,
+        first_frame_image: undefined,
         aspect_ratio: aspectRatio, // only for bytedance/seedance-1-lite
         // resolution: "720p", // only for bytedance/seedance-1-lite
         // fps: 24, // only for bytedance/seedance-1-lite
@@ -22,15 +23,19 @@ async function generateMovie(model, apiKey, prompt, imagePath, aspectRatio, dura
     if (imagePath) {
         const buffer = readFileSync(imagePath);
         const base64Image = `data:image/png;base64,${buffer.toString("base64")}`;
-        if (model === "kwaivgi/kling-v2.1" || model === "kwaivgi/kling-v1.6-pro" || model === "minimax/hailuo-02") {
-            input.start_image = base64Image;
+        const start_image = provider2MovieAgent.replicate.modelParams[model]?.start_image;
+        if (start_image === "first_frame_image" || start_image === "image" || start_image === "start_image") {
+            input[start_image] = base64Image;
+        }
+        else if (start_image === undefined) {
+            throw new Error(`Model ${model} does not support image-to-video generation`);
         }
         else {
             input.image = base64Image;
         }
     }
     try {
-        const output = await replicate.run(model ?? provider2MovieAgent.replicate.defaultModel, { input });
+        const output = await replicate.run(model, { input });
         // Download the generated video
         if (output && typeof output === "object" && "url" in output) {
             const videoUrl = output.url();
@@ -62,13 +67,20 @@ export const getAspectRatio = (canvasSize) => {
 export const movieReplicateAgent = async ({ namedInputs, params, config, }) => {
     const { prompt, imagePath } = namedInputs;
     const aspectRatio = getAspectRatio(params.canvasSize);
-    const duration = params.duration ?? 5;
+    const model = params.model ?? provider2MovieAgent.replicate.defaultModel;
+    if (!provider2MovieAgent.replicate.modelParams[model]) {
+        throw new Error(`Model ${model} is not supported`);
+    }
+    const duration = params.duration ?? provider2MovieAgent.replicate.modelParams[model].durations[0] ?? 5;
+    if (!provider2MovieAgent.replicate.modelParams[model].durations.includes(duration)) {
+        throw new Error(`Duration ${duration} is not supported for model ${model}. Supported durations: ${provider2MovieAgent.replicate.modelParams[model].durations.join(", ")}`);
+    }
     const apiKey = config?.apiKey;
     if (!apiKey) {
         throw new Error("REPLICATE_API_TOKEN environment variable is required");
     }
     try {
-        const buffer = await generateMovie(params.model, apiKey, prompt, imagePath, aspectRatio, duration);
+        const buffer = await generateMovie(model, apiKey, prompt, imagePath, aspectRatio, duration);
         if (buffer) {
             return { buffer };
         }

package/lib/agents/sound_effect_replicate_agent.d.ts

@@ -0,0 +1,5 @@
+import type { AgentFunction, AgentFunctionInfo } from "graphai";
+import type { AgentBufferResult, SoundEffectAgentInputs, ReplicateSoundEffectAgentParams, ReplicateSoundEffectAgentConfig } from "../types/agent.js";
+export declare const soundEffectReplicateAgent: AgentFunction<ReplicateSoundEffectAgentParams, AgentBufferResult, SoundEffectAgentInputs, ReplicateSoundEffectAgentConfig>;
+declare const soundEffectReplicateAgentInfo: AgentFunctionInfo;
+export default soundEffectReplicateAgentInfo;