Spaces:
Running
Running
| # from responses import start | |
| import gradio as gr | |
| from language_directions import * | |
| from transformers import pipeline | |
| from examples import example_sentences | |
| source_lang_dict = get_all_source_languages() | |
| target_lang_dict = {} | |
| source_languages = source_lang_dict.keys() | |
| def get_auto_detect_source_dropdown(input_text): | |
| source, _ = auto_detect_language_code(input_text) | |
| language_name = get_name_from_iso_code(source) | |
| source_dropdown_text = "Detected - " + language_name | |
| update_source_languages_dict(source_lang_dict, source_dropdown_text) | |
| source_language_dropdown = gr.Dropdown(choices=source_languages, | |
| value=source_dropdown_text, | |
| label="Source Language") | |
| return source_language_dropdown, language_name | |
| def get_target_dropdown(source_language_name, input_text): | |
| global target_lang_dict | |
| target_lang_dict, source_language = get_target_languages(source_lang_dict[source_language_name], input_text) | |
| target_languages = list(target_lang_dict.keys()) | |
| default_target_value = None | |
| if "English" in target_languages or "english" in target_languages: | |
| default_target_value = "English" | |
| else: | |
| default_target_value = target_languages[0] | |
| target_dropdown = gr.Dropdown(choices=target_languages, | |
| value=default_target_value, | |
| label="Target Language") | |
| return target_dropdown | |
| def get_dropdown_value(dropdown): | |
| if isinstance(dropdown, gr.Dropdown): | |
| dropdown_value = dropdown.constructor_args.get('value') | |
| elif isinstance(dropdown, str): | |
| dropdown_value = dropdown | |
| return dropdown_value | |
| def get_dropdowns(source_dropdown, input_text): | |
| source_language_name = get_dropdown_value(source_dropdown) | |
| if input_text and source_language_name == "Auto Detect" or source_language_name.startswith("Detected"): | |
| source_dropdown, source_language_name = get_auto_detect_source_dropdown(input_text) | |
| target_dropdown = get_target_dropdown(source_language_name=source_language_name, | |
| input_text=input_text) | |
| return source_dropdown, target_dropdown | |
| def input_changed(source_language_dropdown, input_text=""): | |
| return get_dropdowns(source_dropdown=source_language_dropdown, | |
| input_text=input_text) | |
| def translate(input_text, source, target): | |
| source_readable = source | |
| if source == "Auto Detect" or source.startswith("Detected"): | |
| source, _ = auto_detect_language_code(input_text) | |
| if source in source_lang_dict.keys(): | |
| source = source_lang_dict[source] | |
| target_lang_dict, _ = get_target_languages(source) | |
| try: | |
| target = target_lang_dict[target] | |
| model = f"Helsinki-NLP/opus-mt-{source}-{target}" | |
| pipe = pipeline("translation", model=model) | |
| translation = pipe(input_text) | |
| return translation[0]['translation_text'], "" | |
| except KeyError: | |
| return "", f"Error: Translation direction {source_readable} to {target} is not supported by Helsinki Translation Models" | |
| with gr.Blocks() as demo: | |
| gr.HTML("""<html> | |
| <head> | |
| <style> | |
| h1 { | |
| text-align: center; | |
| } | |
| </style> | |
| </head> | |
| <body> | |
| <h1>Open Translate</h1> | |
| </body> | |
| </html>""") | |
| with gr.Row(): | |
| with gr.Column(): | |
| source_language_dropdown = gr.Dropdown(choices=source_languages, | |
| value="Auto Detect", | |
| label="Source Language") | |
| input_textbox = gr.Textbox(lines=5, placeholder="Enter text to translate", label="Input Text") | |
| with gr.Column(): | |
| target_language_dropdown = gr.Dropdown(choices=["English", "French", "Spanish"], | |
| value="English", | |
| label="Target Language") | |
| translated_textbox = gr.Textbox(lines=5, placeholder="", label="Translated Text") | |
| info_label = gr.HTML("") | |
| btn = gr.Button("Translate") | |
| source_language_dropdown.change(input_changed, inputs=[source_language_dropdown, input_textbox], outputs=[source_language_dropdown, target_language_dropdown]) | |
| input_textbox.change(input_changed, inputs=[source_language_dropdown, input_textbox], outputs=[source_language_dropdown, target_language_dropdown]) | |
| btn.click(translate, inputs=[input_textbox, | |
| source_language_dropdown, | |
| target_language_dropdown], | |
| outputs=[translated_textbox, info_label]) | |
| gr.Examples(example_sentences, inputs=[input_textbox]) | |
| if __name__ == "__main__": | |
| demo.launch() |