KZTech commited on
Commit
09f6800
·
verified ·
1 Parent(s): 5213116

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +38 -6
app.py CHANGED
@@ -4,22 +4,54 @@ from gtts import gTTS
4
  import uuid
5
  import os
6
 
7
- # Language options and model map (simplified to just one pair)
8
  MODEL_MAP = {
9
  ("en", "fr"): "Helsinki-NLP/opus-mt-en-fr",
10
- ("fr", "en"): "Helsinki-NLP/opus-mt-fr-en"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  }
12
 
13
  TRANSLATORS = {}
14
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
15
  # UI
16
  st.title("🌍 Multilingual Translator with Text-to-Speech")
17
 
18
- src_lang_name = st.selectbox("Select source language", ["English", "French"], index=0)
19
- tgt_lang_name = st.selectbox("Select target language", ["English", "French"], index=1)
 
20
 
21
- src_lang = "en" if src_lang_name == "English" else "fr"
22
- tgt_lang = "en" if tgt_lang_name == "English" else "fr"
 
23
 
24
  text_input = st.text_area("Enter text to translate:")
25
 
 
4
  import uuid
5
  import os
6
 
7
+ # Language options and model map (now with 10 languages)
8
  MODEL_MAP = {
9
  ("en", "fr"): "Helsinki-NLP/opus-mt-en-fr",
10
+ ("fr", "en"): "Helsinki-NLP/opus-mt-fr-en",
11
+ ("en", "de"): "Helsinki-NLP/opus-mt-en-de",
12
+ ("de", "en"): "Helsinki-NLP/opus-mt-de-en",
13
+ ("en", "es"): "Helsinki-NLP/opus-mt-en-es",
14
+ ("es", "en"): "Helsinki-NLP/opus-mt-es-en",
15
+ ("en", "it"): "Helsinki-NLP/opus-mt-en-it",
16
+ ("it", "en"): "Helsinki-NLP/opus-mt-it-en",
17
+ ("en", "zh"): "Helsinki-NLP/opus-mt-en-zh", # Chinese
18
+ ("zh", "en"): "Helsinki-NLP/opus-mt-zh-en", # Chinese
19
+ ("en", "ur"): "Helsinki-NLP/opus-mt-en-ur", # Urdu
20
+ ("ur", "en"): "Helsinki-NLP/opus-mt-ur-en", # Urdu
21
+ ("en", "ar"): "Helsinki-NLP/opus-mt-en-ar", # Arabic
22
+ ("ar", "en"): "Helsinki-NLP/opus-mt-ar-en", # Arabic
23
+ ("en", "fa"): "Helsinki-NLP/opus-mt-en-fa", # Persian
24
+ ("fa", "en"): "Helsinki-NLP/opus-mt-fa-en", # Persian
25
+ ("en", "tr"): "Helsinki-NLP/opus-mt-en-tr", # Turkish
26
+ ("tr", "en"): "Helsinki-NLP/opus-mt-tr-en", # Turkish
27
  }
28
 
29
  TRANSLATORS = {}
30
 
31
+ # Language map for the dropdowns
32
+ lang_map = {
33
+ "English": "en",
34
+ "French": "fr",
35
+ "German": "de",
36
+ "Spanish": "es",
37
+ "Italian": "it",
38
+ "Chinese": "zh", # Simplified Chinese
39
+ "Urdu": "ur",
40
+ "Arabic": "ar",
41
+ "Persian": "fa",
42
+ "Turkish": "tr"
43
+ }
44
+
45
  # UI
46
  st.title("🌍 Multilingual Translator with Text-to-Speech")
47
 
48
+ lang_options = list(lang_map.keys())
49
+ src_lang_name = st.selectbox("Select source language", lang_options, index=0)
50
+ tgt_lang_name = st.selectbox("Select target language", lang_options, index=1)
51
 
52
+ # Map language names to codes
53
+ src_lang = lang_map[src_lang_name]
54
+ tgt_lang = lang_map[tgt_lang_name]
55
 
56
  text_input = st.text_area("Enter text to translate:")
57