Spaces:
Runtime error
Runtime error
| from transformers import pipeline | |
| import gradio as gr | |
| # Define el modelo de traducción inglés-español | |
| modelo_traduccion = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es") | |
| # Define el modelo de clasificación de idiomas (cambiado a fasttext) | |
| modelo_idioma = pipeline("text-classification", model="facebook/fasttext-language-identification") | |
| def detectar_idioma(texto): | |
| """ | |
| Detecta el idioma del texto utilizando el modelo fasttext. | |
| Args: | |
| texto (str): El texto del cual se quiere detectar el idioma. | |
| Returns: | |
| str: El código del idioma detectado (ej: 'en', 'es'). | |
| """ | |
| try: | |
| resultado = modelo_idioma(texto, top_k=1)[0] | |
| # El modelo fasttext devuelve códigos de idioma precedidos por '__label__' | |
| return resultado['label'].replace('__label__', '') | |
| except Exception as e: | |
| print(f"Error en la detección de idioma: {e}") | |
| return None | |
| def traducir_texto(texto_en): | |
| """ | |
| Traduce el texto de entrada del inglés al español, previa verificación del idioma. | |
| Args: | |
| texto_en (str): El texto en inglés que se va a traducir. | |
| Returns: | |
| str: El texto traducido al español o un mensaje de error si el idioma no es inglés. | |
| """ | |
| if not texto_en.strip(): | |
| return "" | |
| idioma_detectado = detectar_idioma(texto_en) | |
| if idioma_detectado == 'en': | |
| try: | |
| resultado = modelo_traduccion(texto_en) | |
| return resultado[0]['translation_text'] | |
| except Exception as e: | |
| return f"Error en la traducción: {e}" | |
| else: | |
| return "Por favor, introduce texto en inglés para traducir." | |
| def borrar_entrada(texto_entrada): | |
| """ | |
| Borra el texto del cuadro de texto de entrada. | |
| Args: | |
| texto_entrada (str): El texto actual en el cuadro de texto de entrada. | |
| Returns: | |
| str: Una cadena vacía. | |
| """ | |
| return "" | |
| with gr.Blocks() as interfaz_traductor: | |
| """ | |
| Define la interfaz gráfica de la aplicación de traducción de inglés a español con detección de idioma (fasttext). | |
| """ | |
| gr.Markdown("# Traductor Inglés a Español") | |
| with gr.Row(): | |
| with gr.Column(): | |
| texto_entrada_en = gr.Textbox(label="Texto en inglés", placeholder="Pega aquí el texto en inglés", lines=10) | |
| boton_borrar = gr.Button("Borrar todo") | |
| with gr.Column(): | |
| texto_salida_es = gr.Textbox(label="Traducción al español", placeholder="La traducción aparecerá aquí", interactive=False, lines=10) | |
| gr.Markdown("*Puedes copiar el texto traducido seleccionándolo y usando Ctrl+C (o Cmd+C en macOS).*") | |
| texto_entrada_en.change(traducir_texto, inputs=texto_entrada_en, outputs=texto_salida_es) | |
| boton_borrar.click(borrar_entrada, inputs=texto_entrada_en, outputs=texto_entrada_en) | |
| interfaz_traductor.launch() |