53 lines
1.8 KiB
JavaScript
53 lines
1.8 KiB
JavaScript
import { app } from "../../scripts/app.js";
|
|
|
|
app.registerExtension({
|
|
name: "BiRefNet_Hugo",
|
|
async setup() {
|
|
console.log("BiRefNet_Hugo loaded");
|
|
|
|
function queue_button_pressed() {
|
|
console.log("Queue button was pressed!");
|
|
}
|
|
|
|
document.getElementById("queue-button").addEventListener("click", queue_button_pressed);
|
|
},
|
|
|
|
async nodeCreated(node) {
|
|
if (node.title === "🔥BiRefNet") {
|
|
// 查找 load_local_model 和 local_model_path 的 widgets
|
|
let loadLocalModelWidget = node.widgets.find(widget => widget.name === "load_local_model");
|
|
let localModelPathWidget = node.widgets.find(widget => widget.name === "local_model_path");
|
|
|
|
// 确保找到相关控件
|
|
if (!loadLocalModelWidget || !localModelPathWidget) {
|
|
console.error("Required widgets not found: load_local_model or local_model_path.");
|
|
return;
|
|
}
|
|
|
|
// 初始状态控制：根据 load_local_model 的值决定是否显示 local_model_path
|
|
if (!loadLocalModelWidget.value) {
|
|
node.widgets = node.widgets.filter(widget => widget.name !== "local_model_path");
|
|
}
|
|
|
|
// 当 load_local_model 变化时，动态更新 local_model_path 的显示或隐藏
|
|
loadLocalModelWidget.callback = function() {
|
|
console.log("load_local_model changed:", loadLocalModelWidget.value);
|
|
|
|
// 移除或添加 local_model_path 控件
|
|
if (loadLocalModelWidget.value) {
|
|
if (!node.widgets.includes(localModelPathWidget)) {
|
|
node.widgets.push(localModelPathWidget); // 显示 local_model_path
|
|
}
|
|
} else {
|
|
node.widgets = node.widgets.filter(widget => widget.name !== "local_model_path"); // 隐藏 local_model_path
|
|
}
|
|
};
|
|
}
|
|
|
|
},
|
|
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
// 在注册节点前的逻辑
|
|
}
|
|
});
|