diff --git a/interface.php b/interface.php
index f0544a4..8282b7c 100644
--- a/interface.php
+++ b/interface.php
@@ -383,44 +383,34 @@ function handleKeydownUserPost(event){
 			}
 	
 			const decodedData = new TextDecoder().decode(value);
+			console.log(decodedData);
 			let chunks = decodedData.split("data: ");
 			chunks.forEach((chunk, index) => {
 				if(chunk.indexOf('finish_reason":"stop"') > 0) return false;
 				if(chunk.indexOf('DONE') > 0) return false;
 				if(chunk.indexOf('role') > 0) return false;
-				if(chunk.length === 0) return false;
-				// First check if chunk is valid json.
-				// Otherwise we do not see the correct error message.
-				try {
-					const json = JSON.parse(chunk);
-					if ("choices" in json) {
-						// console.log(json["choices"]);
-						// normal response
-						document.querySelector(".message:last-child").querySelector(".message-text").innerHTML +=
-							json["choices"][0]["delta"].content;
-					} else {
-						if ("error" in json) {
-							if ("message" in json.error) {
-								// console.log(json.error.message);
-								document.querySelector(".message:last-child").querySelector(".message-text").innerHTML =
-									'<em>' + json.error.message + '</em>';
-							} else {
-								console.log(json.error);
-							}
-						} else {
-							console.log(json);
-						}
-					}
-				} catch(error) {
-					console.log(chunk);
-					console.error(error.message);
-				}
+				if(chunk.length == 0) return false;
+				if(chunk != "") console.log(JSON.parse(chunk)["choices"][0]["delta"])
+				console.log(JSON.parse(chunk)["choices"][0]["delta"]);
+				document.querySelector(".message:last-child").querySelector(".message-text").innerHTML +=  escapeHTML(JSON.parse(chunk)["choices"][0]["delta"].content);
 			})
 
+			// Check if the content has code block
+			document.querySelector(".message:last-child").querySelector(".message-text").innerHTML = document.querySelector(".message:last-child").querySelector(".message-text").innerHTML.replace(/```([\s\S]+?)```/g, '<pre><code>$1</code></pre>').replace(/\*\*.*?\*\*/g, '');;
+			hljs.highlightAll();
 			scrollToLast();
 		}
 	}
 
+	function escapeHTML(str) {
+    return str.replace(/&/g, '&amp;')
+              .replace(/</g, '&lt;')
+              .replace(/>/g, '&gt;')
+              .replace(/"/g, '&quot;')
+              .replace(/'/g, '&#039;');
+}
+	}
+
 	
 	function addMessage(message){
 		const messagesElement = document.querySelector(".messages");