fix for output type

jbilcke-hf · Aug 5, 2024 · a88b592 · a88b592
1 parent f4a3eff
commit a88b592
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/services/assistant/updateStoryAndScene.ts b/src/services/assistant/updateStoryAndScene.ts
@@ -1,4 +1,4 @@
-import { ClapSegment, ClapSegmentCategory, newSegment } from '@aitube/clap'
+import { ClapOutputType, ClapSegment, ClapSegmentCategory, newSegment } from '@aitube/clap'
 import {
   AssistantMessage,
   AssistantSceneSegment,
@@ -76,6 +76,8 @@ export async function updateStoryAndScene({
       prompt: prompt,
       label: prompt,
       category,
+      // TODO put SOUND in here
+      // outputType: category === ClapSegmentCategory.SOUND || category === ClapSegmentCategory.MUSIC || category === ClapSegmentCategory.DIALOGUE
     }
 
     const segment: TimelineSegment = await clapSegmentToTimelineSegment(
@@ -150,6 +152,7 @@ export async function updateStoryAndScene({
             newSegment({
               ...segmentProperties,
               category: ClapSegmentCategory.VIDEO,
+              outputType: ClapOutputType.VIDEO,
             })
           )
         )
@@ -158,6 +161,7 @@ export async function updateStoryAndScene({
             newSegment({
               ...segmentProperties,
               category: ClapSegmentCategory.STORYBOARD,
+              outputType: ClapOutputType.IMAGE
             })
           )
         )