import gradio as gr
import torch
from transformers import pipeline
modelo = pipeline("text-to-image", model="runwayml/stable-diffusion-v1-5")

def generarImagen(texto):
    imagen = modelo(prompt=texto).images[0]
    return imagen

with gr.Blocks() as demo:
    gr.Markdown("# Generador Visual de Historia")
    gr.Markdown("Introduce la historia a la que quieras generar el texto")

    with gr.Row():
        with gr.Column(scale=1):
            entrada = gr.Textbox(
                label="Fragmento de la Historia",
                placeholder="Introduce"
            )
            btn = gr.Button("Generar Imagen", variant="primary")

        with gr.Column(scale=1):
            salida = gr.Image(label="Imagen Generada", type="pil")

    btn.click(fn=generarImagen, inputs=entrada, outputs=salida)

demo.launch(theme=gr.themes.Monochrome())