From d7d005e71fe10476db7fa57d4f8211c7e02e42e3 Mon Sep 17 00:00:00 2001
From: Nicolas Mowen <nickmowen213@gmail.com>
Date: Mon, 16 Feb 2026 19:20:14 -0700
Subject: [PATCH] Add copy button

---
 web/public/locales/en/common.json       |  1 +
 web/src/components/chat/ChatMessage.tsx | 49 ++++++++++++++++++++++---
 2 files changed, 45 insertions(+), 5 deletions(-)

diff --git a/web/public/locales/en/common.json b/web/public/locales/en/common.json
index f75e0c97d..1181554af 100644
--- a/web/public/locales/en/common.json
+++ b/web/public/locales/en/common.json
@@ -129,6 +129,7 @@
     "cancel": "Cancel",
     "close": "Close",
     "copy": "Copy",
+    "copiedToClipboard": "Copied to clipboard",
     "back": "Back",
     "history": "History",
     "fullscreen": "Fullscreen",
diff --git a/web/src/components/chat/ChatMessage.tsx b/web/src/components/chat/ChatMessage.tsx
index 48117221f..3e87a495e 100644
--- a/web/src/components/chat/ChatMessage.tsx
+++ b/web/src/components/chat/ChatMessage.tsx
@@ -1,4 +1,14 @@
 import ReactMarkdown from "react-markdown";
+import { useTranslation } from "react-i18next";
+import copy from "copy-to-clipboard";
+import { toast } from "sonner";
+import { FaCopy } from "react-icons/fa";
+import { Button } from "@/components/ui/button";
+import {
+  Tooltip,
+  TooltipContent,
+  TooltipTrigger,
+} from "@/components/ui/tooltip";
 import { cn } from "@/lib/utils";
 
 type MessageBubbleProps = {
@@ -7,18 +17,47 @@ type MessageBubbleProps = {
 };
 
 export function MessageBubble({ role, content }: MessageBubbleProps) {
+  const { t } = useTranslation(["views/chat", "common"]);
   const isUser = role === "user";
 
+  const handleCopy = () => {
+    const text = content?.trim() || "";
+    if (!text) return;
+    if (copy(text)) {
+      toast.success(t("button.copiedToClipboard", { ns: "common" }));
+    }
+  };
+
   return (
     <div
       className={cn(
-        "rounded-lg px-3 py-2",
-        isUser
-          ? "self-end bg-primary text-primary-foreground"
-          : "self-start bg-muted",
+        "flex flex-col gap-1",
+        isUser ? "items-end self-end" : "items-start self-start",
       )}
     >
-      {isUser ? content : <ReactMarkdown>{content}</ReactMarkdown>}
+      <div
+        className={cn(
+          "rounded-lg px-3 py-2",
+          isUser ? "bg-primary text-primary-foreground" : "bg-muted",
+        )}
+      >
+        {isUser ? content : <ReactMarkdown>{content}</ReactMarkdown>}
+      </div>
+      <Tooltip>
+        <TooltipTrigger asChild>
+          <Button
+            variant="ghost"
+            size="icon"
+            className="size-7 text-muted-foreground hover:text-foreground"
+            onClick={handleCopy}
+            disabled={!content?.trim()}
+            aria-label={t("button.copy", { ns: "common" })}
+          >
+            <FaCopy className="size-3" />
+          </Button>
+        </TooltipTrigger>
+        <TooltipContent>{t("button.copy", { ns: "common" })}</TooltipContent>
+      </Tooltip>
     </div>
   );
 }