Risuai 0.6.3 first commit
This commit is contained in:
kwaroran
686
src/ts/characters.ts
Normal file
686
src/ts/characters.ts
Normal file
@@ -0,0 +1,686 @@
|
||||
import { get, writable } from "svelte/store";
|
||||
import { DataBase, saveImage, setDatabase, type character, type Chat, defaultSdDataFunc } from "./database";
|
||||
import exifr from 'exifr'
|
||||
import { alertConfirm, alertError, alertNormal, alertSelect, alertStore } from "./alert";
|
||||
import { language } from "../lang";
|
||||
import { PngMetadata } from "./exif";
|
||||
import { encode as encodeMsgpack, decode as decodeMsgpack } from "@msgpack/msgpack";
|
||||
import { checkNullish, findCharacterbyId, selectMultipleFile, selectSingleFile, sleep } from "./util";
|
||||
import { v4 as uuidv4 } from 'uuid';
|
||||
import { selectedCharID } from "./stores";
|
||||
import { downloadFile, getFileSrc, readImage } from "./globalApi";
|
||||
|
||||
export function createNewCharacter() {
|
||||
let db = get(DataBase)
|
||||
db.characters.push(createBlankChar())
|
||||
setDatabase(db)
|
||||
return db.characters.length - 1
|
||||
}
|
||||
|
||||
export function createNewGroup(){
|
||||
let db = get(DataBase)
|
||||
db.characters.push({
|
||||
type: 'group',
|
||||
name: "",
|
||||
firstMessage: "",
|
||||
chats: [{
|
||||
message: [],
|
||||
note: '',
|
||||
name: 'Chat 1',
|
||||
localLore: []
|
||||
}], chatPage: 0,
|
||||
viewScreen: 'none',
|
||||
globalLore: [],
|
||||
characters: [],
|
||||
autoMode: false,
|
||||
useCharacterLore: true,
|
||||
emotionImages: [],
|
||||
customscript: [],
|
||||
chaId: uuidv4(),
|
||||
})
|
||||
setDatabase(db)
|
||||
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'){
|
||||
return ''
|
||||
}
|
||||
return null
|
||||
}
|
||||
const filesrc = await getFileSrc(loc)
|
||||
if(type === 'plain'){
|
||||
return filesrc
|
||||
}
|
||||
else if(type ==='css'){
|
||||
return `background: url("${filesrc}");background-size: cover;`
|
||||
}
|
||||
else{
|
||||
return `background: url("${filesrc}");background-size: contain;background-repeat: no-repeat;background-position: center;`
|
||||
}
|
||||
}
|
||||
|
||||
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){
|
||||
return
|
||||
}
|
||||
const img = selected.data
|
||||
let db = get(DataBase)
|
||||
const imgp = await saveImage(img)
|
||||
db.characters[charId].image = imgp
|
||||
setDatabase(db)
|
||||
}
|
||||
|
||||
export async function selectUserImg() {
|
||||
const selected = await selectSingleFile(['png'])
|
||||
if(!selected){
|
||||
return
|
||||
}
|
||||
const img = selected.data
|
||||
let db = get(DataBase)
|
||||
const imgp = await saveImage(img)
|
||||
db.userIcon = imgp
|
||||
setDatabase(db)
|
||||
}
|
||||
|
||||
export const addingEmotion = writable(false)
|
||||
|
||||
export async function addCharEmotion(charId:number) {
|
||||
addingEmotion.set(true)
|
||||
const selected = await selectMultipleFile(['png', 'webp', 'gif'])
|
||||
if(!selected){
|
||||
addingEmotion.set(false)
|
||||
return
|
||||
}
|
||||
let db = get(DataBase)
|
||||
for(const f of selected){
|
||||
console.log(f)
|
||||
const img = f.data
|
||||
const imgp = await saveImage(img)
|
||||
const name = f.name.replace('.png','').replace('.webp','')
|
||||
let dbChar = db.characters[charId]
|
||||
if(dbChar.type !== 'group'){
|
||||
dbChar.emotionImages.push([name,imgp])
|
||||
db.characters[charId] = dbChar
|
||||
}
|
||||
setDatabase(db)
|
||||
}
|
||||
addingEmotion.set(false)
|
||||
}
|
||||
|
||||
export async function rmCharEmotion(charId:number, emotionId:number) {
|
||||
let db = get(DataBase)
|
||||
let dbChar = db.characters[charId]
|
||||
if(dbChar.type !== 'group'){
|
||||
dbChar.emotionImages.splice(emotionId, 1)
|
||||
db.characters[charId] = dbChar
|
||||
}
|
||||
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 {
|
||||
|
||||
const mode = await alertSelect(['Export as JSON', "Export as TXT"])
|
||||
const selectedID = get(selectedCharID)
|
||||
const db = get(DataBase)
|
||||
const chat = db.characters[selectedID].chats[page]
|
||||
const char = db.characters[selectedID]
|
||||
const date = new Date().toJSON();
|
||||
console.log(mode)
|
||||
if(mode === '0'){
|
||||
const stringl = Buffer.from(JSON.stringify({
|
||||
type: 'risuChat',
|
||||
ver: 1,
|
||||
data: chat
|
||||
}), 'utf-8')
|
||||
|
||||
await downloadFile(`${char.name}_${date}_chat`.replace(/[<>:"/\\|?*\.\,]/g, "") + '.json', stringl)
|
||||
|
||||
}
|
||||
else{
|
||||
|
||||
let stringl = chat.message.map((v) => {
|
||||
if(v.saying){
|
||||
return `${findCharacterbyId(v.saying).name}\n${v.data}`
|
||||
}
|
||||
else{
|
||||
return `${v.role === 'char' ? char.name : db.username}\n${v.data}`
|
||||
}
|
||||
}).join('\n\n')
|
||||
|
||||
if(char.type !== 'group'){
|
||||
stringl = `${char.name}\n${char.firstMessage}\n\n` + stringl
|
||||
}
|
||||
|
||||
await downloadFile(`${char.name}_${date}_chat`.replace(/[<>:"/\\|?*\.\,]/g, "") + '.txt', Buffer.from(stringl, 'utf-8'))
|
||||
|
||||
}
|
||||
alertNormal(language.successExport)
|
||||
} catch (error) {
|
||||
alertError(`${error}`)
|
||||
}
|
||||
}
|
||||
|
||||
export async function importChat(){
|
||||
const dat =await selectSingleFile(['json','jsonl'])
|
||||
if(!dat){
|
||||
return
|
||||
}
|
||||
try {
|
||||
const selectedID = get(selectedCharID)
|
||||
let db = get(DataBase)
|
||||
|
||||
if(dat.name.endsWith('jsonl')){
|
||||
const lines = Buffer.from(dat.data).toString('utf-8').split('\n')
|
||||
let newChat:Chat = {
|
||||
message: [],
|
||||
note: "",
|
||||
name: "Imported Chat",
|
||||
localLore: []
|
||||
}
|
||||
|
||||
let isFirst = true
|
||||
for(const line of lines){
|
||||
|
||||
const presedLine = JSON.parse(line)
|
||||
if(presedLine.name && presedLine.is_user, presedLine.mes){
|
||||
if(!isFirst){
|
||||
newChat.message.push({
|
||||
role: presedLine.is_user ? "user" : 'char',
|
||||
data: formatTavernChat(presedLine.mes, db.characters[selectedID].name)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
isFirst = false
|
||||
}
|
||||
|
||||
if(newChat.message.length === 0){
|
||||
alertError(language.errors.noData)
|
||||
return
|
||||
}
|
||||
|
||||
db.characters[selectedID].chats.push(newChat)
|
||||
setDatabase(db)
|
||||
alertNormal(language.successImport)
|
||||
}
|
||||
else{
|
||||
const json = JSON.parse(Buffer.from(dat.data).toString('utf-8'))
|
||||
if(json.type === 'risuChat' && json.ver === 1){
|
||||
const das:Chat = json.data
|
||||
if(!(checkNullish(das.message) || checkNullish(das.note) || checkNullish(das.name) || checkNullish(das.localLore))){
|
||||
db.characters[selectedID].chats.push(das)
|
||||
setDatabase(db)
|
||||
alertNormal(language.successImport)
|
||||
return
|
||||
}
|
||||
else{
|
||||
alertError(language.errors.noData)
|
||||
return
|
||||
}
|
||||
}
|
||||
else{
|
||||
alertError(language.errors.noData)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
} catch (error) {
|
||||
alertError(`${error}`)
|
||||
}
|
||||
}
|
||||
|
||||
function formatTavernChat(chat:string, charName:string){
|
||||
const db = get(DataBase)
|
||||
return chat.replace(/<([Uu]ser)>|\{\{([Uu]ser)\}\}/g, db.username).replace(/((\{\{)|<)([Cc]har)(=.+)?((\}\})|>)/g, charName)
|
||||
}
|
||||
|
||||
export function characterFormatUpdate(index:number|character){
|
||||
let db = get(DataBase)
|
||||
let cha = typeof(index) === 'number' ? db.characters[index] : index
|
||||
if(cha.chats.length === 0){
|
||||
cha.chats = [{
|
||||
message: [],
|
||||
note: '',
|
||||
name: 'Chat 1',
|
||||
localLore: []
|
||||
}]
|
||||
}
|
||||
if(!cha.chats[cha.chatPage]){
|
||||
cha.chatPage = 0
|
||||
}
|
||||
if(!cha.chats[cha.chatPage].message){
|
||||
cha.chats[cha.chatPage].message = []
|
||||
}
|
||||
if(!cha.type){
|
||||
cha.type = 'character'
|
||||
}
|
||||
if(!cha.chaId){
|
||||
cha.chaId = uuidv4()
|
||||
}
|
||||
if(cha.type !== 'group'){
|
||||
if(checkNullish(cha.sdData)){
|
||||
cha.sdData = defaultSdDataFunc()
|
||||
}
|
||||
if(checkNullish(cha.utilityBot)){
|
||||
cha.utilityBot = false
|
||||
}
|
||||
}
|
||||
if(checkNullish(cha.customscript)){
|
||||
cha.customscript = []
|
||||
}
|
||||
if(typeof(index) === 'number'){
|
||||
db.characters[index] = cha
|
||||
setDatabase(db)
|
||||
}
|
||||
return cha
|
||||
}
|
||||
|
||||
|
||||
export function createBlankChar():character{
|
||||
return {
|
||||
name: '',
|
||||
firstMessage: '',
|
||||
desc: '',
|
||||
notes: '',
|
||||
chats: [{
|
||||
message: [],
|
||||
note: '',
|
||||
name: 'Chat 1',
|
||||
localLore: []
|
||||
}],
|
||||
chatPage: 0,
|
||||
emotionImages: [],
|
||||
bias: [],
|
||||
viewScreen: 'none',
|
||||
globalLore: [],
|
||||
chaId: uuidv4(),
|
||||
type: 'character',
|
||||
sdData: defaultSdDataFunc(),
|
||||
utilityBot: false,
|
||||
customscript: [],
|
||||
exampleMessage: ''
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
export async function makeGroupImage() {
|
||||
try {
|
||||
alertStore.set({
|
||||
type: 'wait',
|
||||
msg: `Loading..`
|
||||
})
|
||||
const db = get(DataBase)
|
||||
const charID = get(selectedCharID)
|
||||
const group = db.characters[charID]
|
||||
if(group.type !== 'group'){
|
||||
return
|
||||
}
|
||||
|
||||
const imageUrls = await Promise.all(group.characters.map((v) => {
|
||||
return getCharImage(findCharacterbyId(v).image, 'plain')
|
||||
}))
|
||||
|
||||
|
||||
|
||||
const canvas = document.createElement("canvas");
|
||||
canvas.width = 256
|
||||
canvas.height = 256
|
||||
const ctx = canvas.getContext("2d");
|
||||
|
||||
// Load the images
|
||||
const images = [];
|
||||
let loadedImages = 0;
|
||||
|
||||
await Promise.all(
|
||||
imageUrls.map(
|
||||
(url) =>
|
||||
new Promise<void>((resolve) => {
|
||||
const img = new Image();
|
||||
img.crossOrigin="anonymous"
|
||||
img.onload = () => {
|
||||
images.push(img);
|
||||
resolve();
|
||||
};
|
||||
img.src = url;
|
||||
})
|
||||
)
|
||||
);
|
||||
|
||||
// Calculate dimensions and draw the grid
|
||||
const numImages = images.length;
|
||||
const numCols = Math.ceil(Math.sqrt(images.length));
|
||||
const numRows = Math.ceil(images.length / numCols);
|
||||
const cellWidth = canvas.width / numCols;
|
||||
const cellHeight = canvas.height / numRows;
|
||||
|
||||
for (let row = 0; row < numRows; row++) {
|
||||
for (let col = 0; col < numCols; col++) {
|
||||
const index = row * numCols + col;
|
||||
if (index >= numImages) break;
|
||||
ctx.drawImage(
|
||||
images[index],
|
||||
col * cellWidth,
|
||||
row * cellHeight,
|
||||
cellWidth,
|
||||
cellHeight
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
// Return the image URI
|
||||
|
||||
const uri = canvas.toDataURL()
|
||||
console.log(uri)
|
||||
canvas.remove()
|
||||
db.characters[charID].image = await saveImage(dataURLtoBuffer(uri));
|
||||
setDatabase(db)
|
||||
alertStore.set({
|
||||
type: 'none',
|
||||
msg: ''
|
||||
})
|
||||
} catch (error) {
|
||||
alertError(`${error}`)
|
||||
}
|
||||
}
|
||||
|
||||
function dataURLtoBuffer(string:string){
|
||||
const regex = /^data:.+\/(.+);base64,(.*)$/;
|
||||
|
||||
const matches = string.match(regex);
|
||||
const ext = matches[1];
|
||||
const data = matches[2];
|
||||
return Buffer.from(data, 'base64');
|
||||
}
|
||||
|
||||
export async function addDefaultCharacters() {
|
||||
const imgs = [fetch('/sample/rika.png'),fetch('/sample/yuzu.png')]
|
||||
|
||||
alertStore.set({
|
||||
type: 'wait',
|
||||
msg: `Loading Sample bots...`
|
||||
})
|
||||
|
||||
for(const img of imgs){
|
||||
const imgBuffer = await (await img).arrayBuffer()
|
||||
const readed = (await exifr.parse(imgBuffer, true))
|
||||
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++){
|
||||
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(Buffer.from(imgBuffer)))
|
||||
char.chaId = uuidv4()
|
||||
db.characters.push(characterFormatUpdate(char))
|
||||
setDatabase(db)
|
||||
}
|
||||
|
||||
alertStore.set({
|
||||
type: 'none',
|
||||
msg: ''
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user