[WIP] feat: 自动更新

ChuanhuChatbot.py

@@ -282,6 +282,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
                         )
                         changeProxyBtn = gr.Button(i18n("🔄 设置代理地址"))
                         default_btn = gr.Button(i18n("🔙 恢复默认设置"))
+                        updateChuanhuBtn = gr.Button(visible=False, elem_classes="invisible_btn", elem_id="update_chuanhu_btn")
 
     gr.Markdown(CHUANHU_DESCRIPTION, elem_id="description")
     gr.HTML(get_html("footer.html").format(versions=versions_html()), elem_id="footer")
@@ -483,6 +484,14 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     )
     checkUpdateBtn.click(fn=None, _js='()=>{manualCheckUpdate();}')
 
+    # Invisible elements
+    updateChuanhuBtn.click(
+        update_chuanhu,
+        [],
+        [status_display],
+        show_progress=True,
+    )
+
 logging.info(
     colorama.Back.GREEN
     + "\n川虎的温馨提示：访问 http://localhost:7860 查看界面"

assets/custom.js

@@ -22,6 +22,7 @@ var updateToast = null;
 var sendBtn = null;
 var cancelBtn = null;
 var sliders = null;
+var updateChuanhuBtn = null;
 
 var userLogged = false;
 var usernameGotten = false;
@@ -87,6 +88,7 @@ function gradioLoaded(mutations) {
             sendBtn = document.getElementById("submit_btn");
             cancelBtn = document.getElementById("cancel_btn");
             sliders = document.querySelectorAll('input[type="range"]');
+            updateChuanhuBtn = document.getElementById("update_chuanhu_btn");
 
             if (loginUserForm) {
                 localStorage.setItem("userLogged", true);

assets/html/update.html

@@ -15,7 +15,7 @@
         </div>
         <div id="goto-update-btn" class="btn-update-group">
             <button class="btn-update lg secondary svelte-1ipelgc" id="cancel-button" onclick="cancelUpdate()">{cancel_btn}</button>
-            <button class="btn-update lg primary svelte-1ipelgc" id="update-button" onclick="getUpdate()">{update_btn}</button>
+            <button class="btn-update lg primary svelte-1ipelgc" id="update-button" onclick="getUpdate(); updateChuanhuBtn.click();">{update_btn}</button>
         </div>
         <div id="close-update-btn" class="btn-update-group hideK">
             <button class="btn-update lg secondary svelte-1jrzxu" id="update-button" onclick="getUpdate()">{seenew_btn}</button>

modules/utils.py

@@ -590,6 +590,16 @@ def version_time():
         commit_time = "unknown"
     return commit_time
 
+def update_chuanhu():
+    git = os.environ.get('GIT', "git")
+    try:
+        run(f"{git} fetch --all && {git} stash && {git} pull https://github.com/GaiZhenbiao/ChuanhuChatGPT.git main -f && {git} stash pop")
+        logging.info("更新成功，重启中……")
+        return gr.Markdown.update(value="更新成功，重启中……")
+    except Exception:
+        logging.info("更新失败")
+        return gr.Markdown.update(value="更新失败")
+
 def get_html(filename):
     path = os.path.join(shared.chuanhu_path, "assets", "html", filename)
     if os.path.exists(path):