From a884ff5d1f3e07259cd8d8cb9db746cdda1ff514 Mon Sep 17 00:00:00 2001
From: Julian Bilcke <julian.bilcke@huggingface.co>
Date: Tue, 17 Sep 2024 13:20:17 +0200
Subject: [PATCH] add api key setting for luma

---
 .../app/src/components/settings/provider.tsx  | 11 ++++++++
 .../workflows/lumalabs/index.ts               | 27 +------------------
 .../workflow-editor/workflows/piapi/index.ts  |  2 +-
 3 files changed, 13 insertions(+), 27 deletions(-)

diff --git a/packages/app/src/components/settings/provider.tsx b/packages/app/src/components/settings/provider.tsx
index 427cff20..ceead330 100644
--- a/packages/app/src/components/settings/provider.tsx
+++ b/packages/app/src/components/settings/provider.tsx
@@ -101,6 +101,9 @@ export function SettingsSectionProvider() {
 
   const letzAiApiKey = useSettings((s) => s.letzAiApiKey)
   const setLetzAiApiKey = useSettings((s) => s.setLetzAiApiKey)
+  
+  const lumaLabsApiKey = useSettings((s) => s.lumaLabsApiKey)
+  const setLumaLabsApiKey = useSettings((s) => s.setLumaLabsApiKey)
 
   const bigModelApiKey = useSettings((s) => s.bigModelApiKey)
   const setBigModelApiKey = useSettings((s) => s.setBigModelApiKey)
@@ -337,6 +340,14 @@ export function SettingsSectionProvider() {
               type={apiKeyType}
             />
 
+            <FormInput
+              label="Luma AI API Key"
+              value={lumaLabsApiKey}
+              defaultValue={defaultSettings.lumaLabsApiKey}
+              onChange={setLumaLabsApiKey}
+              type={apiKeyType}
+            />
+
             <FormInput
               label="BigModel API Key"
               value={bigModelApiKey}
diff --git a/packages/app/src/services/editors/workflow-editor/workflows/lumalabs/index.ts b/packages/app/src/services/editors/workflow-editor/workflows/lumalabs/index.ts
index cd29fe97..e15b1499 100644
--- a/packages/app/src/services/editors/workflow-editor/workflows/lumalabs/index.ts
+++ b/packages/app/src/services/editors/workflow-editor/workflows/lumalabs/index.ts
@@ -3,38 +3,13 @@ import {
   ClapWorkflowEngine,
   ClapWorkflowCategory,
   ClapWorkflowProvider,
-  ClapInputCategory,
 } from '@aitube/clap'
 
 import {
   genericAspectRatio,
-  genericAudio,
-  genericBaseImageUrl,
-  genericDrivingVideo,
-  genericFaceImage,
-  genericGuidanceScale,
-  genericHeight1024,
-  genericHeight2048,
-  genericIdWeight,
-  genericImage,
-  genericImageUrl,
-  genericInferenceSteps,
   genericKeyframes,
-  genericLora,
-  genericNegativePrompt,
   genericPrompt,
-  genericReferenceImages,
-  genericSeed,
-  genericStartStep,
-  genericSwapImage,
-  genericSwapImageUrl,
-  genericTargetImage,
-  genericTrueCFG,
-  genericVideo,
-  genericWidth1024,
-  genericWidth2048,
 } from '../common/defaultValues'
-import { sampleDrivingVideo } from '@/lib/core/constants'
 
 // ------------------------------------------------------------------------------
 // if a user is already using one of those workflows and you change its settings,
@@ -54,7 +29,7 @@ export const lumalabsWorkflows: ClapWorkflow[] = [
     engine: ClapWorkflowEngine.REST_API,
     provider: ClapWorkflowProvider.LUMALABS,
     category: ClapWorkflowCategory.VIDEO_GENERATION,
-    data: '',
+    data: 'dream-machine/v1',
     schema: '',
     /**
      * Inputs of the workflow (this is used to build an UI for the workflow automatically)
diff --git a/packages/app/src/services/editors/workflow-editor/workflows/piapi/index.ts b/packages/app/src/services/editors/workflow-editor/workflows/piapi/index.ts
index 7b90144b..4d35bcb7 100644
--- a/packages/app/src/services/editors/workflow-editor/workflows/piapi/index.ts
+++ b/packages/app/src/services/editors/workflow-editor/workflows/piapi/index.ts
@@ -41,7 +41,7 @@ export const piApiWorkflows: ClapWorkflow[] = [
   },
   {
     id: 'piapi://luma/v1/video',
-    label: 'Luma Labs Dream Machine',
+    label: 'Luma AI Dream Machine',
     description: '',
     tags: ['Dream Machine'],
     author: '',