25 lines
980 B
Python
25 lines
980 B
Python
class ArchToneExtractor:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"image": ("IMAGE",),
|
|
"modern_style_weight": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 2.0, "step": 0.1}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("STRING", "STRING")
|
|
RETURN_NAMES = ("color_palette_prompt", "atmosphere_keywords")
|
|
FUNCTION = "extract_tone"
|
|
CATEGORY = "Image/Aesthetic Analysis"
|
|
|
|
def extract_tone(self, image, modern_style_weight):
|
|
# TODO: Implement deep learning based color extraction
|
|
# to avoid cold or outdated architectural rendering styles.
|
|
# Currently a placeholder structure for open source community testing.
|
|
|
|
base_palette = "warm cinematic lighting, high-end modern materials, desaturated earthy tones"
|
|
atmosphere = "photorealistic, architectural visualization, masterpiece, sharp focus"
|
|
|
|
return (base_palette, atmosphere)
|