# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="somosnlp/Phi-2-LenguajeClaro", trust_remote_code=True)

import gradio as gr
mis_ejemplos = ['''El crecimiento económico, que el año pasado fue cinco veces superior a la media de la zona euro,
 y la mejora del empleo con cifras récord de afiliación, permiten mejorar las previsiones de déficit sin aplicar recortes
y reforzando el Estado de Bienestar.
El inicio de 2024 continúa con la senda de descenso y el déficit del Estado hasta febrero se sitúa en el 0,61%''']
def eval_text (text):
    new_prompt = f"""###System: Lee el siguiente texto y hazlo más claro: 
    ###Texto: 
    {text}  
    ###Texto aclarado:
    """
    result = pipe(new_prompt, max_length=2000)
    return (result[0]['generated_text'])


demo = gr.Interface(fn=eval_text, inputs="text", outputs="text", title="Lenguaje Claro", description="modelo de lenguaje entrenado para aclarar el lenguaje de la administración pública española", examples=mis_ejemplos) 
demo.launch()