From stable_diffusion import StableDiffusionPipeline import torch from PIL impor

from stable_diffusion import StableDiffusionPipeline import torch from PIL import Image # Load the Stable Diffusion model pipe = StableDiffusionPipeline.from_pretrained("stable-diffusion-v1-4") pipe = pipe.to("cuda") # Use GPU if available # Define the detailed puzzle game prompt prompts = { "puzzle_game_scene": ( "A first-person perspective scene from a mind-bending puzzle game. The setting is a dimly lit, futuristic laboratory filled with intricate, high-tech machinery and holographic interfaces. " "The player's viewpoint reveals a complex puzzle consisting of glowing, interconnected gears and levers, with enigmatic symbols floating in the air. " "In the foreground, a large, transparent console displays cryptic code and a partially solved puzzle, while in the background, a vast array of shifting, rotating structures adds to the sense of depth and complexity. " "The room's atmosphere is tense and immersive, with shadows playing across the walls and soft, ambient lighting illuminating key elements of the puzzle. " "The player's hand, seen in the foreground, reaches out to interact with a control panel, emphasizing the sense of immersion and involvement in solving the puzzle. " "The overall scene combines high-tech aesthetics with a feeling of isolation and challenge, inviting the player to engage deeply with the puzzle mechanics." ) } # Generate the image for the prompt for name, prompt in prompts.items(): # Generate the image using the prompt image = pipe(prompt).images[0] # Save the generated image filename = f"{name.lower().replace(' ', '_')}.png" image.save(filename) print(f"Saved image: {filename}") # Optionally, show the generated image image.show()