DebasishDhal99 commited on
Commit
26f365f
·
1 Parent(s): 3757521

Add target lang option to txt translation

Browse files
Files changed (2) hide show
  1. app.py +4 -2
  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
- def src_txt_to_eng_translator(input_text):
4
- translated = GoogleTranslator(source='auto', target='en').translate(input_text)
 
 
 
 
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