From a88b592136778db353b68dd485b1df06fd09f7f6 Mon Sep 17 00:00:00 2001
From: Julian Bilcke <julian.bilcke@huggingface.co>
Date: Mon, 5 Aug 2024 17:45:45 +0200
Subject: [PATCH] fix for output type

---
 src/services/assistant/updateStoryAndScene.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/services/assistant/updateStoryAndScene.ts b/src/services/assistant/updateStoryAndScene.ts
index 178aa116..b837de58 100644
--- 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
             })
           )
         )