bizyengine 0.4.2__py3-none-any.whl

bizyengine/__init__.py

@@ -0,0 +1,35 @@
+__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
+
+from bizyengine.core import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+
+WEB_DIRECTORY = "./js"
+import bizyengine.bizyair_extras
+from bizyengine.misc import (
+    auth,
+    llm,
+    mzkolors,
+    nodes,
+    nodes_controlnet_aux,
+    nodes_controlnet_union_sdxl,
+    segment_anything,
+    supernode,
+)
+
+
+def update_mappings(module):
+    NODE_CLASS_MAPPINGS.update(**module.NODE_CLASS_MAPPINGS)
+    NODE_DISPLAY_NAME_MAPPINGS.update(**module.NODE_DISPLAY_NAME_MAPPINGS)
+
+
+update_mappings(auth)
+update_mappings(supernode)
+update_mappings(llm)
+update_mappings(nodes_controlnet_aux)
+update_mappings(nodes_controlnet_union_sdxl)
+update_mappings(mzkolors)
+update_mappings(segment_anything)
+
+try:
+    import bizyengine.bizy_server as bizy_server
+except Exception as e:
+    print("\n\n\033[91m[BizyAir]\033[0m Fail to import 'bizy_server':" f" {e}\n\n")

bizyengine/bizy_server/__init__.py

@@ -0,0 +1,7 @@
+from .execution import upload_worker
+from .server import BizyAirServer
+
+bizy_server = BizyAirServer()
+
+if bizy_server:
+    print("\n\n\033[92m[BizyAir]\033[0m Model hosting service initialized.\n\n")