First commit

__init__.py

@@ -0,0 +1,14 @@
+from .bilbox_photo_prompt import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+
+__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
+
+# ensure .js and web copy
+print("### Copying: BilboX web...")
+
+import shutil, os
+import folder_paths
+
+bx_path = os.path.dirname(__file__)
+comfy_path = os.path.dirname(folder_paths.__file__)
+dest_path = os.path.join(comfy_path, "web", "extensions")
+shutil.copytree(os.path.join(bx_path,"web_to_copy"), os.path.join(dest_path,"bilbox"), dirs_exist_ok=True)  # Fin

bilbox_photo_prompt.py

@@ -0,0 +1,114 @@
+import json
+import os
+
+EMPTY_TEXT = ""
+def read_json_file(file_path):
+    try:
+        # Open file, load JSON content into python dictionary, and return it.
+        with open(file_path, 'r') as file:
+            json_data = json.load(file)
+            return json_data
+    except Exception as e:
+        print(f"An error occurred: {str(e)}")
+
+def get_list(data, name, tag="name", can_be_null=True):
+    ret = [EMPTY_TEXT] if can_be_null else []
+    if(isinstance(data, dict) and name in data):
+        ret.extend([d[tag] for d in data[name] if tag in d])
+    return ret
+
+## Don' need anymore, done in frontend
+# def replace_and_combine(json_data, style, subject, framing, setting_background, lighting, camera_angle, camera_properties, film_types, lenses, filters_effects, photographer):
+#     try:
+#         style = "" if style == EMPTY_TEXT else f"{style} photo of "
+#         framing = "" if framing == EMPTY_TEXT else f", {framing}"
+#         lighting = "" if lighting == EMPTY_TEXT else f", {lighting}"
+#         camera_angle = "" if camera_angle == EMPTY_TEXT else f", {camera_angle}"
+#         camera_properties = "" if camera_properties == EMPTY_TEXT else f", {camera_properties}"
+#         film_types = "" if film_types == EMPTY_TEXT else f", {film_types}"
+#         lenses = "" if lenses == EMPTY_TEXT else f", {lenses}"
+#         filters_effects = "" if filters_effects == EMPTY_TEXT else f", {filters_effects}"
+#         photographer = "" if photographer == EMPTY_TEXT else f", in the style of {photographer}"
+
+#         setting_background = "" if setting_background == "" else f", {setting_background}"
+
+#         return f"{style}{subject}{framing}{setting_background}{lighting}{camera_angle}{camera_properties}{film_types}{lenses}{filters_effects}{photographer}"
+#         #raise ValueError(f"No template found with name '{template_name}'.")
+
+#     except Exception as e:
+#         print(f"An error occurred: {str(e)}")
+
+
+class BilboXPhotoPrompt:
+
+    def __init__(self):
+        pass
+
+
+    @classmethod
+    def INPUT_TYPES(self):
+        # Get current file's directory
+        p = os.path.dirname(os.path.realpath(__file__))
+
+        try:
+            file_path = os.path.join(p, '../../web/extensions/bilbox/PromptGeek/photo_data.json')
+            with open(file_path) as f:
+                self.json_data = json.load(f)
+        except Exception as e:
+            print(f"An error occurred during BilboX's PromptHero Photo Promt initialization: {str(e)}")
+
+        # "[STYLE OF PHOTO] photo of a [SUBJECT], [IMPORTANT FEATURE], [MORE DETAILS], [POSE OR ACTION],[FRAMING],
+        #  [SETTING/BACKGROUND], [LIGHTING],[CAMERA ANGLE], [CAMERA PROPERTIES],in style of [PHOTOGRAPHER]"
+        return {
+            "required": {
+                "modal_combos": ("BOOLEAN", {"default":True} ),
+                "style": ((get_list(self.json_data,"style")), ),
+                "subject": ("STRING", {"default": "", "multiline": True,"placeholder": "[SUBJECT], [IMPORTANT FEATURE], [MORE DETAILS], [POSE OR ACTION]"}),
+                "framing": ((get_list(self.json_data,"framing")), ),
+                "setting_background": ("STRING", {"default": "", "multiline": True}),
+                "lighting": ((get_list(self.json_data,"lighting")), ),
+                "camera_angle": ((get_list(self.json_data,"camera angle")), ),
+                "camera_properties": ((get_list(self.json_data,"camera properties")), ),
+                "film_types": ((get_list(self.json_data,"film types")), ),
+                "lenses": ((get_list(self.json_data,"lenses")), ),
+                "filters_effects": ((get_list(self.json_data,"filters effects")), ),
+                "photographers": ((get_list(self.json_data,"photographers")), ),
+                
+                "preview": ("STRING", {"default": "", "multiline": True, "placeholder":"Preview"}),
+                "log_prompt": (["No", "Yes"], {"default":"No"}),
+            },
+        }
+
+    RETURN_TYPES = ('STRING', 'STRING')
+    RETURN_NAMES = ('full_composed_prompt', 'subject_only')
+    FUNCTION = 'prompt_styler'
+    CATEGORY = 'BilboX'
+
+    def prompt_styler(self, modal_combos, style, subject, framing,
+                      setting_background, lighting, camera_angle,
+                      camera_properties, film_types, lenses,
+                      filters_effects, photographers,
+                      preview, log_prompt):
+        # # Process and combine prompts in templates
+        # # The function replaces the positive prompt placeholder in the template,
+        # # and combines the negative prompt with the template's negative prompt, if they exist.
+        # positive_prompt = replace_and_combine(self.json_data,
+        #     style, subject, framing, setting_background, lighting,
+        #     camera_angle, camera_properties, film_types, lenses,
+        #     filters_effects, photographers)
+        # If logging is enabled (log_prompt is set to "Yes"), 
+        # print the style, positive and negative text, and positive and negative prompts to the console
+        if log_prompt == "Yes":
+            print(f"full_composed_prompt: {preview}")
+            print(f"subject_only: {subject}")
+
+        return preview, subject
+
+
+NODE_CLASS_MAPPINGS = {
+    "BilboXPhotoPrompt": BilboXPhotoPrompt,
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "BilboXPhotoPrompt": "BilboX's PromptGeek Photo Prompt",
+}