Commit
·
26f365f
1
Parent(s):
3757521
Add target lang option to txt translation
Browse files- app.py +4 -2
- backend/text_to_tgt.py +6 -2
app.py
CHANGED
|
@@ -1,6 +1,5 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
-
|
| 3 |
-
from backend.text_to_tgt import src_txt_to_eng_translator
|
| 4 |
from backend.audio_to_tgt import src_audio_to_eng_translator
|
| 5 |
from backend.image_to_tgt import src_image_to_eng_translator, language_dict
|
| 6 |
# from backend.video_to_tgt import src_video_to_eng_translator
|
|
@@ -8,10 +7,13 @@ from backend.image_to_tgt import src_image_to_eng_translator, language_dict
|
|
| 8 |
heading_txt = "Text-to-English"
|
| 9 |
description_txt = '''Enter text in any language, and get the translation in English.'''
|
| 10 |
|
|
|
|
|
|
|
| 11 |
txt_interface = gr.Interface(
|
| 12 |
fn=src_txt_to_eng_translator,
|
| 13 |
inputs=[
|
| 14 |
gr.Textbox(label="Text Input"),
|
|
|
|
| 15 |
],
|
| 16 |
outputs=gr.Textbox(label="Translation"),
|
| 17 |
title=heading_txt,
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
+
from backend.text_to_tgt import src_txt_to_eng_translator, formatted_languages
|
|
|
|
| 3 |
from backend.audio_to_tgt import src_audio_to_eng_translator
|
| 4 |
from backend.image_to_tgt import src_image_to_eng_translator, language_dict
|
| 5 |
# from backend.video_to_tgt import src_video_to_eng_translator
|
|
|
|
| 7 |
heading_txt = "Text-to-English"
|
| 8 |
description_txt = '''Enter text in any language, and get the translation in English.'''
|
| 9 |
|
| 10 |
+
language_list = formatted_languages.keys()
|
| 11 |
+
|
| 12 |
txt_interface = gr.Interface(
|
| 13 |
fn=src_txt_to_eng_translator,
|
| 14 |
inputs=[
|
| 15 |
gr.Textbox(label="Text Input"),
|
| 16 |
+
gr.Dropdown(choices=language_list, label="Select Target Language, Default: English", value="English", interactive=True)
|
| 17 |
],
|
| 18 |
outputs=gr.Textbox(label="Translation"),
|
| 19 |
title=heading_txt,
|
backend/text_to_tgt.py
CHANGED
|
@@ -1,5 +1,9 @@
|
|
| 1 |
from deep_translator import GoogleTranslator
|
| 2 |
|
| 3 |
-
|
| 4 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
return translated
|
|
|
|
| 1 |
from deep_translator import GoogleTranslator
|
| 2 |
|
| 3 |
+
available_languages = GoogleTranslator().get_supported_languages(as_dict=True)
|
| 4 |
+
formatted_languages = {key.title(): value for key, value in available_languages.items()}
|
| 5 |
+
|
| 6 |
+
def src_txt_to_eng_translator(input_text, target_lang = 'English'):
|
| 7 |
+
target_lang_code = formatted_languages.get(target_lang, 'en')
|
| 8 |
+
translated = GoogleTranslator(source='auto', target=target_lang_code).translate(input_text)
|
| 9 |
return translated
|