diff --git a/app/components/exporter.tsx b/app/components/exporter.tsx index 0a885d874..d6f345941 100644 --- a/app/components/exporter.tsx +++ b/app/components/exporter.tsx @@ -543,15 +543,22 @@ export function ImagePreviewer(props: { {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 ( -
+
- +