From 37b117a84c2fca1ad27ecb279aebe5c1f75e1b73 Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Mon, 5 Aug 2024 18:00:04 +0200
Subject: [PATCH] fix: styling

---
 .../chat/Messages/ResponseMessage.svelte      |  2 +-
 src/lib/components/common/Image.svelte        | 19 ++++++++++++-------
 2 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/src/lib/components/chat/Messages/ResponseMessage.svelte b/src/lib/components/chat/Messages/ResponseMessage.svelte
index 10ded00fd..d6245d83d 100644
--- a/src/lib/components/chat/Messages/ResponseMessage.svelte
+++ b/src/lib/components/chat/Messages/ResponseMessage.svelte
@@ -411,7 +411,7 @@
 				{/if}
 
 				<div
-					class="prose chat-{message.role} w-full max-w-full dark:prose-invert whitespace-pre-line"
+					class="prose chat-{message.role} w-full max-w-full dark:prose-invert prose-p:my-0 prose-img:my-0 whitespace-pre-line"
 				>
 					<div>
 						{#if (message?.statusHistory ?? [...(message?.status ? [message?.status] : [])]).length > 0}
diff --git a/src/lib/components/common/Image.svelte b/src/lib/components/common/Image.svelte
index 99f1eb405..020e016e8 100644
--- a/src/lib/components/common/Image.svelte
+++ b/src/lib/components/common/Image.svelte
@@ -5,21 +5,26 @@
 	export let src = '';
 	export let alt = '';
 
-	let _src = '';
+	export let className = '';
 
+	let _src = '';
 	$: _src = src.startsWith('/') ? `${WEBUI_BASE_URL}${src}` : src;
 
 	let showImagePreview = false;
 </script>
 
-<div>
-	<button
+<div class={className}>
+	<!-- svelte-ignore a11y-click-events-have-key-events -->
+	<!-- svelte-ignore a11y-no-noninteractive-element-interactions -->
+	<img
 		on:click={() => {
 			showImagePreview = true;
 		}}
-		class=" w-fit"
-	>
-		<img src={_src} {alt} class=" max-h-96 rounded-lg w-full" draggable="false" data-cy="image" />
-	</button>
+		src={_src}
+		{alt}
+		class=" rounded-lg"
+		draggable="false"
+		data-cy="image"
+	/>
 </div>
 <ImagePreview bind:show={showImagePreview} src={_src} {alt} />