bizydraft 0.2.49__py3-none-any.whl → 0.2.87__py3-none-any.whl

bizydraft/static/js/socket.js

@@ -1,109 +1,114 @@
 export class WebSocketClient {
   constructor(url, protocols) {
+    const host = "api.bizyair.cn";
 
-    const host = 'api.bizyair.cn';
-
-    if (url.startsWith('ws://') || url.startsWith('wss://')) {
+    if (url.startsWith("ws://") || url.startsWith("wss://")) {
       this.url = url;
     } else {
-      this.url = `${location.protocol == 'http:' ? 'wss' : 'wss'}://${host}${url}`
+      this.url = `${
+        location.protocol == "http:" ? "wss" : "wss"
+      }://${host}${url}`;
     }
-    this.protocols = protocols
-    this.reconnectDelay = 1000
-    this.maxReconnectDelay = 30000
-    this.keepAliveInterval = 10000
-    this.ws = null
-    this.keepAliveTimer = null
-    this.reconnectTimer = null
-
-    this.connect()
+    this.protocols = protocols;
+    this.reconnectDelay = 1000;
+    this.maxReconnectDelay = 30000;
+    this.keepAliveInterval = 10000;
+    this.ws = null;
+    this.keepAliveTimer = null;
+    this.reconnectTimer = null;
+
+    this.connect();
   }
 
   connect() {
-    this.ws = new WebSocket(this.url, this.protocols)
+    this.ws = new WebSocket(this.url, this.protocols);
     this.ws.onopen = () => {
-      this.onOpen()
-    }
+      this.onOpen();
+    };
 
     this.ws.onmessage = (message) => {
-      if (message.data !== 'pong') {
-        this.onMessage(message)
+      if (message.data !== "pong") {
+        this.onMessage(message);
       }
-    }
+    };
 
     this.ws.onerror = (error) => {
-      this.onError(error)
-    }
+      this.onError(error);
+    };
 
     this.ws.onclose = () => {
-      console.warn('The WebSocket connection has been closed and is ready to be reconnected')
-      this.onClose()
-      this.scheduleReconnect()
-    }
+      console.warn(
+        "The WebSocket connection has been closed and is ready to be reconnected"
+      );
+      this.onClose();
+      this.scheduleReconnect();
+    };
   }
 
   startKeepAlive() {
-    if (this.keepAliveTimer) return
+    if (this.keepAliveTimer) return;
 
     this.keepAliveTimer = setInterval(() => {
       if (this.ws && this.ws.readyState === WebSocket.OPEN) {
-        this.ws.send('ping')
+        this.ws.send("ping");
       }
-    }, this.keepAliveInterval)
+    }, this.keepAliveInterval);
   }
 
   stopKeepAlive() {
     if (this.keepAliveTimer) {
-      clearInterval(this.keepAliveTimer)
-      this.keepAliveTimer = null
+      clearInterval(this.keepAliveTimer);
+      this.keepAliveTimer = null;
     }
   }
 
   scheduleReconnect() {
-    if (this.reconnectTimer) return
+    if (this.reconnectTimer) return;
 
     this.reconnectTimer = setTimeout(() => {
-      console.log(`Attempt to reconnect...`)
-      this.connect()
-      this.reconnectTimer = null
-
-      this.reconnectDelay = Math.min(this.reconnectDelay * 2, this.maxReconnectDelay)
-    }, this.reconnectDelay)
+      console.log(`Attempt to reconnect...`);
+      this.connect();
+      this.reconnectTimer = null;
+
+      this.reconnectDelay = Math.min(
+        this.reconnectDelay * 2,
+        this.maxReconnectDelay
+      );
+    }, this.reconnectDelay);
   }
 
   onOpen() {
-    this.reconnectDelay = 2000
-    this.startKeepAlive()
+    this.reconnectDelay = 2000;
+    this.startKeepAlive();
   }
 
-
   onError(error) {
-    console.error('WebSocket Error: ', error)
+    console.error("WebSocket Error: ", error);
   }
 
   onClose() {
-    this.stopKeepAlive()
+    this.stopKeepAlive();
   }
 
   close() {
     if (this.ws) {
-      this.ws.close()
+      this.ws.close();
     }
-    this.stopKeepAlive()
+    this.stopKeepAlive();
     if (this.reconnectTimer) {
-      clearTimeout(this.reconnectTimer)
-      this.reconnectTimer = null
+      clearTimeout(this.reconnectTimer);
+      this.reconnectTimer = null;
     }
   }
 
   on(event, callback) {
-    if (event === 'message') {
+    if (event === "message") {
       this.onMessage = callback;
-    } else if (event === 'open') {
+    } else if (event === "open") {
       this.onOpen = callback;
-    } else if (event === 'error') {
+    } else if (event === "error") {
       this.onError = callback;
-    } else if (event === 'close') {
+    } else if (event === "close") {
       this.onClose = callback;
     }
     return this;

bizydraft/static/js/tool.js

@@ -1,60 +1,64 @@
 const loadNodeList = [
-    'LoadImage',
-    'LoadImageMask',
-    'LoadAudio',
-    'LoadVideo',
-    'Load3D',
-    'VHS_LoadVideo'
-]
+  "LoadImage",
+  "LoadImageMask",
+  "LoadAudio",
+  "LoadVideo",
+  "Load3D",
+  "VHS_LoadVideo",
+  "VHS_LoadAudioUpload",
+];
 const extMap = {
-    'LoadImage': '.png,.jpg,.jpeg,.webp,.gif,.svg,.ico,.bmp,.tiff,.tif,.heic,.heif',
-    'LoadImageMask': '.png,.jpg,.jpeg,.webp,.gif,.svg,.ico,.bmp,.tiff,.tif,.heic,.heif',
-    'LoadAudio': '.mp3,.wav,.ogg,.m4a,.aac,.flac,.wma,.m4r',
-    'LoadVideo': '.mp4,.mov,.avi,.mkv,.webm,.flv,.wmv,.m4v',
-    'Load3D': '.glb,.gltf,.fbx,.obj,.dae,.ply,.stl',
-    'VHS_LoadAudioUpload': '.mp3,.wav,.ogg,.m4a,.aac,.flac,.wma,.m4r',
-    "VHS_LoadVideo": '.mp4,.mov,.avi,.mkv,.webm,.flv,.wmv,.m4v',
-}
+  LoadImage: ".png,.jpg,.jpeg,.webp,.gif,.svg,.ico,.bmp,.tiff,.tif,.heic,.heif",
+  LoadImageMask:
+    ".png,.jpg,.jpeg,.webp,.gif,.svg,.ico,.bmp,.tiff,.tif,.heic,.heif",
+  LoadAudio: ".mp3,.wav,.ogg,.m4a,.aac,.flac,.wma,.m4r",
+  LoadVideo: ".mp4,.mov,.avi,.mkv,.webm,.flv,.wmv,.m4v",
+  Load3D: ".glb,.gltf,.fbx,.obj,.dae,.ply,.stl",
+  VHS_LoadAudioUpload: ".mp3,.wav,.ogg,.m4a,.aac,.flac,.wma,.m4r",
+  VHS_LoadVideo: ".mp4,.mov,.avi,.mkv,.webm,.flv,.wmv,.m4v",
+};
 export function getCookie(name) {
-    const value = `; ${document.cookie}`;
-    const parts = value.split(`; ${name}=`);
-    if (parts.length === 2) return parts.pop().split(';').shift();
-    return null;
+  const cookies = document.cookie.split(";");
+  for (let cookie of cookies) {
+    const [key, value] = cookie.trim().split("=");
+    if (key === name) {
+      return value;
+    }
+  }
+  return null;
 }
 export const hideWidget = (node, widget_name) => {
-    const widget = node.widgets.find(widget => widget.name === widget_name)
-    if (!widget) {
-        return
-    }
-    const originalComputeSize = widget.computeSize;
-    const originalType = widget.type;
+  const widget = node.widgets.find((widget) => widget.name === widget_name);
+  if (!widget) {
+    return;
+  }
+  const originalComputeSize = widget.computeSize;
+  const originalType = widget.type;
 
-    widget.computeSize = () => [0, -4];
-    widget.type = "hidden";
-    widget.hidden = true
-    widget.options = widget.options || {};
-    widget.show = () => {
-        widget.computeSize = originalComputeSize;
-        widget.type = originalType;
-        widget.height = undefined;
-    };
-}
+  widget.computeSize = () => [0, -4];
+  widget.type = "hidden";
+  widget.hidden = true;
+  widget.options = widget.options || {};
+  widget.show = () => {
+    widget.computeSize = originalComputeSize;
+    widget.type = originalType;
+    widget.height = undefined;
+  };
+};
 export const computeIsLoadNode = (nodeName) => {
-
-    return loadNodeList.includes(nodeName)
-}
+  return loadNodeList.includes(nodeName);
+};
 export const computeExt = (nodeName) => {
-
-    return extMap[nodeName] || ''
-}
+  return extMap[nodeName] || "";
+};
 /**
  * 判断节点名是否为模型加载类（不包含 bizyair）
  * @param {string} nodeName
  * @returns {boolean}
  */
 function isModelLoaderType(nodeName) {
-    const regex = /^(\w+).*Loader.*/i;
-    return regex.test(nodeName);
+  const regex = /^(\w+).*Loader.*/i;
+  return regex.test(nodeName);
 }
 
 /**
@@ -63,25 +67,29 @@ function isModelLoaderType(nodeName) {
  * @returns {Object} 处理后的新对象
  */
 export function processGraphOutput(output) {
-    const newOutput = JSON.parse(JSON.stringify(output));
-    for (const key in newOutput) {
-        const node = newOutput[key];
-        // 1. 如果 class_type 在 loadNodeList 里，删除 inputs.image_name
-        if (loadNodeList.includes(node.class_type) && node.inputs && node.inputs.image_name !== undefined) {
-            delete node.inputs.image_name;
-        }
-        if (isModelLoaderType(node.class_type)) {
-            delete newOutput[key];
-        }
-        // 2. 如果 class_type 满足 Loader 正则且不包含 bizyair，删除 inputs.model_version_id
-        if (isModelLoaderType(node.class_type) && node.inputs) {
-            if (node.inputs.model_version_id !== undefined) {
-                delete node.inputs.model_version_id;
-            }
-            if (node.inputs.ckpt_name !== undefined) {
-                delete node.inputs.ckpt_name;
-            }
-        }
+  const newOutput = JSON.parse(JSON.stringify(output));
+  for (const key in newOutput) {
+    const node = newOutput[key];
+    // 1. 如果 class_type 在 loadNodeList 里，删除 inputs.image_name
+    if (
+      loadNodeList.includes(node.class_type) &&
+      node.inputs &&
+      node.inputs.image_name !== undefined
+    ) {
+      delete node.inputs.image_name;
+    }
+    if (isModelLoaderType(node.class_type)) {
+      delete newOutput[key];
+    }
+    // 2. 如果 class_type 满足 Loader 正则且不包含 bizyair，删除 inputs.model_version_id
+    if (isModelLoaderType(node.class_type) && node.inputs) {
+      if (node.inputs.model_version_id !== undefined) {
+        delete node.inputs.model_version_id;
+      }
+      if (node.inputs.ckpt_name !== undefined) {
+        delete node.inputs.ckpt_name;
+      }
     }
-    return newOutput;
+  }
+  return newOutput;
 }

bizydraft/static/js/uploadFile.js

@@ -1,33 +1,39 @@
-import { getCookie } from './tool.js';
+import { getCookie } from "./tool.js";
 export async function fileToOss(file) {
   try {
-
     // 检查token是否存在
-    const apiHost = 'https://bizyair.cn/api'
+    const apiHost = "https://bizyair.cn/api";
     // const apiHost = 'http://localhost:3000/api'
-    const authToken = getCookie('bizy_token')
+    const authToken = getCookie("bizy_token");
     // console.log(authToken)
     if (!authToken) {
-      throw new Error('未找到认证Token，请先登录');
+      throw new Error("未找到认证Token，请先登录");
     }
 
     // 获取上传凭证
-    const uploadToken = await fetch(`${apiHost}/special/community/upload_token?file_name=${encodeURIComponent(file.name)}&file_type=inputs`, {
-      method: 'GET',
-      headers: {
-        'Content-Type': 'application/json',
-        'Authorization': `Bearer ${authToken}`
+    const uploadToken = await fetch(
+      `${apiHost}/special/community/upload_token?file_name=${encodeURIComponent(
+        file.name
+      )}&file_type=inputs`,
+      {
+        method: "GET",
+        headers: {
+          "Content-Type": "application/json",
+          Authorization: `Bearer ${authToken}`,
+        },
       }
-    });
+    );
 
     // 检查响应状态
     if (!uploadToken.ok) {
       const errorText = await uploadToken.text();
-      console.error('获取上传凭证失败:', uploadToken.status, errorText);
-      throw new Error(`获取上传凭证失败: ${uploadToken.status} ${uploadToken.statusText}`);
+      console.error("获取上传凭证失败:", uploadToken.status, errorText);
+      throw new Error(
+        `获取上传凭证失败: ${uploadToken.status} ${uploadToken.statusText}`
+      );
     }
 
-    const {data} = await uploadToken.json();
+    const { data } = await uploadToken.json();
     // console.log('上传凭证响应:', data);
 
     // 使用STS凭证上传
@@ -37,7 +43,7 @@ export async function fileToOss(file) {
       securityToken: data.data.file.security_token,
       bucket: data.data.storage.bucket,
       region: data.data.storage.region,
-      objectKey: data.data.file.object_key
+      objectKey: data.data.file.object_key,
     };
 
     // console.log('OSS配置:', ossConfig);
@@ -53,10 +59,10 @@ export async function fileToOss(file) {
       expiration: expiration.toISOString(),
       conditions: [
         // 文件大小限制
-        ['content-length-range', 0, 1048576000], // 最大1000MB
+        ["content-length-range", 0, 1048576000], // 最大1000MB
         // 指定允许的文件名前缀
-        ['starts-with', '$key', ossConfig.objectKey.split('/')[0]]
-      ]
+        ["starts-with", "$key", ossConfig.objectKey.split("/")[0]],
+      ],
     };
 
     // Policy Base64编码
@@ -64,23 +70,23 @@ export async function fileToOss(file) {
     // console.log('Policy:', policy);
 
     // 构建表单字段
-    formData.append('key', ossConfig.objectKey);
-    formData.append('OSSAccessKeyId', ossConfig.accessKeyId);
-    formData.append('policy', policy);
-    formData.append('success_action_status', '200');
+    formData.append("key", ossConfig.objectKey);
+    formData.append("OSSAccessKeyId", ossConfig.accessKeyId);
+    formData.append("policy", policy);
+    formData.append("success_action_status", "200");
 
     // 如果有临时token，需要添加
     if (ossConfig.securityToken) {
-      formData.append('x-oss-security-token', ossConfig.securityToken);
+      formData.append("x-oss-security-token", ossConfig.securityToken);
     }
 
     // 计算签名 - 阿里云官方要求使用HMAC-SHA1
     const signature = await hmacSha1(policy, ossConfig.accessKeySecret);
     // console.log('计算的签名:', signature);
-    formData.append('signature', signature);
+    formData.append("signature", signature);
 
     // 最后添加文件内容
-    formData.append('file', file);
+    formData.append("file", file);
 
     // OSS服务端点
     const host = `https://${ossConfig.bucket}.${ossConfig.region}.aliyuncs.com`;
@@ -88,15 +94,17 @@ export async function fileToOss(file) {
 
     // 开始上传
     const uploadResponse = await fetch(host, {
-      method: 'POST',
-      body: formData
+      method: "POST",
+      body: formData,
     });
 
     // 检查响应
     if (!uploadResponse.ok) {
       const errorText = await uploadResponse.text();
-      console.error('上传失败:', uploadResponse.status, errorText);
-      throw new Error(`上传失败: ${uploadResponse.status} ${uploadResponse.statusText}`);
+      console.error("上传失败:", uploadResponse.status, errorText);
+      throw new Error(
+        `上传失败: ${uploadResponse.status} ${uploadResponse.statusText}`
+      );
     }
 
     // 构建公开访问URL
@@ -104,23 +112,23 @@ export async function fileToOss(file) {
 
     // 提交资源
     await fetch(`${apiHost}/special/community/commit_input_resource`, {
-      method: 'POST',
+      method: "POST",
       headers: {
-        'Content-Type': 'application/json',
-        'Authorization': `Bearer ${authToken}`
+        "Content-Type": "application/json",
+        Authorization: `Bearer ${authToken}`,
       },
       body: JSON.stringify({
         object_key: data.data.file.object_key,
         name: file.name,
-      })
-    })
+      }),
+    });
     return {
       url: fileUrl,
       ossTokenFile: data.data.file,
-      ossTokenStorage: data.data.storage
+      ossTokenStorage: data.data.storage,
     };
   } catch (error) {
-    console.error('文件上传到OSS失败:', error);
+    console.error("文件上传到OSS失败:", error);
 
     throw error;
   }
@@ -135,16 +143,16 @@ async function hmacSha1(message, key) {
 
   // 导入密钥
   const cryptoKey = await window.crypto.subtle.importKey(
-    'raw',
+    "raw",
     keyData,
-    { name: 'HMAC', hash: 'SHA-1' },
+    { name: "HMAC", hash: "SHA-1" },
     false,
-    ['sign']
+    ["sign"]
   );
 
   // 计算签名
   const signature = await window.crypto.subtle.sign(
-    'HMAC',
+    "HMAC",
     cryptoKey,
     messageData
   );
@@ -157,7 +165,7 @@ async function hmacSha1(message, key) {
 // 将ArrayBuffer转换为Base64字符串
 function arrayBufferToBase64(buffer) {
   const bytes = new Uint8Array(buffer);
-  let binary = '';
+  let binary = "";
   for (let i = 0; i < bytes.byteLength; i++) {
     binary += String.fromCharCode(bytes[i]);
   }