From bdfdf5141a2756a70c8f76c2db4067c68c8b2d74 Mon Sep 17 00:00:00 2001
From: kwaroran <amasio1234@proton.me>
Date: Tue, 19 Mar 2024 02:43:39 +0900
Subject: [PATCH] Add request info inside chat feature

---
 src/lang/en.ts                                |  1 +
 src/lib/ChatScreens/Chat.svelte               | 26 ++++++++++++++-----
 src/lib/ChatScreens/DefaultChatScreen.svelte  |  3 +++
 .../Pages/AccessibilitySettings.svelte        |  4 +++
 src/ts/storage/database.ts                    |  1 +
 src/ts/util.ts                                |  7 ++++-
 6 files changed, 35 insertions(+), 7 deletions(-)

diff --git a/src/lang/en.ts b/src/lang/en.ts
index 1d3792e1..0c8a930f 100644
--- a/src/lang/en.ts
+++ b/src/lang/en.ts
@@ -540,4 +540,5 @@ export const languageEnglish = {
     persistentStorageDesc: "Your browser supports persistent storage. this is recommended for better performance and user experience.",
     enable: "Enable",
     postFile: "Post File",
+    requestInfoInsideChat: "Show Request Info Inside Chat",
 }
\ No newline at end of file
diff --git a/src/lib/ChatScreens/Chat.svelte b/src/lib/ChatScreens/Chat.svelte
index ca7d3e76..ea79480d 100644
--- a/src/lib/ChatScreens/Chat.svelte
+++ b/src/lib/ChatScreens/Chat.svelte
@@ -1,16 +1,17 @@
 <script lang="ts">
-    import { ArrowLeft, ArrowRight, PencilIcon, LanguagesIcon, RefreshCcwIcon, TrashIcon, CopyIcon, Volume2Icon } from "lucide-svelte";
+    import { ArrowLeft, ArrowRight, PencilIcon, LanguagesIcon, RefreshCcwIcon, TrashIcon, CopyIcon, Volume2Icon, BotIcon } from "lucide-svelte";
     import { ParseMarkdown, type simpleCharacterArgument } from "../../ts/parser";
     import AutoresizeArea from "../UI/GUI/TextAreaResizable.svelte";
     import { alertConfirm, alertError } from "../../ts/alert";
     import { language } from "../../lang";
-    import { DataBase, type groupChat } from "../../ts/storage/database";
-    import { CurrentCharacter, CurrentChat, selectedCharID } from "../../ts/stores";
-    import { translate, translateHTML } from "../../ts/translator/translator";
+    import { DataBase, type MessageGenerationInfo } from "../../ts/storage/database";
+    import { CurrentCharacter, CurrentChat } from "../../ts/stores";
+    import { translateHTML } from "../../ts/translator/translator";
     import { risuChatParser } from "src/ts/process/scripts";
     import { get } from "svelte/store";
-  import { isEqual } from "lodash";
-  import { sayTTS } from "src/ts/process/tts";
+    import { capitalize, isEqual } from "lodash";
+    import { sayTTS } from "src/ts/process/tts";
+  import { getModelShortName } from "src/ts/model/names";
     export let message = ''
     export let name = ''
     export let largePortrait = false
@@ -18,6 +19,7 @@
     export let img:string|Promise<string> = ''
     export let idx = -1
     export let rerollIcon = false
+    export let MessageGenerationInfo:MessageGenerationInfo|null = null
     export let onReroll = () => {}
     export let unReroll = () => {}
     export let character:simpleCharacterArgument|string|null = null
@@ -199,6 +201,18 @@
                     {/if}
                 </div>
             </div>
+            {#if MessageGenerationInfo}
+                <div>
+                    <button class="text-sm p-1 text-textcolor2 border-darkborderc float-end mr-2 my-2
+                                    hover:ring-borderc hover:ring rounded-md hover:text-textcolor transition-all flex justify-center items-center" 
+                    >
+                        <BotIcon size={20} />
+                        <span class="ml-1">
+                            {capitalize(getModelShortName(MessageGenerationInfo.model))}
+                        </span>
+                    </button>
+                </div>
+            {/if}
             {#if editMode}
                 <AutoresizeArea bind:value={message} />
             {:else}
diff --git a/src/lib/ChatScreens/DefaultChatScreen.svelte b/src/lib/ChatScreens/DefaultChatScreen.svelte
index bbf7c823..a83c4865 100644
--- a/src/lib/ChatScreens/DefaultChatScreen.svelte
+++ b/src/lib/ChatScreens/DefaultChatScreen.svelte
@@ -520,6 +520,7 @@
                             isLastMemory={$CurrentChat.lastMemory === (chat.chatId ?? 'none') && $CurrentShowMemoryLimit}
                             character={$CurrentSimpleCharacter}
                             largePortrait={$CurrentCharacter.largePortrait}
+                            MessageGenerationInfo={chat.generationInfo}
                         />
                     {:else}
                         <Chat
@@ -533,6 +534,7 @@
                             isLastMemory={$CurrentChat.lastMemory === (chat.chatId ?? 'none') && $CurrentShowMemoryLimit}
                             character={chat.saying}
                             largePortrait={findCharacterbyId(chat.saying).largePortrait}
+                            MessageGenerationInfo={chat.generationInfo}
                         />
                     {/if}
                 {:else}
@@ -544,6 +546,7 @@
                         img={getCharImage($CurrentUserIcon, 'css')}
                         isLastMemory={$CurrentChat.lastMemory === (chat.chatId ?? 'none') && $CurrentShowMemoryLimit}
                         largePortrait={$DataBase.personas[$DataBase.selectedPersona].largePortrait}
+                        MessageGenerationInfo={chat.generationInfo}
                     />
                 {/if}
             {/each}
diff --git a/src/lib/Setting/Pages/AccessibilitySettings.svelte b/src/lib/Setting/Pages/AccessibilitySettings.svelte
index e6201c73..c7dce13b 100644
--- a/src/lib/Setting/Pages/AccessibilitySettings.svelte
+++ b/src/lib/Setting/Pages/AccessibilitySettings.svelte
@@ -41,4 +41,8 @@
 
 <div class="flex items-center mt-2">
     <Check bind:check={$DataBase.sideMenuRerollButton} name={language.sideMenuRerollButton}/>
+</div>
+
+<div class="flex items-center mt-2">
+    <Check bind:check={$DataBase.requestInfoInsideChat} name={language.requestInfoInsideChat}/>
 </div>
\ No newline at end of file
diff --git a/src/ts/storage/database.ts b/src/ts/storage/database.ts
index 23bfd6bf..5ed2e6b5 100644
--- a/src/ts/storage/database.ts
+++ b/src/ts/storage/database.ts
@@ -599,6 +599,7 @@ export interface Database{
     modules: RisuModule[]
     enabledModules: string[]
     sideMenuRerollButton?:boolean
+    requestInfoInsideChat?:boolean
 }
 
 export interface customscript{
diff --git a/src/ts/util.ts b/src/ts/util.ts
index b901e1a4..a7af2e6b 100644
--- a/src/ts/util.ts
+++ b/src/ts/util.ts
@@ -28,7 +28,8 @@ export function messageForm(arg:Message[], loadPages:number){
             data: reformatContent(m.data),
             index: i,
             saying: m.saying,
-            chatId: m.chatId ?? 'none'
+            chatId: m.chatId ?? 'none',
+            generationInfo: m.generationInfo,
         })
     }
     return a.slice(0, loadPages)
@@ -419,4 +420,8 @@ export const toLangName = (code:string) => {
     }
 }
 
+export const capitalize = (s:string) => {
+    return s.charAt(0).toUpperCase() + s.slice(1)
+}
+
 export const languageCodes = ["af","ak","am","an","ar","as","ay","az","be","bg","bh","bm","bn","br","bs","ca","co","cs","cy","da","de","dv","ee","el","en","eo","es","et","eu","fa","fi","fo","fr","fy","ga","gd","gl","gn","gu","ha","he","hi","hr","ht","hu","hy","ia","id","ig","is","it","iu","ja","jv","ka","kk","km","kn","ko","ku","ky","la","lb","lg","ln","lo","lt","lv","mg","mi","mk","ml","mn","mr","ms","mt","my","nb","ne","nl","nn","no","ny","oc","om","or","pa","pl","ps","pt","qu","rm","ro","ru","rw","sa","sd","si","sk","sl","sm","sn","so","sq","sr","st","su","sv","sw","ta","te","tg","th","ti","tk","tl","tn","to","tr","ts","tt","tw","ug","uk","ur","uz","vi","wa","wo","xh","yi","yo","zh","zu"]
\ No newline at end of file