fix convert tool (#77)

* 服务器模式下导入自动调用convert

* 删除多余日志

Author: dfhfg123

bizyui/js/bizyair_tools.js

@@ -1,36 +1,60 @@
 import { app, ComfyApp } from "../../scripts/app.js";
 import { api } from "../../../scripts/api.js";
 
-app.registerExtension({
-	name: "bizyair.tool",
-	setup() {
+async function handleFile(json_data) {
+    const jsonContent = json_data
 
-        async function handleFile(json_data) {
-            const jsonContent = json_data
+    await app.loadGraphData(
+        jsonContent,
+        true,
+        false,
+        "convert_test"
+      );
+}
 
-            await app.loadGraphData(
-                jsonContent,
-                true,
-                false,
-                "convert_test"
-              );
+// 检查是否为服务器模式
+async function isServerMode() {
+        const serverModeResponse = await fetch("/bizyair/server_mode");
+        const serverModeData = await serverModeResponse.json();
+        return serverModeData.data.server_mode === true;
+}
 
-        }
-        async function convert(){
-            const p2 = await app.graphToPrompt();
-            const json = JSON.stringify(p2["workflow"], null, 2);
+async function convert(){
+    const p2 = await app.graphToPrompt();
+    const json = JSON.stringify(p2["workflow"], null, 2);
+    await api.fetchApi("/bizyair/node_converter", {
+        method: "POST",
+        headers: {
+            "Content-Type": "application/json",
+        },
+        body: json
+      }).then(response => response.json())
+      .then(data => handleFile(data))
+      .catch(error => console.error("Error:", error));
+}
 
-            await api.fetchApi("/bizyair/node_converter", {
-                method: "POST",
-                headers: {
-                    "Content-Type": "application/json",
-                },
-                body: json
-              }).then(response => response.json())
-              .then(data => handleFile(data))
-              .catch(error => console.error("Error:", error));
-        }
-        // Add canvas menu options
+// 全局变量，用于节流控制
+let lastConvertTime = 0;
+const MIN_CONVERT_INTERVAL = 3000; // 最小间隔3秒
+
+// 节流函数确保convert不会被频繁调用
+async function throttledConvert() {
+    const serverMode = await isServerMode();
+    if (!serverMode) {
+        return;
+    }
+    const now = Date.now();
+    if (now - lastConvertTime > MIN_CONVERT_INTERVAL) {
+        lastConvertTime = now;;
+        convert();
+    } 
+}
+
+app.registerExtension({
+    name: "bizyair.tool",
+    setup() {
+        console.log('BizyAir Tools extension setup');      
+        // 添加菜单选项
         const orig = LGraphCanvas.prototype.getCanvasMenuOptions;
         LGraphCanvas.prototype.getCanvasMenuOptions = function () {
             const options = orig.apply(this, arguments);
@@ -48,7 +72,33 @@ app.registerExtension({
                 },
             });
             return options;
-        };
-
-	},
+        };             
+        // 监听导入
+        const origLoadGraphData = app.loadGraphData;
+        if (origLoadGraphData) {
+            app.loadGraphData = async function() {
+                const result = origLoadGraphData.apply(this, arguments);
+                const serverMode = await isServerMode();
+                if (serverMode) {
+                    setTimeout(() => {
+                        throttledConvert();
+                    }, 500); 
+                }            
+                return result;
+            };
+        }
+    },  
+    // 添加init钩子，在ComfyUI初始化后调用convert函数
+    init() {
+        console.log('BizyAir Tools initializing...');
+        lastConvertTime = Date.now(); // 记录初始化时间   
+        // 检查是否为服务器模式
+        setTimeout(async () => {
+            const serverMode = await isServerMode();
+            if (serverMode) {
+                console.log('服务器模式，页面初始化完成，调用convert');
+                convert();
+            } 
+        }, 500); 
+    }
 });