ams-core-api / engines /scene_builder_engine.py
Amssou's picture
Create scene_builder_engine.py
67843ef verified
Raw
History Blame Contribute Delete
1.1 kB
import random
class SceneBuilderEngine:
def __init__(self):
self.foregrounds = [
"ancient stone stairs",
"broken statues",
"glowing crystals",
"burning torches",
"mystical symbols carved in the ground"
]
self.backgrounds = [
"massive mountains in the distance",
"a futuristic city skyline",
"floating islands in the sky",
"a giant moon dominating the horizon",
"storm clouds swirling in the sky"
]
self.environments = [
"thick cinematic fog",
"soft golden sunset light",
"dark storm atmosphere",
"neon reflections everywhere",
"mystical glowing particles floating in the air"
]
def build_scene(self, prompt):
fg = random.choice(self.foregrounds)
bg = random.choice(self.backgrounds)
env = random.choice(self.environments)
scene = f"{prompt}, {fg} in the foreground, {bg} in the background, {env}"
return scene