Refactor & Feat UI Page Chat [Image Exporter]

[+] refactor(exporter.tsx): extract variables for message role and avatar
[+] feat(exporter.tsx): add support for system messages to use a different avatar
This commit is contained in:
H0llyW00dzZ 2023-10-15 02:34:51 +07:00
parent 06e043d3f1
commit 9ed113b52b
No known key found for this signature in database
GPG Key ID: 05C7FFFC0845C930

View File

@ -543,15 +543,22 @@ export function ImagePreviewer(props: {
</div> </div>
</div> </div>
{props.messages.map((m, i) => { {props.messages.map((m, i) => {
const isUserMessage = m.role === "user";
const isSystemMessage = m.role === "system";
const avatar =
isUserMessage && config.avatar
? config.avatar
: isSystemMessage
? "2699-fe0f"
: mask.avatar;
const messageClass = `${styles["message"]} ${
styles["message-" + m.role]
}`;
return ( return (
<div <div className={messageClass} key={i}>
className={styles["message"] + " " + styles["message-" + m.role]}
key={i}
>
<div className={styles["avatar"]}> <div className={styles["avatar"]}>
<ExportAvatar <ExportAvatar avatar={avatar} />
avatar={m.role === "user" ? config.avatar : mask.avatar}
/>
</div> </div>
<div className={styles["body"]}> <div className={styles["body"]}>