File size: 5,472 Bytes
f17d6c8 347aa3e f17d6c8 87d7d29 da86445 f17d6c8 c04ffc4 9d30c4a 224ade3 bd96a6e f17d6c8 7aa9e56 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 |
import gradio as gr
from language_directions import get_all_source_languages,update_source_languages_dict,get_target_languages,auto_detect_language_code
from transformers import pipeline
from adibideak import adibide_esaldiak
from iso639_wrapper import get_name_from_iso_code
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="Input 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="Output 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 == "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 == "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 from {source_readable} to {target} is not supported by Translation Models"
with gr.Blocks() as demo:
gr.HTML("""<html>
<head>
<style>
h2 {
text-align: center;
}
</style>
</head>
<body>
<h2>Basque (EU) Euskera Multilingual Translator (Basque as the Starting Point)</h2>
<h2>Multilingual-to-Basque (EU) Euskera Translator (Basque as the Destination)</h2>
<div style="margin: 0.75em 0;"><a href="https://www.buymeacoffee.com/Artgen" target="_blank"><img src="https://cdn.buymeacoffee.com/buttons/default-orange.png" alt="Buy Me A Coffee" height="41" width="174"></a></div>
<p>This is an AI translator using language models - Hau hizkuntza-ereduak erabiltzen dituen IA itzultzaile bat da. Oraingoz lau hizkuntzetara itzultzen du.</p>
</body>
</html>""")
with gr.Row():
with gr.Column():
source_language_dropdown = gr.Dropdown(choices=source_languages,
value="Detect",
label="Input Language - Input Hizkuntza")
input_textbox = gr.Textbox(lines=5, placeholder="Enter text to translate", label="Input Text")
with gr.Column():
target_language_dropdown = gr.Dropdown(choices=["Basque", "English", "French", "Spanish"],
value="Basque",
label="Output Language - Output Hizkuntza")
translated_textbox = gr.Textbox(lines=5, placeholder="", label="Translated Text")
info_label = gr.HTML("")
btn = gr.Button("Translate - Itzuli")
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(adibide_esaldiak, inputs=[input_textbox])
if __name__ == "__main__":
demo.launch() |