Add camera control node

2026-06-24 09:46:51 +02:00
parent 084702f351
commit 00ac8be640
3 changed files with 371 additions and 3 deletions
@@ -4,20 +4,38 @@ import json

 try:
    from .prompt_builder import (
+        build_camera_config_json,
        build_insta_of_options_json,
        build_insta_of_pair,
        build_prompt,
        build_seed_config_json,
+        camera_angle_choices,
+        camera_distance_choices,
+        camera_lens_choices,
+        camera_mode_choices,
+        camera_orientation_choices,
+        camera_phone_choices,
+        camera_priority_choices,
+        camera_shot_choices,
        category_choices,
        subcategory_choices,
    )
    from .caption_naturalizer import naturalize_caption
 except ImportError:
    from prompt_builder import (
+        build_camera_config_json,
        build_insta_of_options_json,
        build_insta_of_pair,
        build_prompt,
        build_seed_config_json,
+        camera_angle_choices,
+        camera_distance_choices,
+        camera_lens_choices,
+        camera_mode_choices,
+        camera_orientation_choices,
+        camera_phone_choices,
+        camera_priority_choices,
+        camera_shot_choices,
        category_choices,
        subcategory_choices,
    )
@@ -50,6 +68,7 @@ class SxCPPromptBuilder:
            },
            "optional": {
                "seed_config": ("STRING", {"default": "", "multiline": True}),
+                "camera_config": ("STRING", {"default": "", "multiline": True}),
                "extra_positive": ("STRING", {"default": "", "multiline": True}),
                "extra_negative": ("STRING", {"default": "", "multiline": True}),
            },
@@ -81,6 +100,7 @@ class SxCPPromptBuilder:
        trigger,
        prepend_trigger_to_prompt,
        seed_config="",
+        camera_config="",
        extra_positive="",
        extra_negative="",
    ):
@@ -106,6 +126,7 @@ class SxCPPromptBuilder:
            extra_positive=extra_positive or "",
            extra_negative=extra_negative or "",
            seed_config=seed_config or "",
+            camera_config=camera_config or "",
        )
        return (
            row["prompt"],
@@ -167,6 +188,52 @@ class SxCPSeedControl:
        )


+class SxCPCameraControl:
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "camera_mode": (camera_mode_choices(), {"default": "handheld_selfie"}),
+                "shot_size": (camera_shot_choices(), {"default": "auto"}),
+                "angle": (camera_angle_choices(), {"default": "auto"}),
+                "lens": (camera_lens_choices(), {"default": "smartphone_wide"}),
+                "distance": (camera_distance_choices(), {"default": "arm_length"}),
+                "orientation": (camera_orientation_choices(), {"default": "vertical_story"}),
+                "phone_visibility": (camera_phone_choices(), {"default": "phone_visible"}),
+                "priority": (camera_priority_choices(), {"default": "locked"}),
+            }
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("camera_config",)
+    FUNCTION = "build"
+    CATEGORY = "prompt_builder"
+
+    def build(
+        self,
+        camera_mode,
+        shot_size,
+        angle,
+        lens,
+        distance,
+        orientation,
+        phone_visibility,
+        priority,
+    ):
+        return (
+            build_camera_config_json(
+                camera_mode=camera_mode,
+                shot_size=shot_size,
+                angle=angle,
+                lens=lens,
+                distance=distance,
+                orientation=orientation,
+                phone_visibility=phone_visibility,
+                priority=priority,
+            ),
+        )
+
+
 class SxCPCaptionNaturalizer:
    @classmethod
    def INPUT_TYPES(cls):
@@ -223,6 +290,8 @@ class SxCPInstaOFOptions:
                "hardcore_level": (["explicit", "hardcore"], {"default": "hardcore"}),
                "platform_style": (["hybrid", "instagram", "onlyfans"], {"default": "hybrid"}),
                "continuity": (["same_creator_same_room", "same_creator_new_scene"], {"default": "same_creator_same_room"}),
+                "softcore_camera_mode": (camera_mode_choices(), {"default": "handheld_selfie"}),
+                "hardcore_camera_mode": (["same_as_softcore"] + camera_mode_choices(), {"default": "same_as_softcore"}),
            }
        }

@@ -241,6 +310,8 @@ class SxCPInstaOFOptions:
        hardcore_level,
        platform_style,
        continuity,
+        softcore_camera_mode,
+        hardcore_camera_mode,
    ):
        return (
            build_insta_of_options_json(
@@ -252,6 +323,8 @@ class SxCPInstaOFOptions:
                hardcore_level=hardcore_level,
                platform_style=platform_style,
                continuity=continuity,
+                softcore_camera_mode=softcore_camera_mode,
+                hardcore_camera_mode=hardcore_camera_mode,
            ),
        )

@@ -274,6 +347,7 @@ class SxCPInstaOFPromptPair:
            "optional": {
                "seed_config": ("STRING", {"default": "", "multiline": True}),
                "options_json": ("STRING", {"default": "", "multiline": True}),
+                "camera_config": ("STRING", {"default": "", "multiline": True}),
                "extra_positive": ("STRING", {"default": "", "multiline": True}),
                "extra_negative": ("STRING", {"default": "", "multiline": True}),
            },
@@ -306,6 +380,7 @@ class SxCPInstaOFPromptPair:
        prepend_trigger_to_prompt,
        seed_config="",
        options_json="",
+        camera_config="",
        extra_positive="",
        extra_negative="",
    ):
@@ -321,6 +396,7 @@ class SxCPInstaOFPromptPair:
            prepend_trigger_to_prompt=prepend_trigger_to_prompt,
            seed_config=seed_config or "",
            options_json=options_json or "",
+            camera_config=camera_config or "",
            extra_positive=extra_positive or "",
            extra_negative=extra_negative or "",
        )
@@ -339,6 +415,7 @@ class SxCPInstaOFPromptPair:
 NODE_CLASS_MAPPINGS = {
    "SxCPPromptBuilder": SxCPPromptBuilder,
    "SxCPSeedControl": SxCPSeedControl,
+    "SxCPCameraControl": SxCPCameraControl,
    "SxCPCaptionNaturalizer": SxCPCaptionNaturalizer,
    "SxCPInstaOFOptions": SxCPInstaOFOptions,
    "SxCPInstaOFPromptPair": SxCPInstaOFPromptPair,
@@ -347,6 +424,7 @@ NODE_CLASS_MAPPINGS = {
 NODE_DISPLAY_NAME_MAPPINGS = {
    "SxCPPromptBuilder": "SxCP Prompt Builder",
    "SxCPSeedControl": "SxCP Seed Control",
+    "SxCPCameraControl": "SxCP Camera Control",
    "SxCPCaptionNaturalizer": "SxCP Caption Naturalizer",
    "SxCPInstaOFOptions": "SxCP Insta/OF Options",
    "SxCPInstaOFPromptPair": "SxCP Insta/OF Prompt Pair",