diff --git a/src/lib/components/MessageCard.svelte b/src/lib/components/MessageCard.svelte
index 6b2c2e2..6380362 100644
--- a/src/lib/components/MessageCard.svelte
+++ b/src/lib/components/MessageCard.svelte
@@ -155,12 +155,26 @@
class:fork-selected={forkSelected}
>
- {#if isSource}
-
- {:else}
-
- {/if}
-
+ {#if Array.isArray(contentItems)}
+ {#each contentItems as item}
+ {#if item.type === 'text'}
+ {#if isSource}
+
+ {:else}
+
+ {/if}
+ {:else if item.type === 'image_url'}
+
+ {/if}
+ {/each}
+ {:else}
+ {#if isSource}
+
+ {:else}
+
+ {/if}
+ {/if}
+
{#if alternativesCount > 1}