From e69a8382f282bf104936cb77dc5975f80720721f Mon Sep 17 00:00:00 2001
From: sabaimran <narmiabas@gmail.com>
Date: Wed, 9 Oct 2024 23:56:57 -0700
Subject: [PATCH] Add a code icon for code-related train of thought

---
 src/interface/web/app/components/chatMessage/chatMessage.tsx | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/interface/web/app/components/chatMessage/chatMessage.tsx b/src/interface/web/app/components/chatMessage/chatMessage.tsx
index 1836bfde..b75c852e 100644
--- a/src/interface/web/app/components/chatMessage/chatMessage.tsx
+++ b/src/interface/web/app/components/chatMessage/chatMessage.tsx
@@ -26,6 +26,7 @@ import {
     Palette,
     ClipboardText,
     Check,
+    Code,
 } from "@phosphor-icons/react";
 
 import DOMPurify from "dompurify";
@@ -278,6 +279,10 @@ function chooseIconFromHeader(header: string, iconColor: string) {
         return <Palette className={`${classNames}`} />;
     }
 
+    if (compareHeader.includes("code")) {
+        return <Code className={`${classNames}`} />;
+    }
+
     return <Brain className={`${classNames}`} />;
 }