Add "Append" button

gradio_app.py

@@ -63,6 +63,17 @@ FOOTER = """
 Para más información, visite el [repositorio del modelo](https://huggingface.co/bertin-project/bertin-gpt-j-6B).
 """.strip()
 
+EXAMPLES = [
+    "",
+    "¿Cuál es la capital de Francia? Respuesta:",
+    """Los templos egipcios fueron construidos para el culto oficial de los dioses y la conmemoración de los faraones del Antiguo Egipto en las regiones bajo su dominio. Los templos eran vistos como el hogar de los dioses o faraones deificados a quienes eran dedicados, y en ellos los faraones y el clero egipcio llevaban a cabo diversos rituales, las funciones centrales de la religión egipcia: realizar ofrendas a sus dioses, recrear pasajes mitológicos mediante festivales y protegerse de las fuerzas del caos. Estos rituales eran vistos como necesarios para que los dioses mantuvieran la maat, el orden divino del universo.
+
+El cuidado del hogar de los dioses era obligación de los faraones, que dedicaron ingentes cantidades de recursos para la construcción y el mantenimiento de los templos. Por necesidad, los faraones delegaban la mayoría de los rituales en una amplia casta sacerdotal, aunque la mayor parte del pueblo llano permanecía al margen de la participación directa en las ceremonias por tener prohibido el acceso a las zonas más sagradas de los templos. A pesar de ello, el templo siempre fue un importante centro religioso para todos los egipcios, que iban a ellos a rezar, realizar ofrendas y buscar la guía de los oráculos.
+
+Pregunta: ¿Quién cuidaba del hogar los dioses?
+Respuesta:""",
+]
+
 class Normalizer:
     def remove_repetitions(self, text):
         """Remove repetitions"""
@@ -128,14 +139,21 @@ class TextGeneration:
                     generated_text = generated_text.replace(text, "", 1).strip()
                 if generated_text:
                     return (
+                        text,
                         text + " " + generated_text,
                         [(text, None), (generated_text, "BERTIN")]
                     )
             if not generated_text:
                 return (
                     "",
-                    [("Tras 10 intentos BERTIN no generó nada. Pruebe cambiando las opciones", "ERROR")]
+                    "",
+                    [("Tras 10 intentos BERTIN no generó nada. Pruebe cambiando las opciones.", "ERROR")]
                 )
+        return (
+            "",
+            "",
+            [("Debe escribir algo primero.", "ERROR")]
+        )
             # return (text + " " + generated_text,
             #     f'<p class="ltr ltr-box">'
             #     f'<span class="result-text">{text} <span>'
@@ -167,6 +185,17 @@ def complete_with_gpt(text, max_length, top_k, top_p, temperature, do_sample, do
     }
     return generator.generate(text, generation_kwargs)
 
+def expand_with_gpt(hidden, text, max_length, top_k, top_p, temperature, do_sample, do_clean):
+    generation_kwargs = {
+        "max_length": max_length,
+        "top_k": top_k,
+        "top_p": top_p,
+        "temperature": temperature,
+        "do_sample": do_sample,
+        "do_clean": do_clean,
+    }
+    return generator.generate(hidden or text, generation_kwargs)
+
 with gr.Blocks() as demo:
     gr.Markdown(HEADER)
     with gr.Row():
@@ -218,18 +247,23 @@ with gr.Blocks() as demo:
                 # help="Si eliminar o no las palabras repetidas y recortar las últimas frases sin terminar.",
             )
         with gr.Column():
-            textbox = gr.Textbox(label="Texto",placeholder="Escriba algo y pulse 'Generar'...", lines=8)
+            textbox = gr.Textbox(label="Texto", placeholder="Escriba algo (o seleccione un ejemplo) y pulse 'Generar'...", lines=8)
+            examples = gr.Dropdown(label="Ejemplos", choices=EXAMPLES, value=None, type="value")
             hidden = gr.Textbox(visible=False, show_label=False)
             with gr.Box():
                 # output = gr.Markdown()
                 output = gr.HighlightedText(label="Resultado", combine_adjacent=True, color_map={"BERTIN": "green", "ERROR": "red"})
             with gr.Row():
                 btn = gr.Button("Generar")
-                btn.click(complete_with_gpt, inputs=[textbox, max_length, top_k, top_p, temperature, do_sample, do_clean], outputs=[hidden, output])
+                btn.click(complete_with_gpt, inputs=[textbox, max_length, top_k, top_p, temperature, do_sample, do_clean], outputs=[textbox, hidden, output])
+                expand_btn = gr.Button("Añadir")
+                expand_btn.click(expand_with_gpt, inputs=[hidden, textbox, max_length, top_k, top_p, temperature, do_sample, do_clean], outputs=[textbox, hidden, output])
+
                 edit_btn = gr.Button("Editar", variant="secondary")
                 edit_btn.click(lambda x: (x, "", []), inputs=[hidden], outputs=[textbox, hidden, output])
-                clean_btn = gr.Button("Limpiar", variant="secondary")
+                clean_btn = gr.Button("Borrar", variant="secondary")
                 clean_btn.click(lambda: ("", "", []), inputs=[], outputs=[textbox, hidden, output])
+            examples.change(lambda x: x, inputs=[examples], outputs=[textbox])
     gr.Markdown(FOOTER)
 
 demo.launch()