hexxacubic-ComfyUI-Prompt_Library/Prompt_Library.py

import os
import folder_paths
import random
import sys
import subprocess

class Prompt_Library:
    """
    • project dropdown: '<category>/<project>' choices at init  
    • index slider: free int from 1 to 99  
    • randomize flag: integer 0 oder 1 für Zufall  
    • file_content: zeigt und editiert den Dateiinhalt
    • file reloaded every get_prompt call to reflect external changes  
    • "Prompts Folder" Button öffnet den Prompt-Ordner  
    • "Refresh" Button aktualisiert die Liste der Prompt-Dateien  
    • "Save File" Button speichert Änderungen in die Datei
    """

    def __init__(self):
        self.base = os.path.join(folder_paths.models_dir, "prompts")
        self.refresh_projects()
        self.current_file_path = None
        self.last_loaded_project = None

    def refresh_projects(self):
        # Lese die Projekte jedes Mal neu ein
        base = self.base
        cats = [d for d in os.listdir(base) if os.path.isdir(os.path.join(base, d))]
        proj_choices = []
        for cat in sorted(cats):
            folder = os.path.join(base, cat)
            for fn in sorted(os.listdir(folder)):
                if fn.lower().endswith(".txt"):
                    name = fn[:-4]
                    proj_choices.append(f"{cat}/{name}")
        self._proj_choices = proj_choices

    def load_file_content(self, project):
        """Lädt den Inhalt der ausgewählten Datei"""
        try:
            category, name = project.split("/", 1)
            path = os.path.join(self.base, category, name + ".txt")
            if os.path.isfile(path):
                with open(path, encoding="utf-8") as f:
                    return f.read()
        except:
            pass
        return ""

    @classmethod
    def INPUT_TYPES(s):
        instance = s()
        instance.refresh_projects()
        # Lade initial den ersten Eintrag wenn vorhanden
        initial_content = ""
        if instance._proj_choices:
            initial_content = instance.load_file_content(instance._proj_choices[0])
        
        return {
            "required": {
                "project":   (instance._proj_choices,),
                "index":     ("INT",  {"default": 1, "min": 1, "max": 99}),
                "randomize": ("INT",  {"default": 0, "min": 0, "max": 1}),
                "file_content": ("STRING", {"multiline": True, "default": initial_content}),
            }
        }

    RETURN_TYPES = ("STRING", "STRING", "STRING", "INT")
    FUNCTION     = "get_prompt"
    OUTPUT_NODE  = True
    CATEGORY     = "hexxacubic"

    # --- BUTTON UI ---
    @staticmethod
    def ui_open_prompts_folder():
        base = os.path.join(folder_paths.models_dir, "prompts")
        if sys.platform == "win32":
            os.startfile(base)
        elif sys.platform == "darwin":
            subprocess.Popen(["open", base])
        else:
            subprocess.Popen(["xdg-open", base])
        return "Prompts-Ordner geöffnet."

    def ui_refresh_projects(self):
        self.refresh_projects()
        return "Prompt-Liste aktualisiert."

    def ui_save_file(self, project, file_content):
        """Speichert den Inhalt in die aktuell gewählte Datei"""
        try:
            category, name = project.split("/", 1)
            path = os.path.join(self.base, category, name + ".txt")
            with open(path, "w", encoding="utf-8") as f:
                f.write(file_content)
            return f"Datei gespeichert: {name}.txt"
        except Exception as e:
            return f"Fehler beim Speichern: {str(e)}"

    @classmethod
    def UI_BUTTONS(cls):
        return [
            {"label": "Prompts Folder", "method": "ui_open_prompts_folder"},
            {"label": "Refresh", "method": "ui_refresh_projects"},
            {"label": "Save File", "method": "ui_save_file", "params": ["project", "file_content"]},
        ]

    def get_prompt(self, project, index, randomize, file_content):
        # Nach jedem Render automatisch neu einlesen
        self.refresh_projects()

        # Wenn sich das Projekt geändert hat, lade den neuen Dateiinhalt
        if project != self.last_loaded_project:
            self.last_loaded_project = project
            # Hier würde normalerweise die UI aktualisiert werden
            # In ComfyUI muss das über einen speziellen Mechanismus passieren

        # split 'category/project'
        try:
            category, name = project.split("/", 1)
        except ValueError:
            return "", "", project, index

        path = os.path.join(self.base, category, name + ".txt")
        self.current_file_path = path
        
        if not os.path.isfile(path):
            return "", "", project, index

        # Parse den file_content statt die Datei zu lesen
        # So arbeiten wir mit den aktuellen Änderungen im Textfeld
        sections = {}
        current_num = 0
        current_lines = []
        
        for line in file_content.split('\n'):
            line = line.rstrip("\n")
            if line.startswith("###"):
                # Save previous section if exists
                if current_num > 0 and current_lines:
                    sections[current_num] = current_lines
                # Start new section
                current_num += 1
                current_lines = []
            else:
                if current_num > 0:  # Only collect lines after first ###
                    current_lines.append(line)
        
        # Save last section if exists
        if current_num > 0 and current_lines:
            sections[current_num] = current_lines

        # determine final idx
        if sections:
            max_idx = len(sections)
            if randomize == 1:
                idx = random.randint(1, max_idx)
            else:
                idx = min(index, max_idx)
        else:
            idx = index

        # extract prompts
        lines = sections.get(idx, [])
        if '---' in lines:
            sep = lines.index('---')
            pos = "\n".join(lines[:sep]).strip()
            neg = "\n".join(lines[sep+1:]).strip()
        else:
            pos = "\n".join(lines).strip()
            neg = ""

        return pos, neg, project, idx

    @classmethod
    def IS_CHANGED(s, project, index, randomize, file_content):
        # Wichtig für ComfyUI um zu erkennen wenn sich das Projekt ändert
        return f"{project}_{index}_{randomize}"