bizydraft 0.2.31__py3-none-any.whl → 0.2.78.dev20251117024007__py3-none-any.whl

bizydraft/static/js/workflow_io.js

@@ -0,0 +1,193 @@
+// 基础类型检查
+const BASIC_TYPES = ['number', 'string', 'boolean'];
+
+/**
+ * 检查值是否为基础类型
+ * @param {any} val - 要检查的值
+ * @returns {boolean} 是否为基础类型
+ */
+function isBasicType(val) {
+    const type = typeof val;
+    return BASIC_TYPES.includes(type);
+}
+
+/**
+ * 遍历所有节点inputs，保留基础类型参数
+ * @param {Object} prompt - prompt对象
+ * @returns {Array} 输入参数数组
+ */
+function getInputParams(prompt) {
+    const inputParams = [];
+
+    for (const [nodeId, node] of Object.entries(prompt)) {
+        const classType = node.class_type;
+        const inputs = node.inputs || {};
+
+        for (const [k, v] of Object.entries(inputs)) {
+            if (isBasicType(v)) {
+                inputParams.push({
+                    node_id: nodeId,
+                    class_type: classType,
+                    param_name: k,
+                    param_value: v
+                });
+            }
+        }
+    }
+
+    return inputParams;
+}
+
+/**
+ * 通过prompt参数推断叶子节点（即没有被其他节点inputs引用的节点）
+ * @param {Object} prompt - prompt对象
+ * @returns {Set} 叶子节点id集合
+ */
+function getLeafNodesFromPrompt(prompt) {
+    const referenced = new Set();
+
+    for (const [nodeId, node] of Object.entries(prompt)) {
+        const inputs = node.inputs || {};
+
+        for (const v of Object.values(inputs)) {
+            // 如果是引用（如["4", 1]），则v为array且第一个元素为节点id
+            if (Array.isArray(v) && v.length > 0 && typeof v[0] === 'string') {
+                referenced.add(v[0]);
+            }
+        }
+    }
+
+    const allNodes = new Set(Object.keys(prompt));
+    const leafNodes = new Set([...allNodes].filter(x => !referenced.has(x)));
+
+    return leafNodes;
+}
+
+/**
+ * 从工作流节点中查找指定节点id的节点
+ * @param {Array} wfNodes - 工作流节点数组
+ * @param {string} nodeId - 节点id
+ * @returns {Object|null} 找到的节点或null
+ */
+function getNodeIdFromWfNodes(wfNodes, nodeId) {
+    for (const wfNode of wfNodes) {
+        if (String(wfNode.id || 0) === nodeId) {
+            return wfNode;
+        }
+    }
+    return null;
+}
+
+/**
+ * 按节点类型和参数名整理参数信息
+ * @param {Array} inputParams - 输入参数数组
+ * @param {Set} excludeNodeIds - 需要排除的节点id集合
+ * @param {Array} wfNodes - 工作流节点数组
+ * @returns {Array} 整理后的参数信息
+ */
+function summarizeParams(inputParams, excludeNodeIds, wfNodes) {
+    const summary = {};
+
+    for (const param of inputParams) {
+        const nodeId = param.node_id;
+
+        if (excludeNodeIds.has(nodeId)) {
+            continue;
+        }
+
+        const wfNode = getNodeIdFromWfNodes(wfNodes, nodeId);
+        let title = param.class_type;
+
+        if (wfNode && wfNode.title) {
+            title = wfNode.title;
+        }
+
+        const key = `${param.class_type}_${nodeId}`;
+
+        if (!summary[key]) {
+            summary[key] = {
+                name: param.class_type,
+                displayName: param.class_type,
+                nodeId: nodeId,
+                params: [],
+                title: title
+            };
+        }
+
+        // 获取参数值的类型
+        let paramType = typeof param.param_value;
+        if (paramType === 'number') {
+            paramType = Number.isInteger(param.param_value) ? 'INT' : 'FLOAT';
+        } else {
+            paramType = paramType.toUpperCase();
+        }
+
+        summary[key].params.push({
+            name: param.param_name,
+            displayName: param.param_name,
+            type: paramType,
+            defaultValue: param.param_value
+        });
+    }
+
+    return Object.values(summary);
+}
+
+/**
+ * 解析工作流IO数据
+ * @param {Object} requestData - 请求数据
+ * @returns {Object} 解析结果
+ */
+function parseWorkflowIO(requestData) {
+    const prompt = requestData.prompt || {};
+    const extraData = requestData.extra_data || {};
+
+    const extraPnginfo = extraData.extra_pnginfo || {};
+    const workflow = extraPnginfo.workflow || {};
+    const wfNodes = workflow.nodes || [];
+
+    // 1. 输出参数（推断叶子节点）
+    const leafNodes = getLeafNodesFromPrompt(prompt);
+    const outputNodes = [];
+
+    for (const nodeId of leafNodes) {
+        const node = prompt[nodeId];
+        const classType = node.class_type;
+        const wfNode = getNodeIdFromWfNodes(wfNodes, nodeId);
+
+        let title = classType;
+        if (wfNode && wfNode.title) {
+            title = wfNode.title;
+        }
+
+        outputNodes.push({
+            nodeId: nodeId,
+            name: classType,
+            displayName: classType,
+            title: title,
+            params: []
+        });
+    }
+
+    // 2. 输入参数（排除已作为输出的节点）
+    const inputParams = getInputParams(prompt);
+    const inputSummary = summarizeParams(inputParams, leafNodes, wfNodes);
+
+    // 3. 构造response格式
+    return {
+        data: {
+            inputs: inputSummary,
+            outputs: outputNodes
+        }
+    };
+}
+
+// 导出函数
+export {
+    parseWorkflowIO,
+    isBasicType,
+    getInputParams,
+    getLeafNodesFromPrompt,
+    summarizeParams,
+    getNodeIdFromWfNodes
+};

{bizydraft-0.2.31.dist-info → bizydraft-0.2.78.dev20251117024007.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bizydraft
-Version: 0.2.31
+Version: 0.2.78.dev20251117024007
 Summary: bizydraft
 Requires-Dist: loguru
 Requires-Dist: aiohttp

bizydraft-0.2.78.dev20251117024007.dist-info/RECORD

@@ -0,0 +1,34 @@
+bizydraft/__init__.py,sha256=OM-sKCQrPh25nHVJIX-DgF1raMYyoWLSuyduIAHt0Gs,78
+bizydraft/block_nodes.py,sha256=Lqn3oSCaGDHR2OICc8a2iRoRCVVK9v1-9MM3r-qIZgA,1092
+bizydraft/env.py,sha256=VFmGopVL2TGWA6hwxyFhIglCEcQxy6iVvL_raMNd6u4,407
+bizydraft/hijack_nodes.py,sha256=riHsp4DhU60E60bfXUl8kVqjs1QvQWmx_FsgQAdWTa4,4197
+bizydraft/hijack_routes.py,sha256=TVK7zdSIe4wN5dusjnAr1we43K6B_Zup5-dWcBj5rxo,3545
+bizydraft/oss_utils.py,sha256=IjWKFZ98KNvLw7bmFuqux5mEG36B8c9eQbQmL_gZo0M,16254
+bizydraft/patch_handlers.py,sha256=cP6gIYBYghWXI72hBbz3BbQHP1O5IvslS8iNK29MEbI,14159
+bizydraft/postload.py,sha256=XFElKcmCajT_oO7SVJJBaN04XcWro54N5HB5cSCxfvI,1308
+bizydraft/prestartup_patch.py,sha256=4FGjmRcDHELjtlQOrfTfk2Un5OS89QIqfq-gEcB9WDs,998
+bizydraft/resp.py,sha256=8INvKOe5Dgai3peKfqKjrhUoYeuXWXn358w30-_cY-A,369
+bizydraft/server.py,sha256=L2zoJgOisr65IRphOyko74AdsLel59gh55peyMaUrO8,2102
+bizydraft/workflow_io.py,sha256=MYhJbpgkv8hrA5k_aolijOTrWpTtu62nzRznA4hv8JE,4298
+bizydraft/static/js/aiAppHandler.js,sha256=OQRhhoqvc8iZeCvHTtdaD2VTYBGzkeAGdCk1UMO2RZs,17525
+bizydraft/static/js/clipspaceToOss.js,sha256=brfEPs71Tky5Dnc47UXNEFeFlESDE3kQvUH8ducpIew,14265
+bizydraft/static/js/disableComfyWebSocket.js,sha256=ZDOVlU3v3EvWpMRkD8s_AnO43vuWojVLLjkF2pNEVrA,1957
+bizydraft/static/js/freezeModeHandler.js,sha256=SjpHD2nYymR-E13B0YcqkA6e4WycZOVI3c48Ts9qvWE,18027
+bizydraft/static/js/handleStyle.js,sha256=2zAD9zGktGwdi0VLw4cDvqkPGTh0yYBAniytIbfI9kU,3867
+bizydraft/static/js/hookLoadMedia.js,sha256=EYDuitzOFvlNzM_e4IdE0-wXHQfnyCx7KrwzoxVTdr0,6363
+bizydraft/static/js/hookLoadModel.js,sha256=r8ym8ih_w39bo_B0dZwnq-b00XOL-7dhM77R7_5J_Rc,8120
+bizydraft/static/js/main.js,sha256=-pMeAG8vkKO6yq9zXCeHI78KezGn0r9AhYm65DuZrVI,188
+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=_EyA71m5DwGbhHsuf1amKhpLUBxfBWZaMMxViNhuYzU,43481
+bizydraft/static/js/socket.js,sha256=VE3fTAgEfM0FZhL526Skt7OCRokOa3mzTCAjAomI_tE,2432
+bizydraft/static/js/tool.js,sha256=2Hhv2J18OaFZRWmHIlseahjd5Ot7ZVxUPb5z46YeXIo,2928
+bizydraft/static/js/uploadFile.js,sha256=WvglKzHMeOzDhOH3P-fLcPHxCLbKOJpo4DntoRxeJtI,4908
+bizydraft/static/js/workflow_io.js,sha256=FWAjncvWhvy-3nN_legD2fpRwgnIncpRLHU5X016a-U,5236
+bizydraft/static/js/hookLoad/configLoader.js,sha256=R1k0GKdEmv4IIxdT2F-oOI9X9I19ECe77P_1tO3XxgY,2525
+bizydraft/static/js/hookLoad/media.js,sha256=iX_zoZ0OzLAdYZ3bn5GpzpnwU-YjcBuEozQt-MwNuwk,17936
+bizydraft/static/js/hookLoad/model.js,sha256=aHvEPt9k3CcrawHSYnQYcvbtTRwIztk-XDRA3lvgXvA,9890
+bizydraft-0.2.78.dev20251117024007.dist-info/METADATA,sha256=UHr86vHNHIdSNQYagWYilP-23NkFtCZtyZxgGyHIRhQ,180
+bizydraft-0.2.78.dev20251117024007.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizydraft-0.2.78.dev20251117024007.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
+bizydraft-0.2.78.dev20251117024007.dist-info/RECORD,,

bizydraft/static/js/hookLoadImage.js

@@ -1,173 +0,0 @@
-import { app } from "../../scripts/app.js";
-import { getCookie, computeIsLoadNode, computeExt, hideWidget } from './tool.js';
-
-
-app.registerExtension({
-    name: "bizyair.image.to.oss",
-    async beforeRegisterNodeDef(nodeType, nodeData) {
-        let workflowParams = null
-        document.addEventListener('workflowLoaded', (event) => {
-            workflowParams = event.detail;
-        })
-        document.addEventListener('drop', (e) => {
-            e.preventDefault();
-            const files = e.dataTransfer.files;
-
-            Array.from(files).forEach((file) => {
-                if (file.type === 'application/json' || file.name.endsWith('.json')) {
-                    const reader = new FileReader();
-                    reader.onload = function(event) {
-                        try {
-                            const jsonContent = JSON.parse(event.target.result);
-                            if (jsonContent && jsonContent.nodes) {
-                                window.currentWorkflowData = jsonContent;
-                            }
-                        } catch (error) {
-                            console.error('解析JSON文件失败:', error);
-                        }
-                    };
-                    reader.readAsText(file);
-                }
-            });
-        })
-        if (computeIsLoadNode(nodeData.name)) {
-            nodeType.prototype.onNodeCreated = async function() {
-                const apiHost = 'https://bizyair.cn/api'
-                const image_widget = this.widgets.find(w => {
-                    return w.name === 'image'
-                            || w.name === 'file'
-                            || w.name === 'audio'
-                            || w.name === 'model_file'
-                });
-                let image_name_widget = this.widgets.find(w => w.name === 'image_name');
-                let image_list = []
-                const getData = async () => {
-                    const res = await fetch(`${apiHost}/special/community/commit_input_resource?${
-                        new URLSearchParams({
-                            ext: computeExt(nodeData.name),
-                            current: 1,
-                            page_size: 100
-
-                        }).toString()
-                    }`, {
-                        method: 'GET',
-                        headers: {
-                            'Content-Type': 'application/json',
-                            'Authorization': `Bearer ${getCookie('bizy_token')}`
-                        }
-                    })
-                    const {data} = await res.json()
-                    const list = (data && data.data && data.data.data && data.data.data.list) || []
-                    image_list = list.filter(item => item.name).map(item => {
-                        return {
-                            url: item.url,
-                            id: item.id,
-                            name: item.name
-                        }
-                    })
-                    if (!image_name_widget) {
-                        image_name_widget = this.addWidget("combo", "image_name", "", function(e){
-                            const item = image_list.find(item => item.name === e)
-                            const image_url = decodeURIComponent(item.url);
-                            image_widget.value = image_url;
-                            if (image_widget.callback) {
-                                image_widget.callback(e);
-                            }
-                        }, {
-                            serialize: true,
-                            values: image_list.map(item => item.name)
-                        });
-                    }
-                    const val = image_list.find(item => item.url === image_widget.value)?.name || image_widget.value
-                    image_name_widget.label = image_widget.label
-                    image_name_widget.value = val
-
-                    const currentIndex = this.widgets.indexOf(image_name_widget);
-                    if (currentIndex > 1) {
-                        this.widgets.splice(currentIndex, 1);
-                        this.widgets.splice(1, 0, image_name_widget);
-                    }
-                    hideWidget(this, image_widget.name)
-                    image_widget.options.values = image_list.map(item => item.name);
-
-                    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) :
-                                image_list.find(item => item.name === e)
-
-                            const image_url = item ? decodeURIComponent(item.url) : e;
-
-                            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()
-                            image_name_widget.options.values.push(item[item.length - 1])
-                            image_name_widget.value = item[item.length - 1]
-                            image_list.push({
-                                name: item[item.length - 1],
-                                url: e[0]
-                            })
-                            callback(e)
-                        }
-                    }
-                    return true
-                }
-                await getData()
-
-
-                function applyWorkflowImageSettings(workflowParams, image_list, image_widget, image_name_widget) {
-                    if (workflowParams && workflowParams.nodes) {
-                        const imageNode = workflowParams.nodes.find(item => computeIsLoadNode(item.type))
-                        if (imageNode && imageNode.widgets_values) {
-                            const item = imageNode.widgets_values[0].split('/')
-                            image_list.push({
-                                name: item[item.length - 1],
-                                url: imageNode.widgets_values[0]
-                            })
-                            image_widget.value = imageNode.widgets_values[0]
-
-                            image_widget.options.values = image_list.map(item => item.url)
-                            image_name_widget.options.values = image_list.map(item => item.name)
-                            image_widget.callback(imageNode.widgets_values[0])
-                        }
-                    }
-                }
-
-                // 如果有存储的工作流数据，应用图像设置
-                if (window.currentWorkflowData) {
-                    applyWorkflowImageSettings(window.currentWorkflowData, image_list, image_widget, image_name_widget);
-                    // 清除存储的数据，避免重复处理
-                    delete window.currentWorkflowData;
-                } else {
-                    // 原有的调用
-                    applyWorkflowImageSettings(workflowParams, image_list, image_widget, image_name_widget);
-                }
-
-
-            }
-        }
-    }
-})
-
-// app.api.addEventListener('graphChanged', (e) => {
-//     console.log('Graph 发生变化，当前 workflow JSON:', e.detail)
-//     window.parent.postMessage({
-//         type: 'functionResult',
-//         method: 'workflowChanged',
-//         result: e.detail
-//     }, '*');
-
-//     document.dispatchEvent(new CustomEvent('workflowLoaded', {
-//         detail: e.detail
-//     }));
-// })

bizydraft-0.2.31.dist-info/RECORD

@@ -1,28 +0,0 @@
-bizydraft/__init__.py,sha256=OM-sKCQrPh25nHVJIX-DgF1raMYyoWLSuyduIAHt0Gs,78
-bizydraft/block_nodes.py,sha256=Lqn3oSCaGDHR2OICc8a2iRoRCVVK9v1-9MM3r-qIZgA,1092
-bizydraft/env.py,sha256=nVh4xBKXWnRUPbHL3PqZr6VFN8txKsrsmshu8PFGaqo,299
-bizydraft/hijack_nodes.py,sha256=6k2MvC0lhzg32zr14twuUMnAeYxPvUYWL2ID6Tv8cw8,2904
-bizydraft/hijack_routes.py,sha256=lf6x3xDzbo9yQIRwfG_1oxcUNfrX_1ogbiff3WOV9gM,3268
-bizydraft/oss_utils.py,sha256=W3-Td6K4vrB83oMowN82c3nmQ3IrEoR1UX6S8AvO6ig,7896
-bizydraft/patch_handlers.py,sha256=WPf4xeV5sBpoJ2MkHHhx0wem7B0KQNuhONElN7c9v3g,5989
-bizydraft/postload.py,sha256=XFElKcmCajT_oO7SVJJBaN04XcWro54N5HB5cSCxfvI,1308
-bizydraft/prestartup_patch.py,sha256=4FGjmRcDHELjtlQOrfTfk2Un5OS89QIqfq-gEcB9WDs,998
-bizydraft/resp.py,sha256=8INvKOe5Dgai3peKfqKjrhUoYeuXWXn358w30-_cY-A,369
-bizydraft/server.py,sha256=L2zoJgOisr65IRphOyko74AdsLel59gh55peyMaUrO8,2102
-bizydraft/workflow_io.py,sha256=MYhJbpgkv8hrA5k_aolijOTrWpTtu62nzRznA4hv8JE,4298
-bizydraft/static/js/aiAppHandler.js,sha256=t3tefGg9d_2-VeNxqrwMMdX6JYE5uroyXZq0hXNWSzw,17572
-bizydraft/static/js/freezeModeHandler.js,sha256=l1-JK8hyFbJzQ6ylq6occUa_kwszPq9rDbKut48bn6s,17247
-bizydraft/static/js/handleStyle.js,sha256=o9JpjOqs_lLIr-jpqpGNjUN74XQPT8pD3fF5bjsHLAU,1466
-bizydraft/static/js/hookLoadImage.js,sha256=jQGckFr7bTLRiwcFsVmSy1mj_o7TKj1uup0QQ00qwbk,8079
-bizydraft/static/js/hookLoadModel.js,sha256=j_U-z275xxeu3XqgSVlrP8jKURWni5yUaG3pcJQ3ZGI,8582
-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=5rafG9GjaesxmFD_8eSJQYHeSNWfOG3iWXLUPuaRWxg,38296
-bizydraft/static/js/socket.js,sha256=VE3fTAgEfM0FZhL526Skt7OCRokOa3mzTCAjAomI_tE,2432
-bizydraft/static/js/tool.js,sha256=dwVFLrT0pEsdOBkh2RRUDvyVo8GzOIVYg7vU5RiM0m0,2765
-bizydraft/static/js/uploadFile.js,sha256=WvglKzHMeOzDhOH3P-fLcPHxCLbKOJpo4DntoRxeJtI,4908
-bizydraft-0.2.31.dist-info/METADATA,sha256=8Hfs60uIUxadzzAuwZnIHzCwMtWqjV33z40yDVTpCa8,162
-bizydraft-0.2.31.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizydraft-0.2.31.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
-bizydraft-0.2.31.dist-info/RECORD,,