Support image uploads in chat input

13point5 · 13point5 · commit 3fb84d25d9fc · 2024-07-21T17:59:36.000-04:00
diff --git a/components/chat/input.tsx b/components/chat/input.tsx
@@ -5,7 +5,9 @@ import {
   ArrowUpIcon,
   Loader2Icon,
   MicIcon,
+  PaperclipIcon,
   PauseIcon,
+  UploadIcon,
   XIcon,
 } from "lucide-react";
 import { useEffect, useRef, useState } from "react";
@@ -41,6 +43,7 @@ export type Props = {
   onStopRecord: () => void;
   attachments: Attachment[];
   onRemoveAttachment: (attachment: Attachment) => void;
+  onAddAttachment: (newAttachments: Attachment[]) => void;
 };
 
 export const ChatInput = ({
@@ -53,12 +56,44 @@ export const ChatInput = ({
   onStopRecord,
   attachments,
   onRemoveAttachment,
+  onAddAttachment,
 }: Props) => {
   const inputRef = useRef<HTMLTextAreaElement>(null);
   const { onKeyDown } = useEnterSubmit({
     onSubmit,
   });
   const [model, setModel] = useState<Models>(getSettings().model);
+  const fileInputRef = useRef<HTMLInputElement>(null);
+
+  const handleFileUpload = () => {
+    fileInputRef.current?.click();
+  };
+
+  const convertToBase64 = (file: File): Promise<string> => {
+    return new Promise((resolve, reject) => {
+      const reader = new FileReader();
+      reader.readAsDataURL(file);
+      reader.onload = () => resolve(reader.result as string);
+      reader.onerror = (error) => reject(error);
+    });
+  };
+
+  const handleFileChange = async (e: React.ChangeEvent<HTMLInputElement>) => {
+    if (e.target.files) {
+      const filesArray = Array.from(e.target.files);
+      const attachmentsPromises = filesArray.map(async (file) => {
+        const base64 = await convertToBase64(file);
+        return {
+          url: base64,
+          name: file.name,
+          contentType: file.type,
+        } as Attachment;
+      });
+
+      const newAttachments = await Promise.all(attachmentsPromises);
+      onAddAttachment(newAttachments);
+    }
+  };
 
   useEffect(() => {
     if (inputRef.current) {
@@ -107,6 +142,23 @@ export const ChatInput = ({
             onChange={(e) => setInput(e.target.value)}
           />
 
+          <input
+            type="file"
+            accept="image/*"
+            multiple
+            ref={fileInputRef}
+            style={{ display: "none" }}
+            onChange={handleFileChange}
+          />
+          <Button
+            variant="outline"
+            size="icon"
+            className="w-8 h-8 bg-transparent"
+            onClick={handleFileUpload}
+          >
+            <PaperclipIcon className="w-4 h-4" />
+          </Button>
+
           <TooltipProvider>
             <Tooltip>
               <TooltipTrigger asChild>
@@ -116,7 +168,8 @@ export const ChatInput = ({
                     recording ? onStopRecord() : onStartRecord();
                   }}
                   size="icon"
-                  className="w-8 h-8 disabled:pointer-events-auto"
+                  variant="outline"
+                  className="w-8 h-8 bg-transparent disabled:pointer-events-auto"
                 >
                   {recording ? (
                     <PauseIcon className="w-4 h-4" />
diff --git a/components/chat/panel.tsx b/components/chat/panel.tsx
@@ -156,6 +156,12 @@ export const ChatPanel = ({ id }: Props) => {
     });
   };
 
+  const handleAddAttachment: ChatInputProps["onAddAttachment"] = (
+    newAttachments
+  ) => {
+    setAttachments((prev) => [...prev, ...newAttachments]);
+  };
+
   const handleRemoveAttachment: ChatInputProps["onRemoveAttachment"] = (
     attachment
   ) => {
@@ -227,6 +233,7 @@ export const ChatPanel = ({ id }: Props) => {
             onStartRecord={startRecording}
             onStopRecord={stopRecording}
             attachments={attachments}
+            onAddAttachment={handleAddAttachment}
             onRemoveAttachment={handleRemoveAttachment}
           />
         </div>
diff --git a/components/ui/button.tsx b/components/ui/button.tsx
@@ -1,11 +1,11 @@
-import * as React from "react"
-import { Slot } from "@radix-ui/react-slot"
-import { cva, type VariantProps } from "class-variance-authority"
+import * as React from "react";
+import { Slot } from "@radix-ui/react-slot";
+import { cva, type VariantProps } from "class-variance-authority";
 
-import { cn } from "@/lib/utils"
+import { cn } from "@/lib/utils";
 
 const buttonVariants = cva(
-  "inline-flex items-center justify-center whitespace-nowrap rounded-md text-sm font-medium ring-offset-background transition-colors focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-ring focus-visible:ring-offset-2 disabled:pointer-events-none disabled:opacity-50",
+  "inline-flex items-center justify-center whitespace-nowrap rounded-md text-sm font-medium ring-offset-background transition-colors focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-ring focus-visible:ring-offset-2 disabled:pointer-events-none disabled:opacity-50 disabled:cursor-not-allowed",
   {
     variants: {
       variant: {
@@ -31,26 +31,26 @@ const buttonVariants = cva(
       size: "default",
     },
   }
-)
+);
 
 export interface ButtonProps
   extends React.ButtonHTMLAttributes<HTMLButtonElement>,
     VariantProps<typeof buttonVariants> {
-  asChild?: boolean
+  asChild?: boolean;
 }
 
 const Button = React.forwardRef<HTMLButtonElement, ButtonProps>(
   ({ className, variant, size, asChild = false, ...props }, ref) => {
-    const Comp = asChild ? Slot : "button"
+    const Comp = asChild ? Slot : "button";
     return (
       <Comp
         className={cn(buttonVariants({ variant, size, className }))}
         ref={ref}
         {...props}
       />
-    )
+    );
   }
-)
-Button.displayName = "Button"
+);
+Button.displayName = "Button";
 
-export { Button, buttonVariants }
+export { Button, buttonVariants };
