import gradio as gr
from transformers import pipeline

def translate_text(frase_en):
    pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es")
    frase_es = pipe(frase_en)
    return frase_es[0]['translation_text']

# Crear la interfaz con Gradio
demo = gr.Interface(
    fn=translate_text,
    inputs=gr.Textbox(placeholder="Ingrese texto en inglés"),
    outputs=gr.Textbox(label="Traducción al español"),
    title="Traductor Inglés - Español",
    description="Ingrese una frase en inglés y obtendrá la traducción en español usando un modelo de IA."
)

demo.launch()