adding app.py

app.py

@@ -0,0 +1,158 @@
+import gradio as gr
+import json
+import os
+from generate_eban import translate_es_to_mapudungun
+
+__TITLE: str = "# Traductor Español - Mapudungun"
+
+__DESCRIPTION: str = """
+Ingrese texto en español para traducirlo al mapudungun.
+Luego, puede dar feedback sobre la traducción para ayudar a mejorar nuestro modelo.
+"""
+
+__FOOTER: str = """
+---
+(c) 2024 [Inria Chile Research Center](https://inria.cl)
+"""
+
+
+# using eban's model
+def translate_text(text):
+    return translate_es_to_mapudungun(text)
+
+# Updated function to save feedback
+def save_feedback(input_text, output_text, feedback, correction=None):
+    feedback_data = {
+        "input": input_text,
+        "output": output_text,
+        "feedback": feedback,
+        "correction": correction
+    }
+    
+    # Ensure the feedback directory exists
+    os.makedirs("feedback", exist_ok=True)
+    
+    # Save feedback to a JSON file
+    with open(f"feedback/feedback_{len(os.listdir('feedback'))}.json", "w") as f:
+        json.dump(feedback_data, f)
+
+# Main translation function
+def translate_and_store(input_text):
+    translated_text = translate_text(input_text)
+    return translated_text, *clear_feedback()
+
+def clear_feedback():
+    return (gr.update(value=""), 
+            gr.update(value="", visible=False), 
+            gr.update(visible=False), 
+            gr.update(visible=False))
+
+# Updated feedback functions
+def like_translation(input_text, output_text):
+    save_feedback(input_text, output_text, "liked")
+    return ("Gracias por tu feedback positivo!", 
+            gr.update(visible=False), 
+            gr.update(visible=False), 
+            gr.update(visible=False))
+
+def dislike_translation(input_text, output_text):
+    return ("¿Puedes ingresar la traducción correcta?", 
+            gr.update(visible=True), 
+            gr.update(visible=True), 
+            gr.update(visible=True), 
+            gr.update(value="", visible=False), 
+            gr.update(visible=False))
+
+def yes_provide_correction():
+    return (gr.update(visible=True), 
+            gr.update(visible=True), 
+            gr.update(visible=False), 
+            gr.update(visible=False))
+
+def no_provide_correction(input_text, output_text):
+    save_feedback(input_text, output_text, "disliked")
+    return ("Gracias por usar la aplicación.", 
+            gr.update(visible=False), 
+            gr.update(visible=False), 
+            gr.update(visible=False))
+
+def submit_correction(input_text, output_text, correction):
+    save_feedback(input_text, output_text, "disliked", correction)
+    return ("Gracias. Tu corrección nos ayudará a mejorar nuestro modelo de traducción.", 
+            gr.update(value="", visible=False), 
+            gr.update(visible=False), 
+            gr.update(visible=False))
+
+# Create the Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown(__TITLE)
+    gr.Markdown(__DESCRIPTION)
+    
+    with gr.Row():
+        input_text = gr.Textbox(
+            label="Ingrese texto en español",
+            placeholder="Escriba aquí su texto en español...",
+            lines=5,  # Increased number of visible lines
+            max_lines=10,  # Maximum number of lines before scrolling
+        )
+        output_text = gr.Textbox(
+            label="Traducción al mapudungun",
+            lines=5,  # Increased number of visible lines
+            max_lines=10,  # Maximum number of lines before scrolling
+        )
+    
+    translate_btn = gr.Button("Translate")
+
+    gr.Markdown("## ¿Cuál es su opinión sobre la traducción?")
+
+    with gr.Row():
+        like_btn = gr.Button("Me gusta la traducción")
+        dislike_btn = gr.Button("No me gusta")
+    
+    feedback_output = gr.Textbox(label="Feedback", interactive=False)
+    
+    with gr.Row(visible=False) as correction_choice_row:
+        yes_btn = gr.Button("Sí")
+        no_btn = gr.Button("No")
+    
+    correction_input = gr.Textbox(label="Traducción correcta", visible=False)
+    submit_correction_btn = gr.Button("Enviar corrección", visible=False)
+    
+    translate_btn.click(
+        translate_and_store, 
+        inputs=input_text, 
+        outputs=[output_text, feedback_output, correction_input, submit_correction_btn, correction_choice_row]
+    )
+    like_btn.click(
+        like_translation, 
+        inputs=[input_text, output_text], 
+        outputs=[feedback_output, correction_input, submit_correction_btn, correction_choice_row]
+    )
+    dislike_btn.click(
+        dislike_translation, 
+        inputs=[input_text, output_text], 
+        outputs=[feedback_output, correction_choice_row, yes_btn, no_btn, correction_input, submit_correction_btn]
+    )
+    yes_btn.click(
+        yes_provide_correction, 
+        outputs=[correction_input, submit_correction_btn, yes_btn, no_btn]
+    )
+    no_btn.click(
+        no_provide_correction, 
+        inputs=[input_text, output_text], 
+        outputs=[feedback_output, correction_input, submit_correction_btn, correction_choice_row]
+    )
+    submit_correction_btn.click(
+        submit_correction, 
+        inputs=[input_text, output_text, correction_input], 
+        outputs=[feedback_output, correction_input, submit_correction_btn, correction_choice_row]
+    )
+    
+    # Add footer and logo
+    with gr.Row():
+        with gr.Column(scale=2):
+            gr.Markdown(__FOOTER)
+        with gr.Column(scale=1):
+            gr.Image("logo.png", show_label=False, container=False, interactive=False)
+
+demo.launch()

environment.yml

@@ -0,0 +1,177 @@
+name: translator_env
+channels:
+  - conda-forge
+dependencies:
+  - aiofiles=23.2.1
+  - annotated-types=0.7.0
+  - anyio=4.6.0
+  - aom=3.9.1
+  - brotli=1.1.0
+  - brotli-bin=1.1.0
+  - brotli-python=1.1.0
+  - bzip2=1.0.8
+  - ca-certificates=2024.8.30
+  - cairo=1.18.0
+  - certifi=2024.8.30
+  - cffi=1.17.1
+  - charset-normalizer=3.3.2
+  - click=8.1.7
+  - colorama=0.4.6
+  - contourpy=1.3.0
+  - cycler=0.12.1
+  - dav1d=1.2.1
+  - dnspython=2.6.1
+  - email-validator=2.2.0
+  - email_validator=2.2.0
+  - exceptiongroup=1.2.2
+  - expat=2.6.3
+  - fastapi=0.115.0
+  - fastapi-cli=0.0.5
+  - ffmpeg=7.1.0
+  - ffmpy=0.3.0
+  - filelock=3.16.1
+  - font-ttf-dejavu-sans-mono=2.37
+  - font-ttf-inconsolata=3.000
+  - font-ttf-source-code-pro=2.038
+  - font-ttf-ubuntu=0.83
+  - fontconfig=2.14.2
+  - fonts-conda-ecosystem=1
+  - fonts-conda-forge=1
+  - fonttools=4.54.1
+  - freetype=2.12.1
+  - fribidi=1.0.10
+  - fsspec=2024.9.0
+  - gmp=6.3.0
+  - gradio=4.44.1
+  - gradio-client=1.3.0
+  - graphite2=1.3.13
+  - h11=0.14.0
+  - h2=4.1.0
+  - harfbuzz=9.0.0
+  - hpack=4.0.0
+  - httpcore=1.0.6
+  - httpx=0.27.2
+  - huggingface_hub=0.25.1
+  - hyperframe=6.0.1
+  - icu=75.1
+  - idna=3.10
+  - importlib-resources=6.4.5
+  - importlib_resources=6.4.5
+  - jinja2=3.1.4
+  - kiwisolver=1.4.7
+  - lame=3.100
+  - lcms2=2.16
+  - lerc=4.0.0
+  - libabseil=20240722.0
+  - libass=0.17.3
+  - libblas=3.9.0
+  - libbrotlicommon=1.1.0
+  - libbrotlidec=1.1.0
+  - libbrotlienc=1.1.0
+  - libcblas=3.9.0
+  - libcxx=19.1.1
+  - libdeflate=1.21
+  - libexpat=2.6.3
+  - libffi=3.4.2
+  - libgfortran=5.0.0
+  - libgfortran5=13.2.0
+  - libglib=2.82.1
+  - libhwloc=2.11.1
+  - libiconv=1.17
+  - libintl=0.22.5
+  - libjpeg-turbo=3.0.0
+  - liblapack=3.9.0
+  - libopenblas=0.3.27
+  - libopenvino=2024.4.0
+  - libopenvino-arm-cpu-plugin=2024.4.0
+  - libopenvino-auto-batch-plugin=2024.4.0
+  - libopenvino-auto-plugin=2024.4.0
+  - libopenvino-hetero-plugin=2024.4.0
+  - libopenvino-ir-frontend=2024.4.0
+  - libopenvino-onnx-frontend=2024.4.0
+  - libopenvino-paddle-frontend=2024.4.0
+  - libopenvino-pytorch-frontend=2024.4.0
+  - libopenvino-tensorflow-frontend=2024.4.0
+  - libopenvino-tensorflow-lite-frontend=2024.4.0
+  - libopus=1.3.1
+  - libpng=1.6.44
+  - libprotobuf=5.27.5
+  - libsqlite=3.46.1
+  - libtiff=4.7.0
+  - libvpx=1.14.1
+  - libwebp-base=1.4.0
+  - libxcb=1.17.0
+  - libxml2=2.12.7
+  - libzlib=1.3.1
+  - llvm-openmp=19.1.0
+  - markdown-it-py=3.0.0
+  - markupsafe=2.1.5
+  - matplotlib-base=3.9.2
+  - mdurl=0.1.2
+  - munkres=1.1.4
+  - ncurses=6.5
+  - numpy=2.1.1
+  - openh264=2.4.1
+  - openjpeg=2.5.2
+  - openssl=3.3.2
+  - orjson=3.10.7
+  - packaging=24.1
+  - pandas=2.2.3
+  - pcre2=10.44
+  - pillow=10.4.0
+  - pip=24.2
+  - pixman=0.43.4
+  - pthread-stubs=0.4
+  - pugixml=1.14
+  - pycparser=2.22
+  - pydantic=2.0.3
+  - pydantic-core=2.3.0
+  - pydub=0.25.1
+  - pygments=2.18.0
+  - pyparsing=3.1.4
+  - pysocks=1.7.1
+  - python=3.11.10
+  - python-dateutil=2.9.0
+  - python-multipart=0.0.12
+  - python-tzdata=2024.2
+  - python_abi=3.11
+  - pytz=2024.1
+  - pyyaml=6.0.2
+  - qhull=2020.2
+  - readline=8.2
+  - requests=2.32.3
+  - rich=13.9.1
+  - ruff=0.6.8
+  - semantic_version=2.10.0
+  - setuptools=75.1.0
+  - shellingham=1.5.4
+  - six=1.16.0
+  - snappy=1.2.1
+  - sniffio=1.3.1
+  - starlette=0.38.6
+  - svt-av1=2.2.1
+  - tbb=2021.13.0
+  - tk=8.6.13
+  - tomlkit=0.12.0
+  - tqdm=4.66.5
+  - typer=0.12.5
+  - typer-slim=0.12.5
+  - typer-slim-standard=0.12.5
+  - typing-extensions=4.12.2
+  - typing_extensions=4.12.2
+  - tzdata=2024a
+  - urllib3=2.2.3
+  - uvicorn=0.31.0
+  - websockets=12.0
+  - wheel=0.44.0
+  - x264=1!164.3095
+  - x265=3.5
+  - xorg-libxau=1.0.11
+  - xorg-libxdmcp=1.1.5
+  - xz=5.2.6
+  - yaml=0.2.5
+  - zipp=3.20.2
+  - zlib=1.3.1
+  - zstandard=0.23.0
+  - zstd=1.5.6
+prefix: /Users/hernanlira/miniforge3/envs/translator_env

generate_eban.py

@@ -0,0 +1,12 @@
+from transformers import pipeline
+
+def translate_es_to_mapudungun(text_es):
+    pipe = pipeline("text2text-generation", model="ebanebssa/es-arn")
+    translated_text = pipe(text_es)[0]['generated_text']
+    return translated_text
+
+# Example usage (can be commented out or removed)
+if __name__ == "__main__":
+    text_es = input("Enter text in Spanish: ")
+    translated_text = translate_es_to_mapudungun(text_es)
+    print("Translated text (Mapudungun):", translated_text)