@copilotkit/react-ui 1.8.6 → 1.8.7

package/dist/components/chat/Modal.js

@@ -298,6 +298,19 @@ var DownloadIcon = /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
     )
   }
 );
+var UploadIcon = /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
+  "svg",
+  {
+    xmlns: "http://www.w3.org/2000/svg",
+    fill: "none",
+    viewBox: "0 0 24 24",
+    strokeWidth: "1.5",
+    stroke: "currentColor",
+    width: "24",
+    height: "24",
+    children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)("path", { strokeLinecap: "round", strokeLinejoin: "round", d: "M12 4.5v15m7.5-7.5h-15" })
+  }
+);
 var CheckIcon = /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
   "svg",
   {
@@ -363,7 +376,8 @@ var ChatContextProvider = ({
       pushToTalkIcon: MicrophoneIcon,
       copyIcon: CopyIcon,
       thumbsUpIcon: ThumbsUpIcon,
-      thumbsDownIcon: ThumbsDownIcon
+      thumbsDownIcon: ThumbsDownIcon,
+      uploadIcon: UploadIcon
     }), icons),
     [icons]
   );
@@ -1068,6 +1082,7 @@ var Messages = ({
   RenderActionExecutionMessage: RenderActionExecutionMessage2,
   RenderAgentStateMessage: RenderAgentStateMessage2,
   RenderResultMessage: RenderResultMessage2,
+  RenderImageMessage: RenderImageMessage2,
   AssistantMessage: AssistantMessage2,
   UserMessage: UserMessage2,
   onRegenerate,
@@ -1156,6 +1171,23 @@ var Messages = ({
             },
             index
           );
+        } else if (message.isImageMessage && message.isImageMessage()) {
+          return /* @__PURE__ */ (0, import_jsx_runtime10.jsx)(
+            RenderImageMessage2,
+            {
+              message,
+              inProgress,
+              index,
+              isCurrentMessage,
+              AssistantMessage: AssistantMessage2,
+              UserMessage: UserMessage2,
+              onRegenerate,
+              onCopy,
+              onThumbsUp,
+              onThumbsDown
+            },
+            index
+          );
         }
       }),
       interrupt
@@ -1398,7 +1430,7 @@ var usePushToTalk = ({
 // src/components/chat/Input.tsx
 var import_react_core6 = require("@copilotkit/react-core");
 var import_jsx_runtime12 = require("react/jsx-runtime");
-var Input = ({ inProgress, onSend, isVisible = false, onStop }) => {
+var Input = ({ inProgress, onSend, isVisible = false, onStop, onUpload }) => {
   const context = useChatContext();
   const copilotContext = (0, import_react_core6.useCopilotContext)();
   const pushToTalkConfigured = copilotContext.copilotApiConfig.textToSpeechUrl !== void 0 && copilotContext.copilotApiConfig.transcribeAudioUrl !== void 0;
@@ -1462,6 +1494,7 @@ var Input = ({ inProgress, onSend, isVisible = false, onStop }) => {
       }
     ),
     /* @__PURE__ */ (0, import_jsx_runtime12.jsxs)("div", { className: "copilotKitInputControls", children: [
+      onUpload && /* @__PURE__ */ (0, import_jsx_runtime12.jsx)("button", { onClick: onUpload, className: "copilotKitInputControlButton", children: context.icons.uploadIcon }),
       /* @__PURE__ */ (0, import_jsx_runtime12.jsx)("div", { style: { flexGrow: 1 } }),
       showPushToTalk && /* @__PURE__ */ (0, import_jsx_runtime12.jsx)(
         "button",
@@ -1489,7 +1522,7 @@ var Input = ({ inProgress, onSend, isVisible = false, onStop }) => {
 // src/components/chat/messages/UserMessage.tsx
 var import_jsx_runtime13 = require("react/jsx-runtime");
 var UserMessage = (props) => {
-  return /* @__PURE__ */ (0, import_jsx_runtime13.jsx)("div", { className: "copilotKitMessage copilotKitUserMessage", children: props.message });
+  return /* @__PURE__ */ (0, import_jsx_runtime13.jsx)("div", { className: "copilotKitMessage copilotKitUserMessage", children: props.subComponent || props.message });
 };
 
 // src/components/chat/Markdown.tsx
@@ -2343,13 +2376,77 @@ function RenderAgentStateMessage(_a) {
   }
 }
 
+// src/components/chat/messages/RenderImageMessage.tsx
+var import_jsx_runtime21 = require("react/jsx-runtime");
+function RenderImageMessage(_a) {
+  var _b = _a, {
+    UserMessage: UserMessage2 = UserMessage,
+    AssistantMessage: AssistantMessage2 = AssistantMessage
+  } = _b, props = __objRest(_b, [
+    "UserMessage",
+    "AssistantMessage"
+  ]);
+  const {
+    message,
+    inProgress,
+    index,
+    isCurrentMessage,
+    onRegenerate,
+    onCopy,
+    onThumbsUp,
+    onThumbsDown
+  } = props;
+  if (message.isImageMessage()) {
+    const imageData = `data:${message.format};base64,${message.bytes}`;
+    const imageComponent = /* @__PURE__ */ (0, import_jsx_runtime21.jsx)("div", { className: "copilotKitImage", children: /* @__PURE__ */ (0, import_jsx_runtime21.jsx)(
+      "img",
+      {
+        src: imageData,
+        alt: "User uploaded image",
+        style: { maxWidth: "100%", maxHeight: "300px", borderRadius: "8px" }
+      }
+    ) });
+    if (message.role === "user") {
+      return /* @__PURE__ */ (0, import_jsx_runtime21.jsx)(
+        UserMessage2,
+        {
+          "data-message-role": "user",
+          message: "",
+          rawData: message,
+          subComponent: imageComponent
+        },
+        index
+      );
+    } else if (message.role === "assistant") {
+      return /* @__PURE__ */ (0, import_jsx_runtime21.jsx)(
+        AssistantMessage2,
+        {
+          "data-message-role": "assistant",
+          message: "",
+          rawData: message,
+          subComponent: imageComponent,
+          isLoading: inProgress && isCurrentMessage && !message.bytes,
+          isGenerating: inProgress && isCurrentMessage && !!message.bytes,
+          isCurrentMessage,
+          onRegenerate: () => onRegenerate == null ? void 0 : onRegenerate(message.id),
+          onCopy,
+          onThumbsUp,
+          onThumbsDown
+        },
+        index
+      );
+    }
+  }
+  return null;
+}
+
 // src/components/chat/Suggestion.tsx
 var import_react_core9 = require("@copilotkit/react-core");
 var import_shared2 = require("@copilotkit/shared");
 var import_runtime_client_gql3 = require("@copilotkit/runtime-client-gql");
-var import_jsx_runtime21 = require("react/jsx-runtime");
+var import_jsx_runtime22 = require("react/jsx-runtime");
 function Suggestion({ title, message, onClick, partial, className }) {
-  return /* @__PURE__ */ (0, import_jsx_runtime21.jsx)(
+  return /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(
     "button",
     {
       disabled: partial,
@@ -2359,7 +2456,7 @@ function Suggestion({ title, message, onClick, partial, className }) {
       },
       className: className || (partial ? "suggestion loading" : "suggestion"),
       "data-test-id": "suggestion",
-      children: partial ? SmallSpinnerIcon : /* @__PURE__ */ (0, import_jsx_runtime21.jsx)("span", { children: title })
+      children: partial ? SmallSpinnerIcon : /* @__PURE__ */ (0, import_jsx_runtime22.jsx)("span", { children: title })
     }
   );
 }
@@ -2439,7 +2536,87 @@ var import_react_core10 = require("@copilotkit/react-core");
 var import_runtime_client_gql4 = require("@copilotkit/runtime-client-gql");
 var import_shared3 = require("@copilotkit/shared");
 var import_react_core11 = require("@copilotkit/react-core");
-var import_jsx_runtime22 = require("react/jsx-runtime");
+
+// src/components/chat/ImageUploadQueue.tsx
+var import_jsx_runtime23 = require("react/jsx-runtime");
+var ImageUploadQueue = ({
+  images,
+  onRemoveImage,
+  className = ""
+}) => {
+  if (images.length === 0)
+    return null;
+  return /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(
+    "div",
+    {
+      className: `copilotKitImageUploadQueue ${className}`,
+      style: {
+        display: "flex",
+        flexWrap: "wrap",
+        gap: "8px",
+        margin: "8px",
+        padding: "8px"
+      },
+      children: images.map((image, index) => /* @__PURE__ */ (0, import_jsx_runtime23.jsxs)(
+        "div",
+        {
+          className: "copilotKitImageUploadQueueItem",
+          style: {
+            position: "relative",
+            display: "inline-block",
+            width: "60px",
+            height: "60px",
+            borderRadius: "4px",
+            overflow: "hidden"
+          },
+          children: [
+            /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(
+              "img",
+              {
+                src: `data:${image.contentType};base64,${image.bytes}`,
+                alt: `Selected image ${index + 1}`,
+                style: {
+                  width: "100%",
+                  height: "100%",
+                  objectFit: "cover"
+                }
+              }
+            ),
+            /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(
+              "button",
+              {
+                onClick: () => onRemoveImage(index),
+                className: "copilotKitImageUploadQueueRemoveButton",
+                style: {
+                  position: "absolute",
+                  top: "2px",
+                  right: "2px",
+                  background: "rgba(0,0,0,0.6)",
+                  color: "white",
+                  border: "none",
+                  borderRadius: "50%",
+                  width: "18px",
+                  height: "18px",
+                  display: "flex",
+                  alignItems: "center",
+                  justifyContent: "center",
+                  cursor: "pointer",
+                  fontSize: "10px",
+                  padding: 0
+                },
+                children: "\u2715"
+              }
+            )
+          ]
+        },
+        index
+      ))
+    }
+  );
+};
+
+// src/components/chat/Chat.tsx
+var import_jsx_runtime24 = require("react/jsx-runtime");
 function CopilotChat({
   instructions,
   onSubmitMessage,
@@ -2456,14 +2633,64 @@ function CopilotChat({
   RenderActionExecutionMessage: RenderActionExecutionMessage2 = RenderActionExecutionMessage,
   RenderAgentStateMessage: RenderAgentStateMessage2 = RenderAgentStateMessage,
   RenderResultMessage: RenderResultMessage2 = RenderResultMessage,
+  RenderImageMessage: RenderImageMessage2 = RenderImageMessage,
   Input: Input2 = Input,
   className,
   icons,
   labels,
   AssistantMessage: AssistantMessage2 = AssistantMessage,
-  UserMessage: UserMessage2 = UserMessage
+  UserMessage: UserMessage2 = UserMessage,
+  imageUploadsEnabled,
+  inputFileAccept = "image/*"
 }) {
   const { additionalInstructions, setChatInstructions } = (0, import_react_core10.useCopilotContext)();
+  const [selectedImages, setSelectedImages] = (0, import_react13.useState)([]);
+  const fileInputRef = (0, import_react13.useRef)(null);
+  (0, import_react13.useEffect)(() => {
+    if (!imageUploadsEnabled)
+      return;
+    const handlePaste = (e) => __async(this, null, function* () {
+      var _a, _b;
+      const target = e.target;
+      if (!((_a = target.parentElement) == null ? void 0 : _a.classList.contains("copilotKitInput")))
+        return;
+      const items = Array.from(((_b = e.clipboardData) == null ? void 0 : _b.items) || []);
+      const imageItems = items.filter((item) => item.type.startsWith("image/"));
+      if (imageItems.length === 0)
+        return;
+      e.preventDefault();
+      const imagePromises = imageItems.map((item) => {
+        const file = item.getAsFile();
+        if (!file)
+          return Promise.resolve(null);
+        return new Promise((resolve, reject) => {
+          const reader = new FileReader();
+          reader.onload = (e2) => {
+            var _a2, _b2;
+            const base64String = (_b2 = (_a2 = e2.target) == null ? void 0 : _a2.result) == null ? void 0 : _b2.split(",")[1];
+            if (base64String) {
+              resolve({
+                contentType: file.type,
+                bytes: base64String
+              });
+            } else {
+              resolve(null);
+            }
+          };
+          reader.onerror = reject;
+          reader.readAsDataURL(file);
+        });
+      });
+      try {
+        const loadedImages = (yield Promise.all(imagePromises)).filter((img) => img !== null);
+        setSelectedImages((prev) => [...prev, ...loadedImages]);
+      } catch (error) {
+        console.error("Error processing pasted images:", error);
+      }
+    });
+    document.addEventListener("paste", handlePaste);
+    return () => document.removeEventListener("paste", handlePaste);
+  }, [imageUploadsEnabled]);
   (0, import_react13.useEffect)(() => {
     if (!(additionalInstructions == null ? void 0 : additionalInstructions.length)) {
       setChatInstructions(instructions || "");
@@ -2491,6 +2718,14 @@ function CopilotChat({
     onStopGeneration,
     onReloadMessages
   );
+  const handleSendMessage = (text) => {
+    const images = selectedImages;
+    setSelectedImages([]);
+    if (fileInputRef.current) {
+      fileInputRef.current.value = "";
+    }
+    return sendMessage(text, images);
+  };
   const chatContext = import_react13.default.useContext(ChatContext);
   const isVisible = chatContext ? chatContext.open : true;
   const handleRegenerate = (messageId) => {
@@ -2504,8 +2739,42 @@ function CopilotChat({
       onCopy(message);
     }
   };
-  return /* @__PURE__ */ (0, import_jsx_runtime22.jsxs)(WrappedCopilotChat, { icons, labels, className, children: [
-    /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(
+  const handleImageUpload = (event) => __async(this, null, function* () {
+    if (!event.target.files || event.target.files.length === 0) {
+      return;
+    }
+    const files = Array.from(event.target.files).filter((file) => file.type.startsWith("image/"));
+    if (files.length === 0)
+      return;
+    const fileReadPromises = files.map((file) => {
+      return new Promise((resolve, reject) => {
+        const reader = new FileReader();
+        reader.onload = (e) => {
+          var _a, _b;
+          const base64String = ((_b = (_a = e.target) == null ? void 0 : _a.result) == null ? void 0 : _b.split(",")[1]) || "";
+          if (base64String) {
+            resolve({
+              contentType: file.type,
+              bytes: base64String
+            });
+          }
+        };
+        reader.onerror = reject;
+        reader.readAsDataURL(file);
+      });
+    });
+    try {
+      const loadedImages = yield Promise.all(fileReadPromises);
+      setSelectedImages((prev) => [...prev, ...loadedImages]);
+    } catch (error) {
+      console.error("Error reading files:", error);
+    }
+  });
+  const removeSelectedImage = (index) => {
+    setSelectedImages((prev) => prev.filter((_, i) => i !== index));
+  };
+  return /* @__PURE__ */ (0, import_jsx_runtime24.jsxs)(WrappedCopilotChat, { icons, labels, className, children: [
+    /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(
       Messages2,
       {
         AssistantMessage: AssistantMessage2,
@@ -2514,32 +2783,51 @@ function CopilotChat({
         RenderActionExecutionMessage: RenderActionExecutionMessage2,
         RenderAgentStateMessage: RenderAgentStateMessage2,
         RenderResultMessage: RenderResultMessage2,
+        RenderImageMessage: RenderImageMessage2,
         messages: visibleMessages,
         inProgress: isLoading,
         onRegenerate: handleRegenerate,
         onCopy: handleCopy,
         onThumbsUp,
         onThumbsDown,
-        children: currentSuggestions.length > 0 && /* @__PURE__ */ (0, import_jsx_runtime22.jsx)("div", { className: "suggestions", children: currentSuggestions.map((suggestion, index) => /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(
+        children: currentSuggestions.length > 0 && /* @__PURE__ */ (0, import_jsx_runtime24.jsx)("div", { className: "suggestions", children: currentSuggestions.map((suggestion, index) => /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(
           Suggestion,
           {
             title: suggestion.title,
             message: suggestion.message,
             partial: suggestion.partial,
             className: suggestion.className,
-            onClick: (message) => sendMessage(message)
+            onClick: (message) => handleSendMessage(message)
           },
           index
         )) })
       }
     ),
-    /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(
+    imageUploadsEnabled && /* @__PURE__ */ (0, import_jsx_runtime24.jsxs)(import_jsx_runtime24.Fragment, { children: [
+      /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(ImageUploadQueue, { images: selectedImages, onRemoveImage: removeSelectedImage }),
+      /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(
+        "input",
+        {
+          type: "file",
+          multiple: true,
+          ref: fileInputRef,
+          onChange: handleImageUpload,
+          accept: inputFileAccept,
+          style: { display: "none" }
+        }
+      )
+    ] }),
+    /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(
       Input2,
       {
         inProgress: isLoading,
-        onSend: sendMessage,
+        onSend: handleSendMessage,
         isVisible,
-        onStop: stopGeneration
+        onStop: stopGeneration,
+        onUpload: imageUploadsEnabled ? () => {
+          var _a;
+          return (_a = fileInputRef.current) == null ? void 0 : _a.click();
+        } : void 0
       }
     )
   ] });
@@ -2552,10 +2840,10 @@ function WrappedCopilotChat({
 }) {
   const chatContext = import_react13.default.useContext(ChatContext);
   if (!chatContext) {
-    return /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(ChatContextProvider, { icons, labels, open: true, setOpen: () => {
-    }, children: /* @__PURE__ */ (0, import_jsx_runtime22.jsx)("div", { className: `copilotKitChat ${className}`, children }) });
+    return /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(ChatContextProvider, { icons, labels, open: true, setOpen: () => {
+    }, children: /* @__PURE__ */ (0, import_jsx_runtime24.jsx)("div", { className: `copilotKitChat ${className}`, children }) });
   }
-  return /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(import_jsx_runtime22.Fragment, { children });
+  return /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(import_jsx_runtime24.Fragment, { children });
 }
 var SUGGESTIONS_DEBOUNCE_TIMEOUT = 1e3;
 var useCopilotChatLogic = (makeSystemMessage, onInProgress, onSubmitMessage, onStopGeneration, onReloadMessages) => {
@@ -2609,22 +2897,45 @@ var useCopilotChatLogic = (makeSystemMessage, onInProgress, onSubmitMessage, onS
     // global context
     visibleMessages.length == 0
   ]);
-  const sendMessage = (messageContent) => __async(void 0, null, function* () {
+  const sendMessage = (messageContent, imagesToUse) => __async(void 0, null, function* () {
+    const images = imagesToUse || [];
     abortSuggestions();
     setCurrentSuggestions([]);
-    const message = new import_runtime_client_gql4.TextMessage({
-      content: messageContent,
-      role: import_runtime_client_gql4.Role.User
-    });
-    if (onSubmitMessage) {
-      try {
-        yield onSubmitMessage(messageContent);
-      } catch (error) {
-        console.error("Error in onSubmitMessage:", error);
+    let firstMessage = null;
+    if (messageContent.trim().length > 0) {
+      const textMessage = new import_runtime_client_gql4.TextMessage({
+        content: messageContent,
+        role: import_runtime_client_gql4.Role.User
+      });
+      if (onSubmitMessage) {
+        try {
+          yield onSubmitMessage(messageContent);
+        } catch (error) {
+          console.error("Error in onSubmitMessage:", error);
+        }
+      }
+      yield appendMessage(textMessage, { followUp: images.length === 0 });
+      if (!firstMessage) {
+        firstMessage = textMessage;
       }
     }
-    appendMessage(message);
-    return message;
+    if (images.length > 0) {
+      for (let i = 0; i < images.length; i++) {
+        const imageMessage = new import_runtime_client_gql4.ImageMessage({
+          format: images[i].contentType.replace("image/", ""),
+          bytes: images[i].bytes,
+          role: import_runtime_client_gql4.Role.User
+        });
+        yield appendMessage(imageMessage, { followUp: i === images.length - 1 });
+        if (!firstMessage) {
+          firstMessage = imageMessage;
+        }
+      }
+    }
+    if (!firstMessage) {
+      return new import_runtime_client_gql4.TextMessage({ content: "", role: import_runtime_client_gql4.Role.User });
+    }
+    return firstMessage;
   });
   const messages = visibleMessages;
   const { setMessages } = messagesContext;
@@ -2717,7 +3028,7 @@ var useCopilotChatLogic = (makeSystemMessage, onInProgress, onSubmitMessage, onS
 };
 
 // src/components/chat/Modal.tsx
-var import_jsx_runtime23 = require("react/jsx-runtime");
+var import_jsx_runtime25 = require("react/jsx-runtime");
 var CopilotModal = ({
   instructions,
   defaultOpen = false,
@@ -2751,19 +3062,19 @@ var CopilotModal = ({
     onSetOpen == null ? void 0 : onSetOpen(open);
     setOpenState(open);
   };
-  return /* @__PURE__ */ (0, import_jsx_runtime23.jsxs)(ChatContextProvider, { icons, labels, open: openState, setOpen, children: [
+  return /* @__PURE__ */ (0, import_jsx_runtime25.jsxs)(ChatContextProvider, { icons, labels, open: openState, setOpen, children: [
     children,
-    /* @__PURE__ */ (0, import_jsx_runtime23.jsxs)("div", { className, children: [
-      /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(Button2, {}),
-      /* @__PURE__ */ (0, import_jsx_runtime23.jsxs)(
+    /* @__PURE__ */ (0, import_jsx_runtime25.jsxs)("div", { className, children: [
+      /* @__PURE__ */ (0, import_jsx_runtime25.jsx)(Button2, {}),
+      /* @__PURE__ */ (0, import_jsx_runtime25.jsxs)(
         Window2,
         {
           clickOutsideToClose,
           shortcut,
           hitEscapeToClose,
           children: [
-            /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(Header2, {}),
-            /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(
+            /* @__PURE__ */ (0, import_jsx_runtime25.jsx)(Header2, {}),
+            /* @__PURE__ */ (0, import_jsx_runtime25.jsx)(
               CopilotChat,
               {
                 instructions,