allow custom user avatar image of presets

Xaiat · Jul 25, 2024 · d59ef80 · d59ef80
1 parent 0c17d1a
commit d59ef80
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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
@@ -311,14 +311,22 @@ 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({
                         avatarImg: data.value
                       });
                     }} />
                   } />
+                <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
@@ -10,6 +10,7 @@ export type Preset = {
   sourceUrl: string,
   desc: string,
   avatarImg: string,
+  userAvatarImg?: string,
   type: PresetType,
   // chat
   welcomeMessage: string,