import gradio as gr
from transformers import pipeline

en_fr_model = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fr")
fr_en_model = pipeline("translation", model="Helsinki-NLP/opus-mt-fr-en")

def translate(text, model):
    translations = model(text)
    output = translations[0]['translation_text']
    return output

def check(en):
    if en is None:
        return None,None
    fr = translate(en, en_fr_model)
    en2 = translate(fr,fr_en_model)
    return fr,en2

def clear():
    return None,None,None

source=gr.Textbox(label="EN - original", value="") # delaying rendering enables to control layout order

with gr.Blocks() as demo:
    gr.Markdown("# Back translation : insights in target translation quality")
    gr.Markdown("Translating back to source language is one mean to measure the quality of the target translation. Can be useful when you need to check the translation to a language you hardly understand...")
    gr.Markdown("The loop feedback may even guide you to rewrite original text enabling a better translation. Translations using Helsinki-NLP Opus MT models.")

    examples= gr.Examples(
        examples= [
            "Translating a language is a very complex problem and yet this Helsinki-NLP model does it both well and quickly, do you agree?",
            "Reverse translation, is the process of translating content from the target language back to its source language. The final outcome in English should match the original as closely as possible in a quality translation job."
        ],
        inputs= [ source ]
    )
    go_btn = gr.Button("Translate")
    source.render()
    en_back = gr.Textbox(label="EN - back")
    target =gr.Textbox(label="FR - target")
    go_btn.click( fn=check, inputs=[source], outputs= [target,en_back])
    gr.Button("Clear").click( fn=clear, inputs=[], outputs= [source, en_back,target] )

demo.launch(debug=True, enable_queue=True)

# gr.Interface(
#     fn=check,
#     inputs = gr.Textbox(label="EN - original"),
#     outputs = [ gr.Textbox(label="FR - target"), gr.Textbox(label="EN - back") ],
#     title="Back translation : insights in target translation quality",
#     description="Translating back to source language is one mean to measure the quality of the target translation. Can be useful when you need to check the translation to a language you hardly understand... The loop feedback may even guide you to rewrite original text enabling a better translation. Translations using Helsinki-NLP Opus MT models.",
#     examples= [
#         "Translating a language is a very complex problem and yet this Helsinki-NLP model does it both well and quickly, do you agree?",
#         "Reverse translation, is the process of translating content from the target language back to its source language. The final outcome in English should match the original as closely as possible in a quality translation job."
#     ],
#     allow_flagging="never"
# ).launch(debug=True, enable_queue=True)