Update assets/custom.js

assets/custom.js

@@ -16,15 +16,18 @@ var chatbot = null;
 
 var ga = document.getElementsByTagName("gradio-app");
 var targetNode = ga[0];
-var metaTags = document.getElementsByTagName('meta');
 var isInHuggingFace = false;
 
-for (var i = 0; i < metaTags.length; i++) {
-    var tag = metaTags[i];
-    if (tag.getAttribute('name') === 'twitter:site' && tag.getAttribute('content') === '@huggingface') {
-        isInHuggingFace = true;
+// 您是抱抱脸吗?
+function areYouHuggingFace() {
+    const metaTags = document.getElementsByTagName('meta');
+    for (var i = 0; i < metaTags.length; i++) {
+        var tag = metaTags[i];
+        if (tag.getAttribute('name') === 'twitter:site' && tag.getAttribute('content') === '@huggingface') {
+            isInHuggingFace = true;
+        }
+        break;
     }
-    break;
 }
 
 // gradio 页面加载好了么??? 我能动你的元素了么??
@@ -34,6 +37,7 @@ function gradioLoaded(mutations) {
             gradioContainer = document.querySelector(".gradio-container");
             user_input_tb = document.getElementById('user_input_tb');
             userInfoDiv = document.getElementById("user_info");
+            appTitleDiv = document.getElementById("app_title");
             chatbot = document.querySelector('#chuanhu_chatbot');
 
             if (gradioContainer) {  // gradioCainter 加载出来了没?
@@ -42,7 +46,7 @@ function gradioLoaded(mutations) {
             if (user_input_tb) {  // user_input_tb 加载出来了没?
                 selectHistory();
             }
-            if (userInfoDiv) {  // userInfoDiv 加载出来了没?
+            if (userInfoDiv && appTitleDiv) {  // userInfoDiv 加载出来了没?
                 setTimeout(showOrHideUserInfo(), 2000);
             }
             if (chatbot) {  // chatbot 加载出来了没?
@@ -110,8 +114,6 @@ function toggleUserInfoVisibility(shouldHide) {
     }
 }
 function showOrHideUserInfo() {
-    userInfoDiv = document.getElementById("user_info");
-    appTitleDiv = document.getElementById("app_title");
     var sendBtn = document.getElementById("submit_btn");
 
     // Bind mouse/touch events to show/hide user info
@@ -217,6 +219,9 @@ var observer = new MutationObserver(function (mutations) {
 observer.observe(targetNode, { childList: true, subtree: true });
 
 // 监视页面变化
+window.addEventListener("DOMContentLoaded", function () {
+    areYouHuggingFace();
+});
 window.addEventListener('resize', setChatbotHeight);
 window.addEventListener('scroll', setChatbotHeight);
 window.matchMedia("(prefers-color-scheme: dark)").addEventListener("change", adjustDarkMode);