From 25b247354f0c25fed87f82513122e5fb7ceadcce Mon Sep 17 00:00:00 2001
From: starmorph <dylan@starmorph.com>
Date: Mon, 19 May 2025 20:52:38 -0700
Subject: [PATCH] CR: cn utility, refactor accepted files, nextImage

---
 .../thread/ContentBlocksPreview.tsx           |  9 +++-
 src/components/thread/messages/human.tsx      |  4 +-
 src/components/ui/MultimodalPreview.tsx       | 45 +++++++++++--------
 src/hooks/use-file-upload.tsx                 |  7 +--
 4 files changed, 37 insertions(+), 28 deletions(-)

diff --git a/src/components/thread/ContentBlocksPreview.tsx b/src/components/thread/ContentBlocksPreview.tsx
index 1c1caa0..0ee6664 100644
--- a/src/components/thread/ContentBlocksPreview.tsx
+++ b/src/components/thread/ContentBlocksPreview.tsx
@@ -1,6 +1,7 @@
 import React from "react";
 import type { Base64ContentBlock } from "@langchain/core/messages";
 import { MultimodalPreview } from "../ui/MultimodalPreview";
+import { cn } from "@/lib/utils";
 
 interface ContentBlocksPreviewProps {
   blocks: Base64ContentBlock[];
@@ -9,15 +10,19 @@ interface ContentBlocksPreviewProps {
   className?: string;
 }
 
+/**
+ * Renders a preview of content blocks with optional remove functionality.
+ * Uses cn utility for robust class merging.
+ */
 export const ContentBlocksPreview: React.FC<ContentBlocksPreviewProps> = ({
   blocks,
   onRemove,
   size = "md",
-  className = "",
+  className,
 }) => {
   if (!blocks.length) return null;
   return (
-    <div className={`flex flex-wrap gap-2 p-3.5 pb-0 ${className}`}>
+    <div className={cn("flex flex-wrap gap-2 p-3.5 pb-0", className)}>
       {blocks.map((block, idx) => (
         <MultimodalPreview
           key={idx}
diff --git a/src/components/thread/messages/human.tsx b/src/components/thread/messages/human.tsx
index 836d1e3..e95cacc 100644
--- a/src/components/thread/messages/human.tsx
+++ b/src/components/thread/messages/human.tsx
@@ -138,9 +138,7 @@ export function HumanMessage({
               </div>
             )}
             {/* Render text if present, otherwise fallback to file/image name */}
-            {contentString &&
-            contentString !== "Other" &&
-            contentString !== "Multimodal message" ? (
+            {contentString ? (
               <p className="bg-muted ml-auto w-fit rounded-3xl px-4 py-2 text-right whitespace-pre-wrap">
                 {contentString}
               </p>
diff --git a/src/components/ui/MultimodalPreview.tsx b/src/components/ui/MultimodalPreview.tsx
index 714ec91..369b485 100644
--- a/src/components/ui/MultimodalPreview.tsx
+++ b/src/components/ui/MultimodalPreview.tsx
@@ -1,7 +1,8 @@
 import React from "react";
 import { File, Image as ImageIcon, X as XIcon } from "lucide-react";
 import type { Base64ContentBlock } from "@langchain/core/messages";
-
+import { cn } from "@/lib/utils";
+import Image from "next/image";
 export interface MultimodalPreviewProps {
   block: Base64ContentBlock;
   removable?: boolean;
@@ -14,7 +15,7 @@ export const MultimodalPreview: React.FC<MultimodalPreviewProps> = ({
   block,
   removable = false,
   onRemove,
-  className = "",
+  className,
   size = "md",
 }) => {
   // Sizing
@@ -38,13 +39,13 @@ export const MultimodalPreview: React.FC<MultimodalPreviewProps> = ({
     if (size === "sm") imgClass = "rounded-md object-cover h-10 w-10 text-base";
     if (size === "lg") imgClass = "rounded-md object-cover h-24 w-24 text-xl";
     return (
-      <div
-        className={`relative inline-block${className ? ` ${className}` : ""}`}
-      >
-        <img
+      <div className={cn("relative inline-block", className)}>
+        <Image
           src={url}
           alt={String(block.metadata?.name || "uploaded image")}
           className={imgClass}
+          width={size === "sm" ? 16 : size === "md" ? 32 : 48}
+          height={size === "sm" ? 16 : size === "md" ? 32 : 48}
         />
         {removable && (
           <button
@@ -68,20 +69,24 @@ export const MultimodalPreview: React.FC<MultimodalPreviewProps> = ({
   ) {
     const filename =
       block.metadata?.filename || block.metadata?.name || "PDF file";
-    const fileClass = `relative flex items-center gap-2 rounded-md border bg-gray-100 px-3 py-2${className ? ` ${className}` : ""}`;
     return (
-      <div className={fileClass}>
+      <div
+        className={cn(
+          "relative flex items-center gap-2 rounded-md border bg-gray-100 px-3 py-2",
+          className
+        )}
+      >
         <File
-          className={
-            "flex-shrink-0 text-teal-700 " +
-            (size === "sm" ? "h-5 w-5" : "h-7 w-7")
-          }
+          className={cn(
+            "flex-shrink-0 text-teal-700",
+            size === "sm" ? "h-5 w-5" : "h-7 w-7"
+          )}
         />
         <span
-          className={
-            "truncate text-sm text-gray-800 " +
-            (size === "sm" ? "max-w-[80px]" : "max-w-[160px]")
-          }
+          className={cn(
+            "truncate text-sm text-gray-800",
+            size === "sm" ? "max-w-[80px]" : "max-w-[160px]"
+          )}
         >
           {String(filename)}
         </span>
@@ -100,9 +105,13 @@ export const MultimodalPreview: React.FC<MultimodalPreviewProps> = ({
   }
 
   // Fallback for unknown types
-  const fallbackClass = `flex items-center gap-2 rounded-md border bg-gray-100 px-3 py-2 text-gray-500${className ? ` ${className}` : ""}`;
   return (
-    <div className={fallbackClass}>
+    <div
+      className={cn(
+        "flex items-center gap-2 rounded-md border bg-gray-100 px-3 py-2 text-gray-500",
+        className
+      )}
+    >
       <File className="h-5 w-5 flex-shrink-0" />
       <span className="truncate text-xs">Unsupported file type</span>
       {removable && (
diff --git a/src/hooks/use-file-upload.tsx b/src/hooks/use-file-upload.tsx
index 392cff3..27988e1 100644
--- a/src/hooks/use-file-upload.tsx
+++ b/src/hooks/use-file-upload.tsx
@@ -3,14 +3,11 @@ import { toast } from "sonner";
 import type { Base64ContentBlock } from "@langchain/core/messages";
 import { fileToContentBlock } from "@/lib/multimodal-utils";
 
-export const SUPPORTED_IMAGE_TYPES = [
+export const SUPPORTED_FILE_TYPES = [
   "image/jpeg",
   "image/png",
   "image/gif",
   "image/webp",
-];
-export const SUPPORTED_FILE_TYPES = [
-  ...SUPPORTED_IMAGE_TYPES,
   "application/pdf",
 ];
 
@@ -26,7 +23,7 @@ export function useFileUpload({
   const dropRef = useRef<HTMLDivElement>(null);
 
   const isDuplicate = (file: File, blocks: Base64ContentBlock[]) => {
-    if (SUPPORTED_IMAGE_TYPES.includes(file.type)) {
+    if (SUPPORTED_FILE_TYPES.includes(file.type)) {
       return blocks.some(
         (b) =>
           b.type === "image" &&