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

Add source language detection

Browse files
Files changed (2) hide show
  1. app.py +4 -1
  2. backend/text_to_tgt.py +8 -2
app.py CHANGED
@@ -15,7 +15,10 @@ txt_interface = gr.Interface(
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,
20
  description=description_txt,
21
  examples=[
 
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=[
19
+ gr.Textbox(label="Translation"),
20
+ gr.Textbox(label="Source Language")
21
+ ],
22
  title=heading_txt,
23
  description=description_txt,
24
  examples=[
backend/text_to_tgt.py CHANGED
@@ -1,9 +1,15 @@
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
 
1
+ from deep_translator import GoogleTranslator, detection
2
+ import os
3
 
4
  available_languages = GoogleTranslator().get_supported_languages(as_dict=True)
5
  formatted_languages = {key.title(): value for key, value in available_languages.items()}
6
+ formatted_codes = {value: key.title() for key, value in available_languages.items()}
7
+
8
+ lang_detect_key = os.getenv("detect_language_api_key")
9
 
10
  def src_txt_to_eng_translator(input_text, target_lang = 'English'):
11
  target_lang_code = formatted_languages.get(target_lang, 'en')
12
+ src_lang_code = detection.single_detect(input_text, api_key = lang_detect_key)
13
+ src_lang = formatted_codes.get(src_lang_code, 'Source language not detected')
14
  translated = GoogleTranslator(source='auto', target=target_lang_code).translate(input_text)
15
+ return translated, src_lang