Add more elements to the interface

app.py

@@ -1,13 +1,74 @@
 import gradio as gr
 
-def update(name):
-    return f"Welcome to Gradio, {name}!"
+author_set = {'Leopoldo Lugones', 'Nacho Buzón', 'Octavio Paz', 'Luis Cañizal de la Fuente', 'Juan de Salinas', 'Vicente Huidobro', 'Hilario Barrero',
+              'Ramón de Campoamor', 'Anna Ajmátova', 'Víctor Hugo López Cancino', 'Ramón María del Valle-Inclán', 'Infantiles', 'Jorge Luis Borges',
+              'Carlos Bousoño', 'Gonzalo Rojas', 'Juan Ruiz, Arcipreste de Hita', 'Rubén Izaguirre Fiallos', 'Juan Meléndez Valdés',
+              'José María de Heredia', 'Jaime Sabines', 'Alejandra Pizarnik', 'María Cristina Azcona', 'Guillermo Aguirre y Fierro',
+              'Miguel Ramos Carrión', 'José de Espronceda', 'Juan del Encina', 'Jorge Guillén', 'Manuel Machado', 'José Santos Chocano', 'Meira Delmar',
+              'Iván Tubau', 'Tirso de Molina', 'Oliverio Girondo', 'Justo Braga', 'Consuelo Hernández', 'Belén Reyes', 'Pablo Neruda',
+              'Francisco de Aldana', 'Bertolt Brecht', 'José Antonio Labordeta', 'Emilio Prados', 'Porfirio Barba Jacob', 'Leopoldo Marechal',
+              'Aurelio González Ovies', 'Darío Jaramillo Agudelo', 'Víctor Botas', 'Leopoldo María Panero', 'Juan de Mena', 'Tomás de Iriarte',
+              'Gabriela Mistral', 'Gonzalo de Berceo', 'Antonio Machado', 'Santiago Montobbio', 'Pedro Calderón de la Barca', 'Dionisio Ridruejo',
+              'Atahualpa Yupanqui', 'Nicolás Guillén', 'Claudio Rodríguez', 'José María Blanco White', 'Gil Vicente', 'Corina Bruni', 'Gabriel Celaya',
+              'Javier Alvarado', 'Rosalía de Castro', 'Gustavo Pereira', 'Miguel de Cervantes y Saavedra', 'Jorge Riechmann', 'José Juan Tablada',
+              'Lope de Vega', 'Basilio Sánchez', 'Alfonso X el sabio', 'Rafael de León', 'Eunice Odio', 'Manuel Acuña', 'Víctor Jiménez',
+              'José Asunción Silva', 'Omar García Ramírez', 'Luis Cernuda', 'perdón', 'Blas de Otero', 'Luis Benítez', 'Ángeles Carbajal',
+              'Manuel Bretón de los Herreros', 'Baldomero Fernández Moreno', 'Luis Barahona de Soto', 'Guillermo Valencia', 'José María Hinojosa',
+              'Macedonio Fernández', 'Manuel Gutiérrez Nájera', 'Alfredo Buxán', 'Salvador Novo', 'José Ángel Valente', 'José Cadalso',
+              'Juan Ramón Mansilla', 'Ana Istarú', 'Juan Ramón Jiménez', 'Miguel Ángel Asturias', 'Ángel González', 'amistad', 'Alfredo Lavergne',
+              'Xavier Villaurrutia', 'René Chacón Linares', 'Carolina Coronado', 'Bartolomé Leonardo de Argensola', 'Marilina Rébora',
+              'Vicente Aleixandre', 'Alberto Girri', 'Juana de Ibarbourou', 'Ricardo Dávila Díaz Flores', 'Garcilaso de la Vega',
+              'Lupercio Leonardo de Argensola', 'Enrique Lihn', 'Julia de Burgos', 'Mariano Brull', 'Efraín Huerta', 'Roxana Popelka',
+              'Pelayo Fueyo', 'San Juan de la Cruz', 'Vicente García', 'Basilio Fernández', 'Paz Díez Taboada', 'cristianos', 'León Felipe',
+              'Diana Bellessi', 'Genaro Ortega Gutiérrez', 'Cristóbal de Castillejo', 'Gioconda Belli', 'Iacyr Anderson Freitas',
+              'Juan José Vélez Otero', 'Ezequiel Martínez Estrada', 'Juan de Arguijo', 'Gertrudis Gómez de Avellaneda', 'Marcos Rafael Blanco Belmonte',
+              'Julio Aumente', 'Ramón López Velarde', 'para la familia', 'Antonia Álvarez Álvarez', 'José Zorrilla', 'Juan Luis Panero',
+              'Teresa Palazzo Conti', 'Claribel Alegría', 'Francisco de Medrano', 'Antonio Colinas', 'Jordi Doce', 'Ismael Enrique Arciniegas',
+              'Josefina Plá', 'José Agustín Goytisolo', 'Blanca Andreu', 'Enrique González Martínez', 'José García Nieto', 'Ernesto Cardenal',
+              'Pedro Luis Menéndez', 'Carmen Conde Abellán', 'Salvador Rueda', 'Dulce María Loynaz', 'Odette Alonso', 'Manuel Altolaguirre',
+              'Byron Espinoza', 'Francisco Álvarez', 'Vicente Gerbasi', 'César Vallejo', 'Gloria Fuertes', 'Sor Juana Inés de la Cruz',
+              'Francisco de la Torre', 'Francisco Matos Paoli', 'Rubén Darío', 'Rafael Pombo', 'Mara Romero', 'José Albi', 'Francisco de Quevedo',
+              'Juan de Dios Peza', 'Leopoldo Panero', 'Fernando de Herrera', 'Victoriano Crémer', 'Ana Rossetti', 'Gabriel García Márquez',
+              'Teresa Domingo Català', 'Melchor de Palau', 'Miguel Rasch Isla', 'Luis Antonio de Villena', 'Mario Benedetti', 'Ramón Pérez de Ayala',
+              'Antonio Plaza Llamas', 'David Escobar Galindo', 'Mario Meléndez', 'José Gorostiza', 'Delfina Acosta', 'en español', 'Delmira Agustini',
+              'José Luis Piquero', 'Baltasar del Alcázar', 'Nimia Vicéns', 'Rafael Alberti', 'María Eugenia Caseiro', 'Nicomedes Santa Cruz',
+              'Carlos Pellicer', 'Luis de Góngora', 'Manuel Alcántara', 'Toni García Arias', 'Antonio Fernández Lera', 'José María Pemán',
+              'Dina Posada', 'Gaspar Melchor de Jovellanos', 'Alfonso Reyes', 'José María Gabriel y Galán', 'Manuel José Othón', 'Luciano Castañón',
+              'Luis Alberto de Cuenca', 'Andrés Eloy Blanco', 'Luis Antonio Chávez', 'Pedro Salinas', 'Luis Palés Matos', 'Álvaro García',
+              'Pablo de Rokha', 'Dámaso Alonso', 'Luis Álvarez Piner', 'Salvador García Ramírez', 'Roque Dalton', 'Gerardo Diego',
+              'Felipe Benítez Reyes', 'William Shakespeare', 'José Ángel Buesa', 'Miguel Florián', 'Luis Gonzaga Urbina', 'Jesús Hilario Tundidor',
+              'Amado Nervo', 'Miguel Hernández', 'Federico García Lorca', 'José Martí', 'Oscar Ferreiro', 'Pedro Miguel Lamet', 'Fray Luis de León',
+              'Julio Flórez Roa', 'León de Greiff', 'Gustavo Adolfo Bécquer', 'Carlos Edmundo de Ory', 'Miguel de Unamuno', 'Manuel del Cabral',
+              'Oscar Acosta', 'José Lezama Lima', 'Hernando de Acuña', 'Ángel García Aller', 'Salvador Díaz Mirón', 'Ricardo Molinari',
+              'Julio Herrera y Reissig', 'Francisco Luis Bernárdez', 'Fa Claes', 'Jorge Debravo', 'Francisco de Figueroa', 'Marqués de Santillana',
+              'Eugenio Florit', 'José Gautier Benítez', 'Óscar Hahn', 'Andrés Bello', 'Santa Teresa de Jesús, Sánchez de Cep', 'Juan Liscano',
+              'Jorge Teillier', 'Félix María de Samaniego', 'Nicolás Fernández de Moratín', 'Juan Boscán', 'Manuel María Flores', 'Gutierre de Cetina',
+              'Alfonsina Storni', 'José Luis Rey Cano', 'Jorge Manrique', 'Nicanor Parra'}
+
+
+def poem_generate(author, sentiment, words, text):
+    debug_poem = 'Y cuando se despertó, el dinosaurio seguía ahí'
+    # Pending: Translate poem to English, so that text can be the input of the latentdiffussion
+    gen_image = poem_to_image(debug_poem)
+    return debug_poem, gen_image
+
+
+def poem_to_image(poem):
+    print(['poem_to_image', 'start'])
+    poem = " ".join(poem.split('\n'))
+    poem = poem + " oil on canvas."
+    steps, width, height, images, diversity = '50', '256', '256', '1', 15
+    img = gr.Interface.load("spaces/multimodalart/latentdiffusion")(poem,
+                                                                    steps, width, height, images, diversity)[0]
+    print(['poem_to_image', 'end'])
+    return img
+
 
 demo = gr.Blocks()
 
 with demo:
     gr.Markdown(
-    """This space is taking as a base the [Spanish Poem Generation](https://huggingface.co/spaces/hackathon-pln-es/poem-generation-es) created during First Spanish Hackathon of NLP.
+        """This space is taking as a base the [Spanish Poem Generation](https://huggingface.co/spaces/hackathon-pln-es/poem-generation-es) created during First Spanish Hackathon of NLP.
     
     A [t5-small](https://huggingface.co/hackathon-pln-es/poem-gen-spanish-t5-small) was trained with this [dataset](https://huggingface.co/datasets/hackathon-pln-es/spanish-poetry-dataset) of spanish poems. 
     
@@ -15,11 +76,22 @@ with demo:
     
     With ❤️ from [Alberto Carmona Barthelemy](https://huggingface.co/milyiyo) from 🇨🇺.
     """)
-    inp = gr.Textbox(placeholder="What is your name?")
-    out = gr.Textbox()
+    with gr.Row():
+        input_author = gr.inputs.Dropdown(
+            sorted(author_set), type="value", label='Nombre del autor'),
+        input_sentiment = gr.inputs.Radio(
+            ['positivo', 'negativo'], type="value", label='Sentimiento del texto generado'),
+        input_include_words = gr.inputs.Textbox(
+            lines=1, placeholder='palabra_1, palabra_2, ..., palabra_n', label='Palabras que desea input_word = incluir'),
+        input_initial_text = gr.inputs.Textbox(
+            lines=4, placeholder='texto inicial', label='Texto inicial')
+        output_poem_txt = gr.Textbox(lines=7)
+        output_image = gr.Image(type="filepath", shape=(256, 256))
 
-    inp.change(fn=update, 
-               inputs=inp, 
-               outputs=out)
+    b1 = gr.Button("Generate Poem & Illustration")
+    b1.click(poem_generate,
+             inputs=[input_author, input_sentiment,
+                     input_include_words, input_initial_text],
+             outputs=[output_poem_txt, output_image])
 
-demo.launch()
+demo.launch(enable_queue=True, debug=True)