From 867749d7a35896bbcd2670e9cd07615a90377024 Mon Sep 17 00:00:00 2001
From: Henry Li <henry1943@163.com>
Date: Mon, 2 Feb 2026 10:02:31 +0800
Subject: [PATCH] feat: add file icons

---
 .../src/app/workspace/chats/[thread_id]/page.tsx |  2 +-
 .../components/ai-elements/chain-of-thought.tsx  |  5 ++++-
 .../workspace/artifacts/artifact-file-list.tsx   |  2 +-
 .../components/workspace/artifacts/context.tsx   |  4 +++-
 frontend/src/core/utils/files.tsx                | 16 ++++++++++++++--
 5 files changed, 23 insertions(+), 6 deletions(-)

diff --git a/frontend/src/app/workspace/chats/[thread_id]/page.tsx b/frontend/src/app/workspace/chats/[thread_id]/page.tsx
index 1f9cd26..6f9db87 100644
--- a/frontend/src/app/workspace/chats/[thread_id]/page.tsx
+++ b/frontend/src/app/workspace/chats/[thread_id]/page.tsx
@@ -229,7 +229,7 @@ export default function ChatPage() {
                 )}
               </div>
             </header>
-            <main className="flex min-h-0 grow flex-col">
+            <main className="flex min-h-0 max-w-full grow flex-col">
               <div className="flex size-full justify-center">
                 <MessageList
                   className={cn("size-full", !isNewThread && "pt-10")}
diff --git a/frontend/src/components/ai-elements/chain-of-thought.tsx b/frontend/src/components/ai-elements/chain-of-thought.tsx
index b0fde74..cd3a6b7 100644
--- a/frontend/src/components/ai-elements/chain-of-thought.tsx
+++ b/frontend/src/components/ai-elements/chain-of-thought.tsx
@@ -167,7 +167,10 @@ export type ChainOfThoughtSearchResultsProps = ComponentProps<"div">;
 export const ChainOfThoughtSearchResults = memo(
   ({ className, ...props }: ChainOfThoughtSearchResultsProps) => (
     <div
-      className={cn("flex flex-wrap items-center gap-2", className)}
+      className={cn(
+        "flex flex-wrap items-center gap-2 overflow-x-hidden",
+        className,
+      )}
       {...props}
     />
   ),
diff --git a/frontend/src/components/workspace/artifacts/artifact-file-list.tsx b/frontend/src/components/workspace/artifacts/artifact-file-list.tsx
index 844545b..afcde02 100644
--- a/frontend/src/components/workspace/artifacts/artifact-file-list.tsx
+++ b/frontend/src/components/workspace/artifacts/artifact-file-list.tsx
@@ -83,7 +83,7 @@ export function ArtifactFileList({
             <CardTitle className="relative pl-8">
               <div>{getFileName(file)}</div>
               <div className="absolute top-2 -left-0.5">
-                {getFileIcon(file)}
+                {getFileIcon(file, "size-6")}
               </div>
             </CardTitle>
             <CardDescription className="pl-8 text-xs">
diff --git a/frontend/src/components/workspace/artifacts/context.tsx b/frontend/src/components/workspace/artifacts/context.tsx
index 2428d86..22ccb1d 100644
--- a/frontend/src/components/workspace/artifacts/context.tsx
+++ b/frontend/src/components/workspace/artifacts/context.tsx
@@ -37,7 +37,9 @@ export function ArtifactsProvider({ children }: ArtifactsProviderProps) {
 
   const select = (artifact: string, autoSelect = false) => {
     setSelectedArtifact(artifact);
-    setSidebarOpen(false);
+    if (env.NEXT_PUBLIC_STATIC_WEBSITE_ONLY !== "true") {
+      setSidebarOpen(false);
+    }
     if (!autoSelect) {
       setAutoSelect(false);
     }
diff --git a/frontend/src/core/utils/files.tsx b/frontend/src/core/utils/files.tsx
index 8f9b52f..652e132 100644
--- a/frontend/src/core/utils/files.tsx
+++ b/frontend/src/core/utils/files.tsx
@@ -5,6 +5,7 @@ import {
   FileCogIcon,
   FilePlayIcon,
   FileTextIcon,
+  ImageIcon,
 } from "lucide-react";
 
 const extensionMap: Record<string, string> = {
@@ -191,17 +192,28 @@ export function getFileExtensionDisplayName(filepath: string) {
   }
 }
 
-export function getFileIcon(filepath: string) {
+export function getFileIcon(filepath: string, className?: string) {
   const extension = getFileExtension(filepath);
   const { isCodeFile } = checkCodeFile(filepath);
-  const className = "size-6";
   switch (extension) {
     case "skill":
       return <FileCogIcon className={className} />;
     case "html":
       return <CompassIcon className={className} />;
+    case "txt":
     case "md":
       return <BookOpenTextIcon className={className} />;
+    case "jpg":
+    case "jpeg":
+    case "png":
+    case "gif":
+    case "bmp":
+    case "tiff":
+    case "ico":
+    case "webp":
+    case "svg":
+    case "heic":
+      return <ImageIcon className={className} />;
     case "mp3":
     case "wav":
     case "ogg":