From 137d0140e10e917a2a41c8c762a8630cd5a3a6c8 Mon Sep 17 00:00:00 2001
From: Webifi <john@webifi.com>
Date: Wed, 14 Jun 2023 14:44:21 -0500
Subject: [PATCH] Show code highlighting while streaming

---
 src/lib/EditMessage.svelte | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/lib/EditMessage.svelte b/src/lib/EditMessage.svelte
index 672bd56..980c2c7 100644
--- a/src/lib/EditMessage.svelte
+++ b/src/lib/EditMessage.svelte
@@ -1,6 +1,6 @@
 <script lang="ts">
   import Code from './Code.svelte'
-  import { createEventDispatcher, onMount } from 'svelte'
+  import { afterUpdate, createEventDispatcher, onMount } from 'svelte'
   import { deleteMessage, chatsStorage, deleteSummaryMessage, truncateFromMessage, submitExitingPromptsNow, saveChatStore, continueMessage } from './Storage.svelte'
   import { getPrice } from './Stats.svelte'
   import SvelteMarkdown from 'svelte-markdown'
@@ -37,6 +37,7 @@
   let original:string
   let defaultModel:Model
   let imageUrl:string
+  let refreshCounter = 0
 
   onMount(() => {
     defaultModel = chatSettings.model
@@ -47,6 +48,10 @@
     }
   })
 
+  afterUpdate(() => {
+    if (message.content.slice(-5).match(/```/)) refreshCounter++
+  })
+
   const edit = () => {
     if (message.summarized || message.streaming || editing) return
     editing = true
@@ -224,11 +229,13 @@
         {#if message.summary && !message.summary.length}
         <p><b>Summarizing...</b></p>
         {/if}
+        {#key refreshCounter}
         <SvelteMarkdown 
           source={message.content} 
           options={markdownOptions} 
           renderers={{ code: Code, html: Code }}
         />
+        {/key}
         {#if imageUrl}
           <img src={imageUrl} alt="">
         {/if}