From 622854489e97f1ef03c631129b68066cdd7171c9 Mon Sep 17 00:00:00 2001
From: kwaroran <amasio1234@proton.me>
Date: Fri, 31 May 2024 13:15:01 +0900
Subject: [PATCH] feat: Add dalle support

---
 src/lib/Setting/Pages/OtherBotSettings.svelte | 22 +++++++++++++++++--
 src/ts/process/stableDiff.ts                  | 15 +++++++++++++
 2 files changed, 35 insertions(+), 2 deletions(-)

diff --git a/src/lib/Setting/Pages/OtherBotSettings.svelte b/src/lib/Setting/Pages/OtherBotSettings.svelte
index 91a395c5..f4146181 100644
--- a/src/lib/Setting/Pages/OtherBotSettings.svelte
+++ b/src/lib/Setting/Pages/OtherBotSettings.svelte
@@ -14,6 +14,21 @@
     import { getCharImage } from "src/ts/characters";
     import Arcodion from "src/lib/UI/Arcodion.svelte";
     $:{
+        $DataBase.NAIImgConfig ??= {
+            width: 512,
+            height: 512,
+            sampler: 'k_euler',
+            steps: 100,
+            scale: 1,
+            sm: false,
+            sm_dyn: false,
+            strength: 0.5,
+            noise: 0.5,
+            image: '',
+            InfoExtracted: 0.5,
+            RefStrength: 0.5,
+            refimage: ''
+        }
         if ($DataBase.NAIImgConfig.sampler === 'ddim_v3'){
             $DataBase.NAIImgConfig.sm = false
             $DataBase.NAIImgConfig.sm_dyn = false
@@ -29,8 +44,7 @@
         <OptionInput value="" >None</OptionInput>
         <OptionInput value="webui" >Stable Diffusion WebUI</OptionInput>
         <OptionInput value="novelai" >Novel AI</OptionInput>
-        <!-- TODO -->
-        <!-- <OptionInput value="runpod" >Runpod Serverless</OptionInput> -->
+        <OptionInput value="dalle" >Dall-E</OptionInput>
     </SelectInput>
     
     {#if $DataBase.sdProvider === 'webui'}
@@ -178,6 +192,10 @@
             </button>
         {/if}
     {/if}
+
+
+    <span class="text-textcolor">OpenAI API Key</span>
+    <TextInput size="sm" marginBottom placeholder="sk-..." bind:value={$DataBase.openAIKey}/>
 </Arcodion>
 
 <Arcodion name="TTS" styled>
diff --git a/src/ts/process/stableDiff.ts b/src/ts/process/stableDiff.ts
index cde280e4..18125aef 100644
--- a/src/ts/process/stableDiff.ts
+++ b/src/ts/process/stableDiff.ts
@@ -320,5 +320,20 @@ export async function generateAIImage(genPrompt:string, currentChar:character, n
             return false   
         }
     }
+    if(db.sdProvider === 'dalle'){
+        const data = await globalFetch("https://api.openai.com/v1/images/generations", {
+            body: {
+                "prompt": genPrompt,
+                "model": "dall-e-3",
+                "response_format": "b64_json",
+                "style": "natural"
+            },
+            headers: {
+                "Authorization": "Bearer " + db.openAIKey
+            }
+        })
+
+        return data?.data?.url ?? ""
+    }
     return ''
 }
\ No newline at end of file