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

336
src/ts/characterCards.ts Normal file
View File

@@ -0,0 +1,336 @@
import { get } from "svelte/store"
import { alertConfirm, alertError, alertNormal, alertStore } from "./alert"
import { DataBase, defaultSdDataFunc, type character, saveImage, setDatabase } from "./database"
import { checkNullish, selectSingleFile, sleep } from "./util"
import { language } from "src/lang"
import { encode as encodeMsgpack, decode as decodeMsgpack } from "@msgpack/msgpack";
import { v4 as uuidv4 } from 'uuid';
import exifr from 'exifr'
import { PngMetadata } from "./exif"
import { characterFormatUpdate } from "./characters"
import { downloadFile, readImage } from "./globalApi"
type OfficialCardSpec = {
spec: 'chara_card_v2'
spec_version: '2.0' // May 8th addition
data: {
name: string
description: string
personality: string
scenario: string
first_mes: string
mes_example: string
creator_notes: string
system_prompt: string
post_history_instructions: string
alternate_greetings: string[]
character_book?: CharacterBook
tags: string[]
creator: string
character_version: number
extensions: Record<string, any>
}
}
type CharacterBook = null
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(await importSpecv2(da)){
return
}
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.chara){
// standard spec v2 imports
const charaData:CharacterCardV2 = JSON.parse(Buffer.from(readed.chara, 'base64').toString('utf-8'))
if(await importSpecv2(charaData)){
return
}
}
if(readed.risuai){
// old risu imports
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:OldTavernChar = 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 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:OldTavernChar = {
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}`)
}
}
async function importSpecv2(card:CharacterCardV2):Promise<boolean>{
if(!card ||card.spec !== 'chara_card_v2'){
return false
}
let data = card.data
let db = get(DataBase)
const risuext = data.extensions.risuai
if(risuext && risuext.emotions){
for(let i=0;i<risuext.emotions.length;i++){
alertStore.set({
type: 'wait',
msg: `Loading... (Getting Emotions ${i} / ${risuext.emotions.length})`
})
await sleep(10)
const imgp = await saveImage(risuext.emotions[i][1] as any)
data.extensions.risuai.emotions[i][1] = imgp
}
}
}
type CharacterCardV2 = {
spec: 'chara_card_v2'
spec_version: '2.0' // May 8th addition
data: {
name: string
description: string
personality: string
scenario: string
first_mes: string
mes_example: string
creator_notes: string
system_prompt: string
post_history_instructions: string
alternate_greetings: string[]
character_book?: CharacterBook
tags: string[]
creator: string
character_version: number
extensions: {
risuai?:{
emotions:[string, string][]
}
}
}
}
interface OldTavernChar{
avatar: "none"
chat: string
create_date: string
description: string
first_mes: string
mes_example: "<START>"
name: string
personality: ""
scenario: ""
talkativeness: "0.5"
}

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 {