bizydraft 0.2.54__py3-none-any.whl → 0.2.57__py3-none-any.whl

bizydraft/static/js/hookLoadImage.js

@@ -92,7 +92,6 @@ app.registerExtension({
 
                     const callback = image_widget.callback
                     image_widget.callback = async function(e) {
-                        console.log(e)
                         if (typeof e == 'string') {
                             const item = e.includes('http') ?
                                 image_list.find(item => item.url === e) :
@@ -102,12 +101,7 @@ app.registerExtension({
 
                             image_name_widget.value = item ? item.name : e;
                             image_widget.value = image_url;
-                            console.log('=================image_url=================')
-                            console.log(image_url)
-                            console.log(item)
-                            console.log(await app.graphToPrompt())
                             callback([image_url])
-                            console.log(await app.graphToPrompt())
                         } else {
                             const item = e[0].split('/')
                             image_name_widget.options.values.pop()
@@ -125,9 +119,12 @@ app.registerExtension({
                 await getData()
 
 
-                function applyWorkflowImageSettings(workflowParams, image_list, image_widget, image_name_widget) {
+                function applyWorkflowImageSettings(workflowParams, image_list, image_widget, image_name_widget, currentNodeId) {
                     if (workflowParams && workflowParams.nodes) {
-                        const imageNode = workflowParams.nodes.find(item => computeIsLoadNode(item.type))
+                        // 根据当前节点ID查找对应的节点数据，而不是总是选择第一个
+                        const imageNode = workflowParams.nodes.find(item =>
+                            computeIsLoadNode(item.type) && item.id === currentNodeId
+                        )
                         if (imageNode && imageNode.widgets_values) {
                             const item = imageNode.widgets_values[0].split('/')
                             image_list.push({
@@ -145,12 +142,12 @@ app.registerExtension({
 
                 // 如果有存储的工作流数据，应用图像设置
                 if (window.currentWorkflowData) {
-                    applyWorkflowImageSettings(window.currentWorkflowData, image_list, image_widget, image_name_widget);
+                    applyWorkflowImageSettings(window.currentWorkflowData, image_list, image_widget, image_name_widget, this.id);
                     // 清除存储的数据，避免重复处理
                     delete window.currentWorkflowData;
                 } else {
                     // 原有的调用
-                    applyWorkflowImageSettings(workflowParams, image_list, image_widget, image_name_widget);
+                    applyWorkflowImageSettings(workflowParams, image_list, image_widget, image_name_widget, this.id);
                 }
                 //在这里发个postmessage
                 window.parent.postMessage({

bizydraft/static/js/hookLoadModel.js

@@ -34,7 +34,8 @@ const WHITELIST_NODES = {
     "SUPIR_model_loader_v2_clip": "CLIP",
     "LayerMask: LoadSAM2Model": "Detection",
     // "LayerMask: LoadCannyModel": "Detection",
-    "LayerMask: SegmentAnythingUltra V2": "Detection"
+    "LayerMask: SegmentAnythingUltra V2": "Detection",
+    "LoadFramePackModel": "UNet"
 };
 const HAZY_WHITELIST_NODES = {
 
@@ -88,7 +89,8 @@ const possibleWidgetNames=[
     "upscale_model_2",
     "upscale_model_3",
     "sam_model",
-    "sam2_model"
+    "sam2_model",
+    "grounding_dino_model"
 ]
 
 // 根据节点名称匹配模型类型
@@ -163,6 +165,23 @@ function createSetWidgetCallback(modelType, selectedBaseModels = []) {
 
                                         currentModel.value = version.id;
                                         currentNode.setDirtyCanvas(true);
+
+                                    // 删除节点上的感叹号徽章
+                                    if (currentNode && currentNode.badges && Array.isArray(currentNode.badges)) {
+                                        // 移除 text 为 '!' 的徽章
+                                        currentNode.badges = currentNode.badges.filter(badgeFn => {
+                                            try {
+                                                const badge = typeof badgeFn === 'function' ? badgeFn() : badgeFn;
+                                                return badge.text !== '!';
+                                            } catch (e) {
+                                                return true;
+                                            }
+                                        });
+                                        // 同时移除 hasTips 标记
+                                        if (currentNode.hasTips) {
+                                            delete currentNode.hasTips;
+                                        }
+                                    }
                                     }
                                 }
                             });
@@ -198,7 +217,22 @@ function setupNodeMouseBehavior(node, modelType) {
         };
     }
 }
-
+function addBadge(node) {
+    const customBadge = new LGraphBadge({
+        text: '!',
+        fgColor: 'white',
+        bgColor: '#FF6B6B',
+        fontSize: 12,
+        padding: 8,
+        height: 20,
+        cornerRadius: 10
+      })
+    if (node.hasTips) {
+        return
+    }
+    node.badges.push(() => customBadge);
+    node.hasTips = true;
+}
 app.registerExtension({
     name: "bizyair.hook.load.model",
     async beforeRegisterNodeDef(nodeType, nodeData, app) {
@@ -247,9 +281,26 @@ app.registerExtension({
 
                     const result = onNodeCreated?.apply(this, arguments);
                     let selectedBaseModels = [];
-                    // if (modelType === 'Checkpoint') {
-                    //     selectedBaseModels = ['SDXL', 'Pony', 'SD 3.5', 'Illustrious']
-                    // }
+                    targetWidget.forEach((widget, index) => {
+                        let model_version_id;
+                        if (index === 0) {
+                            model_version_id = this.widgets.find(w => w.name === "model_version_id");
+                            setTimeout(() => {
+                                if (widget.value != 'NONE' && model_version_id && !model_version_id.value) {
+                                    addBadge(this);
+                                }
+                            }, 200)
+                        } else {
+                            const fieldName = `model_version_id${index + 1}`;
+                            model_version_id = this.widgets.find(w => w.name === fieldName);
+                            setTimeout(() => {
+                                if (widget.value != 'NONE' && model_version_id && !model_version_id.value) {
+                                    addBadge(this);
+                                }
+                            }, 200)
+                        }
+                    });
+
                     createSetWidgetCallback(modelType, selectedBaseModels).call(this);
                     return result;
                 } catch (error) {

bizydraft/static/js/postEvent.js

@@ -250,6 +250,13 @@ app.registerExtension({
                         case 'promptQueued':
                         case 'logs':
                         case 'b_preview':
+                          if (msg.data.balance_not_enough) {
+                              window.parent.postMessage({
+                                  type: "functionResult",
+                                  method: "balanceNotEnough",
+                                  result: true,
+                              }, "*");
+                          }
                           dispatchCustomEvent(msg.type, msg.data)
                           break
                         case 'feature_flags':
@@ -1005,5 +1012,6 @@ app.registerExtension({
                 detail: e.detail
             }));
         })
+
     }
 });

{bizydraft-0.2.54.dist-info → bizydraft-0.2.57.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bizydraft
-Version: 0.2.54
+Version: 0.2.57
 Summary: bizydraft
 Requires-Dist: loguru
 Requires-Dist: aiohttp

{bizydraft-0.2.54.dist-info → bizydraft-0.2.57.dist-info}/RECORD

@@ -13,16 +13,16 @@ bizydraft/workflow_io.py,sha256=MYhJbpgkv8hrA5k_aolijOTrWpTtu62nzRznA4hv8JE,4298
 bizydraft/static/js/aiAppHandler.js,sha256=SBGms27-DYyc5uBluN9ScliW-zUAdw5cS6VQ4BV-_CQ,17816
 bizydraft/static/js/freezeModeHandler.js,sha256=SjpHD2nYymR-E13B0YcqkA6e4WycZOVI3c48Ts9qvWE,18027
 bizydraft/static/js/handleStyle.js,sha256=rUlgc-_vH76828Fv9zxTHq9UFj2AHANIUELArhl5lvU,2956
-bizydraft/static/js/hookLoadImage.js,sha256=DVUe-Ovc1G5DOOZ7bWjSiWJWEz7FBlmhEb_JKCtscu8,8320
-bizydraft/static/js/hookLoadModel.js,sha256=ovNvz4FRYNcOKJTueotugzIeGQDYAivYB5EQTUo8AO0,9914
+bizydraft/static/js/hookLoadImage.js,sha256=aFRWkgJW-Cp-YHjZh-3j-vsVcNaDZpBVoQqcFZ2Po0g,8186
+bizydraft/static/js/hookLoadModel.js,sha256=OHHTC1Hk8WXHlmDaNzMNGvT6BvQ6dGvxqALqboxUz84,12317
 bizydraft/static/js/main.js,sha256=oEsVEUZSo8ipx93oqs2WFQSRgp46XQ_D-Xao-wen2S8,121
 bizydraft/static/js/nodeFocusHandler.js,sha256=24xXbS4Q-GjJdRqf11i-1pBo8MkOJ24F7MHFV44EG6Q,4683
 bizydraft/static/js/nodeParamsFilter.js,sha256=H7lBB0G8HNqoGhOCH1hNXqPU-rPlrFyTxg_f_JgLEMk,4168
-bizydraft/static/js/postEvent.js,sha256=met19YcUlC8ttKQl9tHJ66NnfltBeOeq28TZBmIGhlM,39249
+bizydraft/static/js/postEvent.js,sha256=jzZcR238XyPdb5h6lICkarXCIf2gP8IV646dmWoMGpo,39604
 bizydraft/static/js/socket.js,sha256=VE3fTAgEfM0FZhL526Skt7OCRokOa3mzTCAjAomI_tE,2432
 bizydraft/static/js/tool.js,sha256=VupamUuh7tYiDnBTrL5Z_yLmhJinskhzRXwE3zfsKZM,2901
 bizydraft/static/js/uploadFile.js,sha256=WvglKzHMeOzDhOH3P-fLcPHxCLbKOJpo4DntoRxeJtI,4908
-bizydraft-0.2.54.dist-info/METADATA,sha256=9HHnNfSi145FUtWlkBxdSa4UcipPMuRe7aBCl6N2ENA,162
-bizydraft-0.2.54.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizydraft-0.2.54.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
-bizydraft-0.2.54.dist-info/RECORD,,
+bizydraft-0.2.57.dist-info/METADATA,sha256=rBDMa_w4XMqK44ISXZasXyq6gO03IrGDgdDioMbQw84,162
+bizydraft-0.2.57.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizydraft-0.2.57.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
+bizydraft-0.2.57.dist-info/RECORD,,