bizydraft 0.2.49__py3-none-any.whl → 0.2.87__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.49.dist-info → bizydraft-0.2.87.dist-info}/METADATA

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

bizydraft-0.2.87.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=j7i9xAfaWvcu-mSoDISQyEwdERd0y-GFhvY2To7MRTU,5015
+bizydraft/oss_utils.py,sha256=_UzbS4e_MW5UynkQd2JI6edfdWxStm-Kbt6k7CvJJGA,16663
+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=aiNOJeoc-jIV9RSuIERS86nEGyj9RbFDVD1eD0aPIH8,15710
+bizydraft/static/js/clipspaceToOss.js,sha256=iiHjYoiQ2GAgj-d3VQhvf5DX7uJMk60dJ_VnmOsb_Mk,14728
+bizydraft/static/js/disableComfyWebSocket.js,sha256=Kx4d7azNg0qGOZxUkG1mb4dFmEjxZ30yfhtKBZ2SEE8,1978
+bizydraft/static/js/freezeModeHandler.js,sha256=P6sBWgaMsLMXz0ADuuGIKPyPbcB69iHJufY93peTCWk,16490
+bizydraft/static/js/handleStyle.js,sha256=sYcyDoSGlLhb6fCCp6RTWZ2gq79TLc_Am1EP0NO26uk,6336
+bizydraft/static/js/hookLoadMedia.js,sha256=dlXlVYIXgEkoWk-VEx22VeewUOw86gJHiI9Vu9SFVOA,6005
+bizydraft/static/js/hookLoadModel.js,sha256=ho9ZTDtKtPQhigSwKjuKClC2X8jfU41-SFbHEonoSt0,7233
+bizydraft/static/js/main.js,sha256=PRe4LdsquEQWrZDnVd4ubpVQuD1eDIedAXFFazKdoKQ,188
+bizydraft/static/js/nodeFocusHandler.js,sha256=wJYfjbpkY4uTLfKLLLOGkIjmB4yU15DI3He4Sn28_c4,4414
+bizydraft/static/js/nodeParamsFilter.js,sha256=muhsNLvXvChivtQfCujgr9r0RxP4kq3FgiQaMP_jfx8,3732
+bizydraft/static/js/postEvent.js,sha256=2YPOcAkwnn3acOcvYGDkxto53x_ihr4WhVdEBnGILwg,37317
+bizydraft/static/js/socket.js,sha256=S21_74fLF6syMcWONeRpQVXIOkniQRP5Evv-t508kVM,2525
+bizydraft/static/js/tool.js,sha256=1Go2IXDTN_r7hDT2ZvOnpRpj1AkU88pBfQItva_IivE,2810
+bizydraft/static/js/uploadFile.js,sha256=maDTrauNj24iOvUGwZxlc32fOKKF1_iifUVquGtKbGM,4995
+bizydraft/static/js/workflow_io.js,sha256=m-WbjkJSYbTC5PvQ1jV9glVDa8pgkyTTQS2HnsLfNec,4771
+bizydraft/static/js/hookLoad/configLoader.js,sha256=vJbraYque8QK8faNM7O8F0ONygGb9abKlm2P-juWrps,2331
+bizydraft/static/js/hookLoad/media.js,sha256=yAag-VmrhEX0k-TRIX6CkOQC31Mk-5ify2_59Gm3GxA,17470
+bizydraft/static/js/hookLoad/model.js,sha256=MF860un_kuODxbEFs8v9DueVWfvddvrsRpyLloVrFgo,9031
+bizydraft-0.2.87.dist-info/METADATA,sha256=Pmt1qtVYBtooCH4CJIYqH_zMCDXjkH2J0yBJMV7C8gk,162
+bizydraft-0.2.87.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizydraft-0.2.87.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
+bizydraft-0.2.87.dist-info/RECORD,,

bizydraft/static/js/hookLoadImage.js

@@ -1,177 +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);
-                }
-                //在这里发个postmessage
-                window.parent.postMessage({
-                    type: 'functionResult',
-                    method: 'hookLoadImageCompleted',
-                    params: {}
-                }, '*');
-            }
-        }
-    }
-})
-
-// 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.49.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=ktonkuI3NEctELUL9C0Hmv2-DaDr1MClGZWeHrP2VCA,3485
-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=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=1a4lDiE0l_nl8ajaEz8I9CFPdW3b7vjhwpyfr0yMUak,9641
-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=AEeph8HpmIn5qeyebkh5FzcKW6GRh7PGDHci66yKaHA,40200
-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.49.dist-info/METADATA,sha256=z1Aq6g8siuV0MlRH4rHNYJ_cHudcPQ6yHvJKIYbY1I0,162
-bizydraft-0.2.49.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizydraft-0.2.49.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
-bizydraft-0.2.49.dist-info/RECORD,,