more image gen improvements

Author: 0xGingi

src/routes/api/generate-message/+server.ts

@@ -1948,14 +1948,15 @@ export const POST: RequestHandler = async ({ request }) => {
 					throw new Error('No valid image data returned from API');
 				}
 
-				const markdownContent = generatedImages
-					.map((img, idx) => `![Generated Image ${idx + 1}](${img.url})`)
-					.join('\n\n');
+				const textContent =
+					generatedImages.length > 1
+						? `Generated Images (${generatedImages.length})`
+						: 'Generated Image';
 
 				await db
 					.update(messages)
 					.set({
-						content: markdownContent,
+						content: textContent,
 						contentHtml: null,
 						tokenCount: 0,
 						images: generatedImages,

src/routes/chat/[id]/message.svelte

@@ -63,6 +63,11 @@
 				)
 			: []
 	);
+	const displayContent = $derived.by(() => {
+		if (safeImages.length === 0) return safeContent;
+		const stripped = safeContent.replace(/!\[[^\]]*]\([^)]*\)/g, '').trim();
+		return stripped.length > 0 ? stripped : 'Generated Image';
+	});
 
 	let imageModal = $state<{ open: boolean; imageUrl: string; fileName: string }>({
 		open: false,
@@ -453,7 +458,7 @@
 				{@html sanitizeHtml(message.contentHtml)}
 			{:else}
 				<svelte:boundary>
-					<MarkdownRenderer content={safeContent} />
+			<MarkdownRenderer content={displayContent} />
 
 					{#snippet failed(error)}
 						<div class="text-destructive">
@@ -557,7 +562,7 @@
 								if (audioPlayer.isPlaying && audioPlayer.currentMessageId === message.id) {
 									audioPlayer.stop();
 								} else {
-							audioPlayer.play(safeContent, message.id);
+							audioPlayer.play(displayContent, message.id);
 								}
 							}}
 							{...tooltip.trigger}