import gradio as gr
from diffusers import StableDiffusionPipeline
import torch

pipe = StableDiffusionPipeline.from_pretrained(
    "stabilityai/sd-turbo",
    torch_dtype=torch.float32
)

pipe = pipe.to("cpu")

def generar_imagen(prompt):
    imagen = pipe(
        prompt,
        num_inference_steps=3,  # antes era 1
        guidance_scale=7.5      # mejora calidad
    ).images[0]
    return imagen

with gr.Blocks() as demo:
    gr.Markdown("# 🎨 ImaginaAI")
    gr.Markdown("Genera imágenes con inteligencia artificial")

    prompt = gr.Textbox(label="Escribe tu idea")

    boton_generar = gr.Button("Generar imagen")
    resultado = gr.Image(label="Imagen generada")

    boton_generar.click(fn=generar_imagen, inputs=prompt, outputs=resultado)

demo.launch()