From 7f21854b9aea22af0c609d6abe4ae07e9c472e25 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Sun, 19 May 2024 13:32:43 +0800
Subject: [PATCH] fix: webui issue with image (#523)

---
 assets/arena.html      | 3 ++-
 assets/playground.html | 6 +++++-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/assets/arena.html b/assets/arena.html
index 07a0cb6..b50ee52 100644
--- a/assets/arena.html
+++ b/assets/arena.html
@@ -101,6 +101,7 @@
     .chat-message {
       display: flex;
       padding: 0.7rem;
+      margin-bottom: 0.7rem;
     }
 
     .chat-avatar svg {
@@ -170,7 +171,7 @@
     .message-toolbox {
       display: flex;
       position: absolute;
-      bottom: -1.25rem;
+      bottom: -1.4rem;
     }
 
     .copy-message-btn,
diff --git a/assets/playground.html b/assets/playground.html
index 420c679..d8c33f7 100644
--- a/assets/playground.html
+++ b/assets/playground.html
@@ -178,6 +178,7 @@
     .chat-message {
       display: flex;
       padding: 0.7rem;
+      margin-bottom: 0.7rem;
     }
 
     .chat-avatar svg {
@@ -247,7 +248,7 @@
     .message-toolbox {
       display: flex;
       position: absolute;
-      bottom: -1.25rem;
+      bottom: -1.4rem;
     }
 
     .copy-message-btn,
@@ -1193,6 +1194,9 @@
 
     function estimateTokenLength(input) {
       let tokenLength = 0;
+      if (Array.isArray(input)) {
+        input = input.map(v => v.text || "").join("");
+      }
 
       for (let i = 0; i < input.length; i++) {
         const charCode = input.charCodeAt(i);