import streamlit as st 
from diffusers import StableDiffusionPipeline 
import torch

st.set_page_config(page_title="Dreamscape Visualizer") 
st.title("Dreamscape Visualizer")

@st.cache_resource 
def load_pipeline(): 
    pipe = StableDiffusionPipeline.from_pretrained( 
        "runwayml/stable-diffusion-v1-5", 
        torch_dtype=torch.float16 
    ) 
    pipe.to("cuda" if torch.cuda.is_available() else "cpu") 
    return pipe

pipe = load_pipeline() 
style_modifiers = { 
    "Fantasy": "dreamlike fantasy art", 
    "Nightmare": "dark horror surrealism", 
    "Lucid": "vivid hyperreal dreamscape", 
    "Sci-Fi": "futuristic dream world", 
    "Mythical": "ancient mythical surrealism" 
}  

prompt = st.text_area("Describe your dream:", height=100) 
style = st.selectbox("Choose a style:", list(style_modifiers.keys())) 
    if st.button("Visualize Dream"): 
        if not prompt.strip(): 
            st.warning("Please enter a dream description.") 
        else: 
            with st.spinner("Generating your dreamscape..."): 
                final_prompt = f"{prompt}, {style_modifiers[style]}" 
                image = pipe(final_prompt).images[0]
                st.image(image, caption="Here is your dreamscape!",use_column_width=True)