import gradio as gr
from transformers import pipeline

translator = pipeline(
    "translation",
    model="projecte-aina/aina-translator-es-ast",
    # Parámetros para una traducción más literal
    do_sample=False,  # Desactiva la generación aleatoria
    num_beams=1,      # Usa búsqueda greedy
    temperature=0.7,  # Temperatura más baja para traducciones más conservadoras
    max_length=200    # Limita la longitud para evitar generación extra
)

def translate(text):
    if not text:
        return ""
    result = translator(text)
    return result[0]['translation_text']

demo = gr.Interface(
    fn=translate,
    inputs="text",
    outputs="text",
    title="Traductor Español-Asturiano"
)

if __name__ == "__main__":
    demo.launch()