import gradio as gr
from diffusers import StableDiffusionPipeline
import torch

model_id = "stabilityai/stable-diffusion-2-1"
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
pipe = pipe.to("cpu")

def generate_room(prompt):
    image = pipe(prompt).images[0]
    return image

gr.Interface(fn=generate_room, 
             inputs=gr.Textbox(placeholder="Describe your room design idea..."), 
             outputs="image",
             title="Virtual Room Decorator (Lite)",
             description="Generate interior room designs using simple text prompts. Powered by Stable Diffusion 2.1."
).launch()