Files
Isulion-ComfyUI_Isulion/theme_handlers/dimension_3d_handler.py
T
2024-11-26 12:13:27 +01:00

71 lines
2.9 KiB
Python

from typing import Dict
from .base_handler import BaseThemeHandler
class Dimension3DThemeHandler(BaseThemeHandler):
"""Handler for 3D dimension-themed prompt generation."""
def generate(self, custom_subject: str = "",
custom_location: str = "",
include_environment: str = "yes",
include_style: str = "yes",
include_effects: str = "yes") -> Dict[str, str]:
"""Generate 3D dimension-themed components."""
components = {}
# Generate subject
if custom_subject:
components["subject"] = (
f"((3D {custom_subject})), "
f"((3D modeling)), ((dimensional design)), "
f"((3D rendering))"
)
else:
subject = self._get_random_choice("dimension_3d.subjects")
style = self._get_random_choice("dimension_3d.styles")
feature = self._get_random_choice("dimension_3d.features")
components["subject"] = (
f"((3D {subject} in {style} style)), "
f"((featuring {feature})), ((3D modeling)), "
f"((dimensional design)), ((3D rendering))"
)
# Generate environment if requested
if include_environment == "yes":
if custom_location:
components["environment"] = (
f"in ((3D {custom_location})) with "
f"((3D environment)), ((dimensional space))"
)
else:
setting = self._get_random_choice("dimension_3d.settings")
element = self._get_random_choice("dimension_3d.elements")
components["environment"] = (
f"in ((3D {setting})) with "
f"((3D {element})), "
f"((dimensional environment)), ((3D space)), "
f"((rendered world))"
)
# Generate style if requested
if include_style == "yes":
render_style = self._get_random_choice("dimension_3d.render_styles")
technique = self._get_random_choice("dimension_3d.techniques")
components["style"] = (
f"((rendered in {render_style} style)), "
f"((using {technique} technique)), "
f"((3D quality)), ((dimensional artistry)), "
f"((professional 3D))"
)
# Generate effects if requested
if include_effects == "yes":
effect = self._get_random_choice("dimension_3d.effects")
lighting = self._get_random_choice("dimension_3d.lighting")
components["effects"] = (
f"with ((3D {effect} effects)), "
f"((dimensional {lighting} lighting)), "
f"((3D finish)), ((render quality))"
)
return components