From e31f45268291509129e4091d936337da684a1d82 Mon Sep 17 00:00:00 2001
From: Eric-2369 <admin@eri.cx>
Date: Mon, 10 Feb 2025 15:08:00 +0800
Subject: [PATCH] feat: use llm icon for exporter

---
 app/components/exporter.tsx | 37 ++++++++++++++-----------------------
 1 file changed, 14 insertions(+), 23 deletions(-)

diff --git a/app/components/exporter.tsx b/app/components/exporter.tsx
index 79ae87be2..f0d9a7f4b 100644
--- a/app/components/exporter.tsx
+++ b/app/components/exporter.tsx
@@ -23,7 +23,6 @@ import CopyIcon from "../icons/copy.svg";
 import LoadingIcon from "../icons/three-dots.svg";
 import ChatGptIcon from "../icons/chatgpt.png";
 import ShareIcon from "../icons/share.svg";
-import BotIcon from "../icons/bot.png";
 
 import DownloadIcon from "../icons/download.svg";
 import { useEffect, useMemo, useRef, useState } from "react";
@@ -407,22 +406,6 @@ export function PreviewActions(props: {
   );
 }
 
-function ExportAvatar(props: { avatar: string }) {
-  if (props.avatar === DEFAULT_MASK_AVATAR) {
-    return (
-      <img
-        src={BotIcon.src}
-        width={30}
-        height={30}
-        alt="bot"
-        className="user-avatar"
-      />
-    );
-  }
-
-  return <Avatar avatar={props.avatar} />;
-}
-
 export function ImagePreviewer(props: {
   messages: ChatMessage[];
   topic: string;
@@ -543,12 +526,16 @@ export function ImagePreviewer(props: {
           <div>
             <div className={styles["main-title"]}>NextChat</div>
             <div className={styles["sub-title"]}>
-              github.com/ChatGPTNextWeb/ChatGPT-Next-Web
+              github.com/ChatGPTNextWeb/NextChat
             </div>
             <div className={styles["icons"]}>
-              <ExportAvatar avatar={config.avatar} />
+              <Avatar avatar={config.avatar} />
               <span className={styles["icon-space"]}>&</span>
-              <ExportAvatar avatar={mask.avatar} />
+              {mask.avatar === DEFAULT_MASK_AVATAR ? (
+                <Avatar model={mask.modelConfig.model} />
+              ) : (
+                <Avatar avatar={mask.avatar} />
+              )}
             </div>
           </div>
           <div>
@@ -576,9 +563,13 @@ export function ImagePreviewer(props: {
               key={i}
             >
               <div className={styles["avatar"]}>
-                <ExportAvatar
-                  avatar={m.role === "user" ? config.avatar : mask.avatar}
-                />
+                {m.role === "user" ? (
+                  <Avatar avatar={config.avatar} />
+                ) : mask.avatar === DEFAULT_MASK_AVATAR ? (
+                  <Avatar model={m.model} />
+                ) : (
+                  <Avatar avatar={mask.avatar} />
+                )}
               </div>
 
               <div className={styles["body"]}>