From 66a5a903c5113da1d9e0a0eb1f8391c37d7459fa Mon Sep 17 00:00:00 2001
From: josc146 <josStorer@outlook.com>
Date: Sat, 20 May 2023 10:38:35 +0800
Subject: [PATCH] improve notifications

---
 frontend/src/_locales/zh-hans/main.json | 3 ++-
 frontend/src/pages/Chat.tsx             | 5 +++++
 frontend/src/pages/Models.tsx           | 2 +-
 3 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/frontend/src/_locales/zh-hans/main.json b/frontend/src/_locales/zh-hans/main.json
index a090ce9..6fc41b5 100644
--- a/frontend/src/_locales/zh-hans/main.json
+++ b/frontend/src/_locales/zh-hans/main.json
@@ -73,5 +73,6 @@
   "Hello! I'm RWKV, an open-source and commercially available large language model.": "你好! 我是RWKV, 一个开源可商用的大语言模型.",
   "This tool’s API is compatible with OpenAI API. It can be used with any ChatGPT tool you like. Go to the settings of some ChatGPT tool, replace the 'https://api.openai.com' part in the API address with '": "本工具的API与OpenAI API兼容. 因此可以配合任意你喜欢的ChatGPT工具使用. 打开某个ChatGPT工具的设置, 将API地址中的'https://api.openai.com'部分替换为'",
   "New Version Available": "新版本可用",
-  "Update": "更新"
+  "Update": "更新",
+  "Please click the button in the top right corner to start the model": "请点击右上角的按钮启动模型"
 }
\ No newline at end of file
diff --git a/frontend/src/pages/Chat.tsx b/frontend/src/pages/Chat.tsx
index f5485fc..aa96ca8 100644
--- a/frontend/src/pages/Chat.tsx
+++ b/frontend/src/pages/Chat.tsx
@@ -16,6 +16,7 @@ import {ToolTipButton} from '../components/ToolTipButton';
 import {ArrowCircleUp28Regular, Delete28Regular, RecordStop28Regular} from '@fluentui/react-icons';
 import {CopyButton} from '../components/CopyButton';
 import {ReadButton} from '../components/ReadButton';
+import {toast} from 'react-toastify';
 
 const userName = 'M E';
 const botName = 'A I';
@@ -88,6 +89,10 @@ const ChatPanel: FC = observer(() => {
     e.stopPropagation();
     if (e.type === 'click' || (e.keyCode === 13 && !e.shiftKey)) {
       e.preventDefault();
+      if (commonStore.modelStatus === ModelStatus.Offline) {
+        toast(t('Please click the button in the top right corner to start the model'), {type: 'warning'});
+        return;
+      }
       if (!message) return;
       onSubmit(message);
       setMessage('');
diff --git a/frontend/src/pages/Models.tsx b/frontend/src/pages/Models.tsx
index 995958b..47edf5b 100644
--- a/frontend/src/pages/Models.tsx
+++ b/frontend/src/pages/Models.tsx
@@ -134,7 +134,7 @@ const columns: TableColumnDefinition<ModelSourceItem>[] = [
             }
             {item.downloadUrl && !item.isLocal &&
               <ToolTipButton desc={t('Download')} icon={<ArrowDownload20Regular/>} onClick={() => {
-                toast(`${t('Downloading')} ${item.name}`);
+                toast(`${t('Downloading')} ${item.name}`, {type: 'info'});
                 DownloadFile(`./${manifest.localModelDir}/${item.name}`, item.downloadUrl!);
               }}/>}
             {item.url && <ToolTipButton desc={t('Open Url')} icon={<Open20Regular/>} onClick={() => {