BREAKING: 默认渲染LaTeX公式

- 去除了页面中的勾选框，现在仅能在config中配置渲染与否
- 重写了js引入mathjax方法

ChuanhuChatbot.py

@@ -93,9 +93,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
                         autogpt_mode = gr.Checkbox(label=i18n("AutoGPT 模式"), value=True)
                         single_turn_checkbox = gr.Checkbox(label=i18n("单轮对话"), value=False)
                         use_websearch_checkbox = gr.Checkbox(label=i18n("使用在线搜索"), value=False)
-                        render_latex_checkbox = gr.Checkbox(
-                            label=i18n("渲染LaTeX公式"), value=render_latex, interactive=True, elem_id="render_latex_checkbox"
-                        )
+                        # render_latex_checkbox = gr.Checkbox(label=i18n("渲染LaTeX公式"), value=render_latex, interactive=True, elem_id="render_latex_checkbox")
                     language_select_dropdown = gr.Dropdown(
                         label=i18n("选择回复语言（针对搜索&索引功能）"),
                         choices=REPLY_LANGUAGES,

assets/custom.js

@@ -17,7 +17,7 @@ var chatbotWrap = null;
 var apSwitch = null;
 var empty_botton = null;
 var messageBotDivs = null;
-var renderLatex = null;
+// var renderLatex = null;
 var loginUserForm = null;
 var logginUser = null;
 
@@ -51,7 +51,7 @@ function gradioLoaded(mutations) {
             chatbot = document.querySelector('#chuanhu_chatbot');
             chatbotWrap = document.querySelector('#chuanhu_chatbot > .wrap');
             apSwitch = document.querySelector('.apSwitch input[type="checkbox"]');
-            renderLatex = document.querySelector("#render_latex_checkbox > label > input");
+            // renderLatex = document.querySelector("#render_latex_checkbox > label > input");
             empty_botton = document.getElementById("empty_btn")
 
             if (loginUserForm) {
@@ -80,10 +80,10 @@ function gradioLoaded(mutations) {
                 }
                 setChatbotScroll();
             }
-            if (renderLatex) {  // renderLatex 加载出来了没?
-                shouldRenderLatex = renderLatex.checked;
-                updateMathJax();
-            }
+            // if (renderLatex) {  // renderLatex 加载出来了没?
+            //     shouldRenderLatex = renderLatex.checked;
+            //     updateMathJax();
+            // }
             if (empty_botton) {
                 emptyHistory();
             }
@@ -98,7 +98,7 @@ function webLocale() {
         var forViewStyle = document.createElement('style');
         forViewStyle.innerHTML = '.wrap>.history-message>:last-child::after { content: "' + forView + '"!important; }';
         document.head.appendChild(forViewStyle);
-        console.log("added forViewStyle", forView);
+        // console.log("added forViewStyle", forView);
     }
 }
 
@@ -439,7 +439,7 @@ function renderMathJax() {
         }
     }
     mathjaxUpdated = true;
-    console.log("MathJax Rendered")
+    // console.log("MathJax Rendered")
 }
 
 function removeMathjax() {
@@ -455,22 +455,21 @@ function removeMathjax() {
 }
 
 function updateMathJax() {
-    renderLatex.addEventListener("change", function() {
-        shouldRenderLatex = renderLatex.checked;
-        // console.log(shouldRenderLatex)
-        if (!mathjaxUpdated) {
-            if (shouldRenderLatex) {
-                renderMathJax();
-            } else {
-                console.log("MathJax Disabled")
-                removeMathjax();
-            }
-        } else {
-            if (!shouldRenderLatex) {
-                mathjaxUpdated = false; // reset
-            }
-        }
-    });
+    // renderLatex.addEventListener("change", function() {
+    //     shouldRenderLatex = renderLatex.checked;
+    //     if (!mathjaxUpdated) {
+    //         if (shouldRenderLatex) {
+    //             renderMathJax();
+    //         } else {
+    //             console.log("MathJax Disabled")
+    //             removeMathjax();
+    //         }
+    //     } else {
+    //         if (!shouldRenderLatex) {
+    //             mathjaxUpdated = false; // reset
+    //         }
+    //     }
+    // });
     if (shouldRenderLatex && !mathjaxUpdated) {
         renderMathJax();
     }
@@ -534,7 +533,7 @@ var loadhistorytime = 0; // for debugging
 function saveHistoryHtml() {
     var historyHtml = document.querySelector('#chuanhu_chatbot > .wrap');
     localStorage.setItem('chatHistory', historyHtml.innerHTML);
-    console.log("History Saved")
+    // console.log("History Saved")
     historyLoaded = false;
 }
 function loadHistoryHtml() {
@@ -595,6 +594,7 @@ observer.observe(targetNode, { childList: true, subtree: true });
 window.addEventListener("DOMContentLoaded", function () {
     isInIframe = (window.self !== window.top);
     historyLoaded = false;
+    shouldRenderLatex = !!document.querySelector('script[src="https://cdnjs.cloudflare.com/ajax/libs/mathjax/2.7.5/MathJax.js?config=TeX-MML-AM_CHTML"]');
 });
 window.addEventListener('resize', setChatbotHeight);
 window.addEventListener('scroll', setChatbotHeight);

assets/external-scripts.js

@@ -1,21 +1,2 @@
 
-// window.MathJax = {
-//     tex: {
-//         inlineMath: [['$', '$'], ['\\(', '\\)']]
-//     },
-//     svg: {
-//         fontCache: 'global'
-//     }
-// };
-
-(function () {
-    var script = document.createElement('script');
-    script.src = "https://cdnjs.cloudflare.com/ajax/libs/mathjax/2.7.5/latest.js?config=TeX-MML-AM_CHTML";
-    script.async = true;
-    document.head.appendChild(script);
-
-    var config = document.createElement("script");
-    config.type = "text/x-mathjax-config";
-    config.text = "MathJax.Hub.Config({skipStartupTypeset: true, tex2jax: {inlineMath: [['$','$'], ['\\\\(','\\\\)']],displayMath: [['$$','$$'], ['\\\\[','\\\\]']]}});";
-    document.head.appendChild(config);
-})();
+// external javascript here

config_example.json

@@ -10,7 +10,7 @@
     // "https_proxy": "http://127.0.0.1:1079",
     // "http_proxy": "http://127.0.0.1:1079",
     // 是否默认渲染LaTeX
-    "render_latex": false,
+    "render_latex": true,
     "users": [], // 用户列表，[[用户名1, 密码1], [用户名2, 密码2], ...]
     "local_embedding": false, //是否在本地编制索引
     "hide_history_when_not_logged_in": false, //未登录情况下是否不展示对话历史

modules/config.py

@@ -76,7 +76,7 @@ my_api_key = os.environ.get("OPENAI_API_KEY", my_api_key)
 xmchat_api_key = config.get("xmchat_api_key", "")
 os.environ["XMCHAT_API_KEY"] = xmchat_api_key
 
-render_latex = config.get("render_latex", False)
+render_latex = config.get("render_latex", True)
 
 if render_latex:
     os.environ["RENDER_LATEX"] = "yes"

modules/overwrites.py

@@ -8,7 +8,7 @@ from gradio_client import utils as client_utils
 
 from modules.presets import *
 from modules.llama_func import *
-
+from modules.config import render_latex
 
 def compact_text_chunks(self, prompt: Prompt, text_chunks: List[str]) -> List[str]:
     logging.debug("Compacting text chunks...🚀🚀🚀")
@@ -85,6 +85,11 @@ with open("./assets/custom.js", "r", encoding="utf-8") as f, \
 def reload_javascript():
     print("Reloading javascript...")
     js = f'<script>{customJS}</script><script async>{externalScripts}</script>'
+    if render_latex:
+        js += """\
+            <script src="https://cdnjs.cloudflare.com/ajax/libs/mathjax/2.7.5/latest.js?config=TeX-MML-AM_CHTML"></script>
+            <script type="text/x-mathjax-config">MathJax.Hub.Config({skipStartupTypeset: false, tex2jax: {inlineMath: [['$','$'], ['\\(','\\)']],displayMath: [['$$','$$'], ['\\[','\\]']]}});</script>
+        """
     def template_response(*args, **kwargs):
         res = GradioTemplateResponseOriginal(*args, **kwargs)
         res.body = res.body.replace(b'</html>', f'{js}</html>'.encode("utf8"))