| import random |
| import yaml |
| import os |
| from typing import Dict, List, Any |
|
|
| class MusicPromptGenerator: |
| """ |
| 音樂 AI Prompt 生成器 - ComfyUI 節點版本 |
| """ |
| |
| def __init__(self): |
| self.keyword_map = { |
| "love": {"mood": "romantic", "genre": "pop", "theme": "love"}, |
| "rain": {"mood": "melancholic", "genre": "lofi", "theme": "rain"}, |
| "battle": {"mood": "energetic", "genre": "orchestral", "theme": "journey"}, |
| "dream": {"mood": "dreamy", "genre": "ambient", "theme": "fantasy"}, |
| "night": {"mood": "serene", "genre": "jazz", "theme": "night"}, |
| "space": {"mood": "mysterious", "genre": "electronic", "theme": "scifi"}, |
| "road": {"mood": "nostalgic", "genre": "rock", "theme": "journey"}, |
| "sunrise": {"mood": "happy", "genre": "ambient", "theme": "nature"}, |
| "sad": {"mood": "melancholic", "genre": "piano", "theme": "memory"} |
| } |
| |
| self.default_data = { |
| "moods": [ |
| {"en": "happy", "zh": "快樂"}, {"en": "dark", "zh": "陰鬱"}, |
| {"en": "nostalgic", "zh": "懷舊"}, {"en": "melancholic", "zh": "憂傷"}, |
| {"en": "euphoric", "zh": "狂喜"}, {"en": "mysterious", "zh": "神秘"}, |
| {"en": "serene", "zh": "寧靜"}, {"en": "tense", "zh": "緊張"}, |
| {"en": "romantic", "zh": "浪漫"}, {"en": "energetic", "zh": "活力"} |
| ], |
| "genres": [ |
| {"en": "pop", "zh": "流行"}, {"en": "jazz", "zh": "爵士"}, |
| {"en": "rock", "zh": "搖滾"}, {"en": "electronic", "zh": "電子"}, |
| {"en": "classical", "zh": "古典"}, {"en": "folk", "zh": "民謠"}, |
| {"en": "lofi", "zh": "低保真"}, {"en": "orchestral", "zh": "管弦樂"}, |
| {"en": "ambient", "zh": "環境音樂"}, {"en": "synthwave", "zh": "合成器浪潮"} |
| ], |
| "themes": [ |
| {"en": "love", "zh": "愛情"}, {"en": "rain", "zh": "雨天"}, |
| {"en": "city", "zh": "都市夜景"}, {"en": "ocean", "zh": "海洋"}, |
| {"en": "fantasy", "zh": "奇幻"}, {"en": "scifi", "zh": "科幻"}, |
| {"en": "journey", "zh": "旅程"}, {"en": "memory", "zh": "回憶"}, |
| {"en": "nature", "zh": "自然"}, {"en": "night", "zh": "夜晚"} |
| ], |
| "instruments": [ |
| {"en": "piano", "zh": "鋼琴"}, {"en": "guitar", "zh": "吉他"}, |
| {"en": "synth", "zh": "合成器"}, {"en": "strings", "zh": "弦樂"}, |
| {"en": "drums", "zh": "鼓組"}, {"en": "bass", "zh": "貝斯"}, |
| {"en": "violin", "zh": "小提琴"}, {"en": "saxophone", "zh": "薩克斯風"}, |
| {"en": "flute", "zh": "長笛"}, {"en": "harp", "zh": "豎琴"} |
| ], |
| "vocal_styles": [ |
| {"en": "female vocals", "zh": "女聲"}, {"en": "male vocals", "zh": "男聲"}, |
| {"en": "choir", "zh": "合唱"}, {"en": "layered vocals", "zh": "層疊人聲"}, |
| {"en": "whisper", "zh": "氣音呢喃"}, {"en": "falsetto", "zh": "假音"}, |
| {"en": "rap", "zh": "說唱"}, {"en": "spoken word", "zh": "念白"} |
| ], |
| "descriptions": [ |
| {"en": "cinematic", "zh": "電影感"}, {"en": "lofi", "zh": "輕鬆低保真"}, |
| {"en": "acoustic", "zh": "純淨原聲"}, {"en": "orchestral", "zh": "管弦編制"}, |
| {"en": "minimal", "zh": "極簡主義"}, {"en": "experimental", "zh": "實驗前衛"}, |
| {"en": "vintage", "zh": "復古風"}, {"en": "futuristic", "zh": "未來感"} |
| ], |
| "tempos": [ |
| {"en": "slow", "zh": "慢板"}, {"en": "medium", "zh": "中板"}, |
| {"en": "fast", "zh": "快板"}, {"en": "adagio", "zh": "柔板"}, |
| {"en": "allegro", "zh": "快板"}, {"en": "andante", "zh": "行板"} |
| ] |
| } |
|
|
| @classmethod |
| def INPUT_TYPES(cls): |
| return { |
| "required": { |
| "mode": (["random", "keyword"], {"default": "random"}), |
| "keyword": ("STRING", {"default": "love", "multiline": False}), |
| "num_prompts": ("INT", {"default": 1, "min": 1, "max": 10}), |
| "language": (["en", "zh", "both"], {"default": "en"}), |
| "template_type": (["default", "cinematic", "minimal", "custom"], {"default": "default"}), |
| "custom_template": ("STRING", { |
| "default": "A {mood} {genre} track about {theme}, featuring {instrument} and {vocal_style}, {description} style, {tempo} tempo.", |
| "multiline": True |
| }), |
| "seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff}), |
| }, |
| "optional": { |
| "yaml_file": ("STRING", {"default": ""}), |
| } |
| } |
|
|
| RETURN_TYPES = ("STRING",) |
| RETURN_NAMES = ("music_prompt",) |
| FUNCTION = "generate_prompt" |
| CATEGORY = "AI Music" |
| DESCRIPTION = "生成音樂 AI 提示詞" |
|
|
| def load_yaml_data(self, file_path): |
| """載入 YAML 詞庫檔案""" |
| if not file_path or not os.path.exists(file_path): |
| return self.default_data |
| |
| try: |
| with open(file_path, 'r', encoding='utf-8') as f: |
| return yaml.safe_load(f) |
| except Exception as e: |
| print(f"載入 YAML 檔案失敗: {e}") |
| return self.default_data |
|
|
| def fill_template(self, mood, genre, theme, instrument, vocal, desc, tempo, template, language): |
| """填充模板""" |
| en_template = template |
| zh_template = "一首{mood}{genre}的音樂，主題是{theme}，包含{instrument}與{vocal_style}，風格{description}，節奏{tempo}。" |
| |
| |
| mood_dict = mood if isinstance(mood, dict) else {"en": mood, "zh": mood} |
| genre_dict = genre if isinstance(genre, dict) else {"en": genre, "zh": genre} |
| theme_dict = theme if isinstance(theme, dict) else {"en": theme, "zh": theme} |
| instrument_dict = instrument if isinstance(instrument, dict) else {"en": instrument, "zh": instrument} |
| vocal_dict = vocal if isinstance(vocal, dict) else {"en": vocal, "zh": vocal} |
| desc_dict = desc if isinstance(desc, dict) else {"en": desc, "zh": desc} |
| tempo_dict = tempo if isinstance(tempo, dict) else {"en": tempo, "zh": tempo} |
| |
| en = en_template.format( |
| mood=mood_dict["en"], genre=genre_dict["en"], theme=theme_dict["en"], |
| instrument=instrument_dict["en"], vocal_style=vocal_dict["en"], |
| description=desc_dict["en"], tempo=tempo_dict["en"] |
| ) |
| zh = zh_template.format( |
| mood=mood_dict["zh"], genre=genre_dict["zh"], theme=theme_dict["zh"], |
| instrument=instrument_dict["zh"], vocal_style=vocal_dict["zh"], |
| description=desc_dict["zh"], tempo=tempo_dict["zh"] |
| ) |
| |
| if language == "en": |
| return en |
| elif language == "zh": |
| return zh |
| else: |
| return f"{en}\n{zh}" |
|
|
| def get_template(self, template_type, custom_template): |
| """取得模板""" |
| templates = { |
| "default": "A {mood} {genre} track about {theme}, featuring {instrument} and {vocal_style}, {description} style, {tempo} tempo.", |
| "cinematic": "Cinematic {genre} music with {mood} atmosphere, theme: {theme}, featuring {instrument} and {vocal_style}, {tempo} tempo.", |
| "minimal": "{mood} {genre} | {theme} | {instrument} | {vocal_style} | {description} | {tempo}", |
| } |
| return custom_template if template_type == "custom" else templates.get(template_type, templates["default"]) |
|
|
| def generate_prompt(self, mode, keyword, num_prompts, language, template_type, custom_template, seed, yaml_file=""): |
| |
| if seed > 0: |
| random.seed(seed) |
| |
| |
| data = self.load_yaml_data(yaml_file) if yaml_file else self.default_data |
| |
| |
| template = self.get_template(template_type, custom_template) |
| |
| results = [] |
| |
| for _ in range(num_prompts): |
| if mode == "keyword": |
| |
| match = self.keyword_map.get(keyword.lower(), {}) |
| |
| def find_item(category, default_value): |
| target = match.get(category, default_value) |
| for item in data[category + "s"]: |
| if isinstance(item, dict) and item.get("en") == target: |
| return item |
| return random.choice(data[category + "s"]) |
| |
| mood = find_item("mood", "melancholic") |
| genre = find_item("genre", "pop") |
| theme = find_item("theme", "memory") |
| instrument = random.choice(data["instruments"]) |
| vocal = random.choice(data["vocal_styles"]) |
| desc = random.choice(data["descriptions"]) |
| tempo = random.choice(data["tempos"]) |
| |
| else: |
| |
| mood = random.choice(data["moods"]) |
| genre = random.choice(data["genres"]) |
| theme = random.choice(data["themes"]) |
| instrument = random.choice(data["instruments"]) |
| vocal = random.choice(data["vocal_styles"]) |
| desc = random.choice(data["descriptions"]) |
| tempo = random.choice(data["tempos"]) |
| |
| prompt = self.fill_template(mood, genre, theme, instrument, vocal, desc, tempo, template, language) |
| results.append(prompt) |
| |
| |
| final_prompt = "\n\n".join(results) if num_prompts > 1 else results[0] |
| |
| return (final_prompt,) |
|
|
| |
| NODE_CLASS_MAPPINGS = { |
| "MusicPromptGenerator": MusicPromptGenerator |
| } |
|
|
| NODE_DISPLAY_NAME_MAPPINGS = { |
| "MusicPromptGenerator": "🎵 Music AI Prompt Generator" |
| } |
