diff --git a/surfsense_web/app/dashboard/[search_space_id]/new-chat/[[...chat_id]]/page.tsx b/surfsense_web/app/dashboard/[search_space_id]/new-chat/[[...chat_id]]/page.tsx
index a6c01dd40..3f6893169 100644
--- a/surfsense_web/app/dashboard/[search_space_id]/new-chat/[[...chat_id]]/page.tsx
+++ b/surfsense_web/app/dashboard/[search_space_id]/new-chat/[[...chat_id]]/page.tsx
@@ -131,6 +131,7 @@ const TOOLS_WITH_UI = new Set([
 	"generate_podcast",
 	"generate_report",
 	"generate_video_presentation",
+	"display_image",
 	"generate_image",
 	"delete_notion_page",
 	"create_notion_page",
diff --git a/surfsense_web/components/assistant-ui/assistant-message.tsx b/surfsense_web/components/assistant-ui/assistant-message.tsx
index 8e1b455a2..781bc8e3a 100644
--- a/surfsense_web/components/assistant-ui/assistant-message.tsx
+++ b/surfsense_web/components/assistant-ui/assistant-message.tsx
@@ -55,7 +55,7 @@ const AssistantMessageInner: FC = () => {
 								generate_report: GenerateReportToolUI,
 								generate_podcast: GeneratePodcastToolUI,
 								generate_video_presentation: GenerateVideoPresentationToolUI,
-								display_image: () => null,
+								display_image: GenerateImageToolUI,
 								generate_image: GenerateImageToolUI,
 								save_memory: SaveMemoryToolUI,
 								recall_memory: RecallMemoryToolUI,
diff --git a/surfsense_web/components/public-chat/public-thread.tsx b/surfsense_web/components/public-chat/public-thread.tsx
index 99221ec2e..1510aa4ce 100644
--- a/surfsense_web/components/public-chat/public-thread.tsx
+++ b/surfsense_web/components/public-chat/public-thread.tsx
@@ -79,6 +79,7 @@ const UserAvatar: FC<AuthorMetadata & { hasError: boolean; onError: () => void }
 
 	if (avatarUrl && !hasError) {
 		return (
+			// biome-ignore lint/performance/noImgElement: external OAuth/profile avatar URL
 			<img
 				src={avatarUrl}
 				alt={displayName || "User"}
@@ -148,7 +149,7 @@ const PublicAssistantMessage: FC = () => {
 								generate_podcast: GeneratePodcastToolUI,
 								generate_report: GenerateReportToolUI,
 								generate_video_presentation: GenerateVideoPresentationToolUI,
-								display_image: () => null,
+								display_image: GenerateImageToolUI,
 								generate_image: GenerateImageToolUI,
 								link_preview: () => null,
 								multi_link_preview: () => null,