feat: config中加入是否启用更新检查

ChuanhuChatbot.py

@@ -38,7 +38,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
         status_display = gr.Markdown(get_geoip(), elem_id="status_display")
     with gr.Row(elem_id="float_display"):
         user_info = gr.Markdown(value="getting user info...", elem_id="user_info")
-        update_info = gr.HTML(get_html("update.html").format(current_version=repo_html(),version_time=version_time()))
+        update_info = gr.HTML(get_html("update.html").format(current_version=repo_html(),version_time=version_time()), visible=check_update)
 
     with gr.Row().style(equal_height=True):
         with gr.Column(scale=5):
@@ -267,7 +267,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
                         )
                         changeProxyBtn = gr.Button(i18n("🔄 设置代理地址"))
                         default_btn = gr.Button(i18n("🔙 恢复默认设置"))
-                    checkUpdateBtn = gr.Button(i18n("🔄 检查更新..."))
+                    checkUpdateBtn = gr.Button(i18n("🔄 检查更新..."), visible=check_update)
 
     gr.Markdown(CHUANHU_DESCRIPTION, elem_id="description")
     gr.HTML(get_html("footer.html").format(versions=versions_html()), elem_id="footer")

assets/custom.js

@@ -24,6 +24,7 @@ var userLogged = false;
 var usernameGotten = false;
 var historyLoaded = false;
 var updateInfoGotten = false;
+var isLatestVersion = localStorage.getItem('isLatestVersion') || false;
 
 var ga = document.getElementsByTagName("gradio-app");
 var targetNode = ga[0];
@@ -94,7 +95,8 @@ function gradioLoaded(mutations) {
             }
             if (updateToast) {
                 const lastCheckTime = localStorage.getItem('lastCheckTime') || 0;
-                if (currentTime - lastCheckTime > 3 * 24 * 60 * 60 * 1000 && !updateInfoGotten) {
+                const longTimeNoCheck = currentTime - lastCheckTime > 3 * 24 * 60 * 60 * 1000;
+                if (longTimeNoCheck && !updateInfoGotten && !isLatestVersion || isLatestVersion && !updateInfoGotten ) {
                     updateLatestVersion();
                 }
             }
@@ -544,6 +546,7 @@ async function updateLatestVersion() {
         if (latestVersionTime) {
             if (localVersionTime < latestVersionTime) {
                 latestVersionElement.textContent = latestVersion;
+                console.log(`New version ${latestVersion} found!`);
                 if (!isInIframe) {openUpdateToast();}      
             } else {
                 noUpdate();
@@ -577,6 +580,8 @@ function manualCheckUpdate() {
     localStorage.setItem('lastCheckTime', currentTime);
 }
 function noUpdate() {
+    localStorage.setItem('isLatestVersion', 'true');
+    isLatestVersion = true;
     const versionInfoElement = document.getElementById('version-info-title');
     const releaseNoteWrap = document.getElementById('release-note-wrap');
     const gotoUpdateBtn = document.getElementById('goto-update-btn');

config_example.json

@@ -18,6 +18,7 @@
     "users": [], // 用户列表，[[用户名1, 密码1], [用户名2, 密码2], ...]
     "local_embedding": false, //是否在本地编制索引
     "hide_history_when_not_logged_in": false, //未登录情况下是否不展示对话历史
+    "check_update": true, //是否启用检查更新
     "default_model": "gpt-3.5-turbo", // 默认模型
 
     //川虎助理设置

modules/config.py

@@ -23,6 +23,7 @@ __all__ = [
     "server_name",
     "server_port",
     "share",
+    "check_update",
     "hide_history_when_not_logged_in",
     "default_chuanhu_assistant_model"
 ]
@@ -39,6 +40,7 @@ lang_config = config.get("language", "auto")
 language = os.environ.get("LANGUAGE", lang_config)
 
 hide_history_when_not_logged_in = config.get("hide_history_when_not_logged_in", False)
+check_update = config.get("check_update", True)
 
 if os.path.exists("api_key.txt"):
     logging.info("检测到api_key.txt文件，正在进行迁移...")