first commit

__init__.py

@@ -0,0 +1,127 @@
+import os
+import shutil
+import folder_paths
+root_dir = os.getcwd()
+now_dir = os.path.dirname(os.path.abspath(__file__))
+
+shutil.copy(os.path.join(now_dir,"index.html"),os.path.join(root_dir,"web", "index.html"))
+shutil.copytree(os.path.join(now_dir,"live2djs"),os.path.join(root_dir,"web", "lib","live2djs"),dirs_exist_ok=True)
+
+input_path = folder_paths.get_input_directory()
+vtuber_list = os.listdir(os.path.join(now_dir,"web","vtuber"))
+class Live2DViewer:
+    """
+    A example node
+
+    Class methods
+    -------------
+    INPUT_TYPES (dict): 
+        Tell the main program input parameters of nodes.
+    IS_CHANGED:
+        optional method to control when the node is re executed.
+
+    Attributes
+    ----------
+    RETURN_TYPES (`tuple`): 
+        The type of each element in the output tulple.
+    RETURN_NAMES (`tuple`):
+        Optional: The name of each output in the output tulple.
+    FUNCTION (`str`):
+        The name of the entry-point method. For example, if `FUNCTION = "execute"` then it will run Example().execute()
+    OUTPUT_NODE ([`bool`]):
+        If this node is an output node that outputs a result/image from the graph. The SaveImage node is an example.
+        The backend iterates on these output nodes and tries to execute all their parents if their parent graph is properly connected.
+        Assumed to be False if not present.
+    CATEGORY (`str`):
+        The category the node should appear in the UI.
+    execute(s) -> tuple || None:
+        The entry point method. The name of this method must be the same as the value of property `FUNCTION`.
+        For example, if `FUNCTION = "execute"` then this method's name must be `execute`, if `FUNCTION = "foo"` then it must be `foo`.
+    """
+    def __init__(self):
+        pass
+    
+    @classmethod
+    def INPUT_TYPES(s):
+        """
+            Return a dictionary which contains config for all input fields.
+            Some types (string): "MODEL", "VAE", "CLIP", "CONDITIONING", "LATENT", "IMAGE", "INT", "STRING", "FLOAT".
+            Input types "INT", "STRING" or "FLOAT" are special values for fields on the node.
+            The type can be a list for selection.
+
+            Returns: `dict`:
+                - Key input_fields_group (`string`): Can be either required, hidden or optional. A node class must have property `required`
+                - Value input_fields (`dict`): Contains input fields config:
+                    * Key field_name (`string`): Name of a entry-point method's argument
+                    * Value field_config (`tuple`):
+                        + First value is a string indicate the type of field or a list for selection.
+                        + Secound value is a config for type "INT", "STRING" or "FLOAT".
+        """
+        return {
+            "required": {
+                "audio": ("AUDIO",),
+                "live2d_model_name": (vtuber_list,{
+                    "default":"kei_vowels_pro"
+                })
+            },
+        }
+
+    RETURN_TYPES = ()
+    #RETURN_NAMES = ("image_output_name",)
+
+    FUNCTION = "show"
+
+    OUTPUT_NODE = True
+
+    CATEGORY = "AIFSH_Live2D"
+
+    def show(self,audio,live2d_model_name):
+        audio_name = os.path.basename(audio)
+        audio_type = os.path.basename(os.path.dirname(audio))
+        return {"ui": {"vtuber":[audio_name,audio_type,live2d_model_name]}}
+
+    """
+        The node will always be re executed if any of the inputs change but
+        this method can be used to force the node to execute again even when the inputs don't change.
+        You can make this node return a number or a string. This value will be compared to the one returned the last time the node was
+        executed, if it is different the node will be executed again.
+        This method is used in the core repo for the LoadImage node where they return the image hash as a string, if the image hash
+        changes between executions the LoadImage node is executed again.
+    """
+    #@classmethod
+    #def IS_CHANGED(s, image, string_field, int_field, float_field, print_to_screen):
+    #    return ""
+
+
+class LoadAudio:
+    @classmethod
+    def INPUT_TYPES(s):
+        files = [f for f in os.listdir(input_path) if os.path.isfile(os.path.join(input_path, f)) and f.split('.')[-1].lower() in ["wav", "mp3","flac","m4a"]]
+        return {"required":
+                    {"audio": (sorted(files),)},
+                }
+
+    CATEGORY = "AIFSH_Live2D"
+
+    RETURN_TYPES = ("AUDIO",)
+    FUNCTION = "load_audio"
+
+    def load_audio(self, audio):
+        audio_path = folder_paths.get_annotated_filepath(audio)
+        return (audio_path,)
+
+# Set the web directory, any .js file in that directory will be loaded by the frontend as a frontend extension
+WEB_DIRECTORY = "./web"
+
+# A dictionary that contains all nodes you want to export with their names
+# NOTE: names should be globally unique
+NODE_CLASS_MAPPINGS = {
+    "LoadAudio": LoadAudio,
+    "Live2DViewer": Live2DViewer
+}
+
+# A dictionary that contains the friendly/humanly readable titles for the nodes
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "LoadAudio": "LoadAudio",
+    "Live2DViewer": "Live2DViewer Node"
+}