Update Prompt_Library.py

Prompt editor and save button implemented.

Prompt_Library.py

@@ -9,16 +9,18 @@ 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  
-    • "Reload File" Button lädt die aktuelle Datei neu
+    • "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
@@ -33,15 +35,33 @@ class Prompt_Library:
                     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}),
             }
         }
 
@@ -66,23 +86,35 @@ class Prompt_Library:
         self.refresh_projects()
         return "Prompt-Liste aktualisiert."
 
-    def ui_reload_current_file(self):
-        if self.current_file_path and os.path.isfile(self.current_file_path):
-            return f"Datei neu geladen: {os.path.basename(self.current_file_path)}"
-        return "Keine Datei geladen."
+    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": "Reload File", "method": "ui_reload_current_file"},
+            {"label": "Save File", "method": "ui_save_file", "params": ["project", "file_content"]},
         ]
 
-    def get_prompt(self, project, index, randomize):
+    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)
@@ -95,24 +127,24 @@ class Prompt_Library:
         if not os.path.isfile(path):
             return "", "", project, index
 
-        # parse file - sections now auto-numbered based on ### markers
+        # Parse den file_content statt die Datei zu lesen
+        # So arbeiten wir mit den aktuellen Änderungen im Textfeld
         sections = {}
         current_num = 0
         current_lines = []
         
-        with open(path, encoding="utf-8") as f:
-            for line in f:
-                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)
+        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:
@@ -139,3 +171,8 @@ class Prompt_Library:
             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}"