mincomk/risuai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[feat] spec v1 basic boilerplate

Browse Source
This commit is contained in:
kwaroran
2023-05-09 01:17:07 +09:00
parent ba8273d4e6
commit ba286099eb
4 changed files with 340 additions and 240 deletions
Download Patch File Download Diff File Expand all files Collapse all files

238
src/ts/characters.ts
View File

@@ -42,143 +42,6 @@ export function createNewGroup(){
return db.characters.length - 1
}
export async function importCharacter() {
try {
const f = await selectSingleFile(['png', 'json'])
if(!f){
return
}
if(f.name.endsWith('json')){
const da = JSON.parse(Buffer.from(f.data).toString('utf-8'))
if((da.char_name || da.name) && (da.char_persona || da.description) && (da.char_greeting || da.first_mes)){
let db = get(DataBase)
db.characters.push({
name: da.char_name ?? da.name,
firstMessage: da.char_greeting ?? da.first_mes,
desc: da.char_persona ?? da.description,
notes: '',
chats: [{
message: [],
note: '',
name: 'Chat 1',
localLore: []
}],
chatPage: 0,
image: '',
emotionImages: [],
bias: [],
globalLore: [],
viewScreen: 'none',
chaId: uuidv4(),
sdData: defaultSdDataFunc(),
utilityBot: false,
customscript: [],
exampleMessage: ''
})
DataBase.set(db)
alertNormal(language.importedCharacter)
return
}
else{
alertError(language.errors.noData)
return
}
}
alertStore.set({
type: 'wait',
msg: 'Loading... (Reading)'
})
await sleep(10)
const img = f.data
const readed = (await exifr.parse(img, true))
console.log(readed)
if(readed.risuai){
await sleep(10)
const va = decodeMsgpack(Buffer.from(readed.risuai, 'base64')) as any
if(va.type !== 101){
alertError(language.errors.noData)
return
}
let char:character = va.data
let db = get(DataBase)
if(char.emotionImages && char.emotionImages.length > 0){
for(let i=0;i<char.emotionImages.length;i++){
alertStore.set({
type: 'wait',
msg: `Loading... (Getting Emotions ${i} / ${char.emotionImages.length})`
})
await sleep(10)
const imgp = await saveImage(char.emotionImages[i][1] as any)
char.emotionImages[i][1] = imgp
}
}
char.chats = [{
message: [],
note: '',
name: 'Chat 1',
localLore: []
}]
if(checkNullish(char.sdData)){
char.sdData = defaultSdDataFunc()
}
char.chatPage = 0
char.image = await saveImage(PngMetadata.filter(img))
db.characters.push(characterFormatUpdate(char))
char.chaId = uuidv4()
setDatabase(db)
alertNormal(language.importedCharacter)
return db.characters.length - 1
}
else if(readed.chara){
const charaData:TavernChar = JSON.parse(Buffer.from(readed.chara, 'base64').toString('utf-8'))
if(charaData.first_mes && charaData.name && charaData.description){
const imgp = await saveImage(PngMetadata.filter(img))
let db = get(DataBase)
db.characters.push({
name: charaData.name,
firstMessage: charaData.first_mes,
desc: charaData.description,
notes: '',
chats: [{
message: [],
note: '',
name: 'Chat 1',
localLore: []
}],
chatPage: 0,
image: imgp,
emotionImages: [],
bias: [],
globalLore: [],
viewScreen: 'none',
chaId: uuidv4(),
sdData: defaultSdDataFunc(),
utilityBot: false,
customscript: [],
exampleMessage: ''
})
DataBase.set(db)
alertNormal(language.importedCharacter)
return db.characters.length - 1
}
alertError(language.errors.noData)
return null
}
else{
alertError(language.errors.noData)
return null
}
} catch (error) {
alertError(`${error}`)
return null
}
}
export async function getCharImage(loc:string, type:'plain'|'css'|'contain') {
if(!loc || loc === ''){
if(type ==='css'){
@@ -198,19 +61,6 @@ export async function getCharImage(loc:string, type:'plain'|'css'|'contain') {
}
}
interface TavernChar{
avatar: "none"
chat: string
create_date: string
description: string
first_mes: string
mes_example: "<START>"
name: string
personality: ""
scenario: ""
talkativeness: "0.5"
}
export async function selectCharImg(charId:number) {
const selected = await selectSingleFile(['png'])
if(!selected){
@@ -270,94 +120,6 @@ export async function rmCharEmotion(charId:number, emotionId:number) {
setDatabase(db)
}
export async function exportChar(charaID:number) {
const db = get(DataBase)
let char:character = JSON.parse(JSON.stringify(db.characters[charaID]))
if(!char.image){
alertError('Image Required')
return
}
const conf = await alertConfirm(language.exportConfirm)
if(!conf){
return
}
alertStore.set({
type: 'wait',
msg: 'Loading...'
})
let img = await readImage(char.image)
try{
if(char.emotionImages && char.emotionImages.length > 0){
for(let i=0;i<char.emotionImages.length;i++){
alertStore.set({
type: 'wait',
msg: `Loading... (Getting Emotions ${i} / ${char.emotionImages.length})`
})
const rData = await readImage(char.emotionImages[i][1])
char.emotionImages[i][1] = rData as any
}
}
char.chats = []
alertStore.set({
type: 'wait',
msg: 'Loading... (Compressing)'
})
await sleep(10)
const data = Buffer.from(encodeMsgpack({
data: char,
type: 101
})).toString('base64')
alertStore.set({
type: 'wait',
msg: 'Loading... (Writing Exif)'
})
const tavernData:TavernChar = {
avatar: "none",
chat: "",
create_date: `${Date.now()}`,
description: char.desc,
first_mes: char.firstMessage,
mes_example: "<START>",
name: char.name,
personality: "",
scenario: "",
talkativeness: "0.5"
}
await sleep(10)
img = PngMetadata.write(img, {
'chara': Buffer.from(JSON.stringify(tavernData)).toString('base64'),
'risuai': data
})
alertStore.set({
type: 'wait',
msg: 'Loading... (Writing)'
})
char.image = ''
await sleep(10)
await downloadFile(`${char.name.replace(/[<>:"/\\|?*\.\,]/g, "")}_export.png`, img)
alertNormal(language.successExport)
}
catch(e){
alertError(`${e}`)
}
}
export async function exportChat(page:number){
try {