Add kei stablediff
This commit is contained in:
@@ -6,6 +6,7 @@ import { globalFetch, readImage } from "../storage/globalApi"
|
|||||||
import { CharEmotion } from "../stores"
|
import { CharEmotion } from "../stores"
|
||||||
import type { OpenAIChat } from "."
|
import type { OpenAIChat } from "."
|
||||||
import { processZip } from "./processzip"
|
import { processZip } from "./processzip"
|
||||||
|
import { keiServerURL } from "../kei/kei"
|
||||||
export async function stableDiff(currentChar:character,prompt:string){
|
export async function stableDiff(currentChar:character,prompt:string){
|
||||||
let db = get(DataBase)
|
let db = get(DataBase)
|
||||||
|
|
||||||
@@ -488,5 +489,39 @@ export async function generateAIImage(genPrompt:string, currentChar:character, n
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
if(db.sdProvider === 'kei'){
|
||||||
|
const db = get(DataBase)
|
||||||
|
let auth = db?.account?.token
|
||||||
|
if(!auth){
|
||||||
|
db.account = JSON.parse(localStorage.getItem("fallbackRisuToken"))
|
||||||
|
auth = db?.account?.token
|
||||||
|
db.account.useSync = true
|
||||||
|
}
|
||||||
|
const da = await globalFetch(keiServerURL() + '/imaggen', {
|
||||||
|
body: {
|
||||||
|
"prompt": genPrompt,
|
||||||
|
},
|
||||||
|
headers: {
|
||||||
|
"x-api-key": auth
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
if(!da.ok || !da.data.success){
|
||||||
|
alertError(Buffer.from(da.data.message || da.data).toString())
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if(returnSdData === 'inlay'){
|
||||||
|
return da.data.data
|
||||||
|
}
|
||||||
|
else{
|
||||||
|
let charemotions = get(CharEmotion)
|
||||||
|
const img = da.data.data
|
||||||
|
const emos:[string, string,number][] = [[img, img, Date.now()]]
|
||||||
|
charemotions[currentChar.chaId] = emos
|
||||||
|
CharEmotion.set(charemotions)
|
||||||
|
}
|
||||||
|
return returnSdData
|
||||||
|
|
||||||
|
}
|
||||||
return ''
|
return ''
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user