TNK21 commited on
Commit
6fe4dfb
·
1 Parent(s): 87c354f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -18
app.py CHANGED
@@ -1,31 +1,37 @@
1
  import gradio as gr
2
  from translate import Translator
3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  def translate_text(text, target_language):
5
- translator = Translator(to_lang=target_language)
 
 
6
  translation = translator.translate(text)
7
  return translation
8
 
9
- target_languages = [
10
- "ta", "hi", "te", "en", "de", "fr", "es", "it", "ja", "ko"
11
- ]
12
-
13
- # Create a separate list for language names
14
- target_language_names = [
15
- "Tamil", "Hindi", "Telugu", "English", "German",
16
- "French", "Spanish", "Italian", "Japanese", "Korean"
17
- ]
18
-
19
- # Combine language codes and names for the dropdown labels
20
- target_languages_with_names = [f"{code} ({name})" for code, name in zip(target_languages, target_language_names)]
21
-
22
  interface = gr.Interface(
23
  fn=translate_text,
24
  inputs=[
25
- gr.components.Textbox(label="Text to translate"),
26
- gr.components.Dropdown(target_languages_with_names, label="Target Language")
27
  ],
28
- outputs=gr.components.Textbox(label="Translation")
29
  )
30
 
31
- interface.launch()
 
1
  import gradio as gr
2
  from translate import Translator
3
 
4
+ # Define a mapping of language codes to language names
5
+ language_map = {
6
+ "ta": "Tamil",
7
+ "hi": "Hindi",
8
+ "te": "Telugu",
9
+ "en": "English",
10
+ "de": "German",
11
+ "fr": "French",
12
+ "es": "Spanish",
13
+ "it": "Italian",
14
+ "ja": "Japanese",
15
+ "ko": "Korean"
16
+ }
17
+
18
+ # Create a list of target languages with their names
19
+ target_languages = list(language_map.values())
20
+
21
  def translate_text(text, target_language):
22
+ # Reverse lookup to get the language code from the language name
23
+ target_code = [code for code, name in language_map.items() if name == target_language][0]
24
+ translator = Translator(to_lang=target_code)
25
  translation = translator.translate(text)
26
  return translation
27
 
 
 
 
 
 
 
 
 
 
 
 
 
 
28
  interface = gr.Interface(
29
  fn=translate_text,
30
  inputs=[
31
+ gr.inputs.Textbox(label="Text to translate"),
32
+ gr.inputs.Dropdown(target_languages, label="Target Language")
33
  ],
34
+ outputs=gr.outputs.Textbox(label="Translation")
35
  )
36
 
37
+ interface.launch()