From 243f1a5ea53c8a9fc199ac756f6c9f17e71ab27a Mon Sep 17 00:00:00 2001
From: heimoshuiyu <heimoshuiyu@gmail.com>
Date: Fri, 2 Aug 2024 16:41:46 +0800
Subject: [PATCH] Update ChatBOX component to display total tokens in the short
 status bar

---
 src/chatbox.tsx | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/chatbox.tsx b/src/chatbox.tsx
index 61dc3d6..b91f645 100644
--- a/src/chatbox.tsx
+++ b/src/chatbox.tsx
@@ -522,7 +522,7 @@ export default function ChatBOX(props: {
           </div>
         </div>
         <div
-          class="navbar-center cursor-pointer"
+          class="navbar-center cursor-pointer py-1"
           onClick={() => {
             setShowSettings(true);
           }}
@@ -587,9 +587,11 @@ export default function ChatBOX(props: {
 
           {/* the short status bar */}
           <div class="indicator lg:hidden">
-            <span class="indicator-item badge badge-primary">
-              {chatStore.streamMode ? Tr("STREAM") : Tr("FETCH")}
-            </span>
+            {chatStore.totalTokens !== 0 && (
+              <span class="indicator-item badge badge-primary">
+                Tokens: {chatStore.totalTokens}
+              </span>
+            )}
             <a class="btn btn-ghost text-base sm:text-xl p-0">
               <SparklesIcon className="h-4 w-4 hidden sm:block" />
               {chatStore.model}