From d59ef80d41f42e761bf42303fe50a7e4f68261f6 Mon Sep 17 00:00:00 2001
From: josc146 <josStorer@outlook.com>
Date: Thu, 25 Jul 2024 23:49:23 +0800
Subject: [PATCH] allow custom user avatar image of presets

---
 frontend/src/_locales/ja/main.json                  |  3 ++-
 frontend/src/_locales/zh-hans/main.json             |  3 ++-
 frontend/src/pages/Chat.tsx                         |  2 ++
 frontend/src/pages/PresetsManager/PresetsButton.tsx | 10 +++++++++-
 frontend/src/types/presets.ts                       |  1 +
 5 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/frontend/src/_locales/ja/main.json b/frontend/src/_locales/ja/main.json
index 053138a8..c9389e85 100644
--- a/frontend/src/_locales/ja/main.json
+++ b/frontend/src/_locales/ja/main.json
@@ -185,7 +185,8 @@
   "Edit Character Settings": "キャラクター設定を編集",
   "Go Back": "戻る",
   "Description": "説明",
-  "Avatar Url": "アバターURL",
+  "Assistant Avatar Url": "アシスタントアバターURL",
+  "User Avatar Url": "ユーザーアバターURL",
   "Welcome Message": "ウェルカムメッセージ",
   "Display Preset Messages": "プリセットメッセージの表示",
   "Tag": "タグ",
diff --git a/frontend/src/_locales/zh-hans/main.json b/frontend/src/_locales/zh-hans/main.json
index cc9724e2..0e26ea6e 100644
--- a/frontend/src/_locales/zh-hans/main.json
+++ b/frontend/src/_locales/zh-hans/main.json
@@ -185,7 +185,8 @@
   "Edit Character Settings": "编辑人设",
   "Go Back": "返回",
   "Description": "描述",
-  "Avatar Url": "头像图片地址",
+  "Assistant Avatar Url": "AI头像图片地址",
+  "User Avatar Url": "用户头像图片地址",
   "Welcome Message": "欢迎语",
   "Display Preset Messages": "显示预设中的对话",
   "Tag": "标签",
diff --git a/frontend/src/pages/Chat.tsx b/frontend/src/pages/Chat.tsx
index a45ab576..6649ac00 100644
--- a/frontend/src/pages/Chat.tsx
+++ b/frontend/src/pages/Chat.tsx
@@ -135,6 +135,8 @@ const ChatMessageItem: FC<{
   let avatarImg: string | undefined;
   if (commonStore.activePreset && messageItem.sender === botName) {
     avatarImg = absPathAsset(commonStore.activePreset.avatarImg);
+  } else if (commonStore.activePreset && messageItem.sender === userName) {
+    avatarImg = commonStore.activePreset.userAvatarImg && absPathAsset(commonStore.activePreset.userAvatarImg);
   } else if (messageItem.avatarImg) {
     avatarImg = messageItem.avatarImg;
   }
diff --git a/frontend/src/pages/PresetsManager/PresetsButton.tsx b/frontend/src/pages/PresetsManager/PresetsButton.tsx
index 58ea976c..d9fce4e6 100644
--- a/frontend/src/pages/PresetsManager/PresetsButton.tsx
+++ b/frontend/src/pages/PresetsManager/PresetsButton.tsx
@@ -311,7 +311,7 @@ const ChatPresetEditor: FC<{
                       });
                     }} />
                   } />
-                <Labeled flex breakline label={t('Avatar Url')}
+                <Labeled flex breakline label={t('Assistant Avatar Url')}
                   content={
                     <Input value={editingPreset.avatarImg} onChange={(e, data) => {
                       setEditingPreset({
@@ -319,6 +319,14 @@ const ChatPresetEditor: FC<{
                       });
                     }} />
                   } />
+                <Labeled flex breakline label={t('User Avatar Url')}
+                  content={
+                    <Input value={editingPreset.userAvatarImg} onChange={(e, data) => {
+                      setEditingPreset({
+                        userAvatarImg: data.value
+                      });
+                    }} />
+                  } />
                 <Labeled flex breakline label={t('Welcome Message')}
                   content={
                     <Input disabled value={editingPreset.welcomeMessage} onChange={(e, data) => {
diff --git a/frontend/src/types/presets.ts b/frontend/src/types/presets.ts
index d5d6f5b3..5680bed6 100644
--- a/frontend/src/types/presets.ts
+++ b/frontend/src/types/presets.ts
@@ -10,6 +10,7 @@ export type Preset = {
   sourceUrl: string,
   desc: string,
   avatarImg: string,
+  userAvatarImg?: string,
   type: PresetType,
   // chat
   welcomeMessage: string,