Chandrabhan01 commited on
Commit
cbe2be4
·
1 Parent(s): 4bc28b7

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +109 -0
app.py ADDED
@@ -0,0 +1,109 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from gtts import gTTS
3
+ import gradio as gr
4
+ from translate import Translator
5
+
6
+ # os.environ['TF_CPP_MIN_LOG_LEVEL']='3'
7
+
8
+ auth_token = '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' # For accessing HuggingFace Facebook Model.
9
+
10
+ class Languages:
11
+ """ Languages currently supported by the application. """
12
+ # Language code
13
+ # lang (string, optional) – The language (IETF language tag) to read the text in. Default is en.
14
+ lang = {'Afrikaans': 'af','Arabic':'ar','Bulgarian':'bg','Bengali':'bn','Bosnian':'bs',
15
+ 'Catalan':'ca','Czech':'cs','Danish':'da','German':'de','Greek':'el','English':'en',
16
+ 'Spanish':'es','Estonian':'et','Finnish':'fi','French':'fr','Gujarati':'gu','Hindi':'hi',
17
+ 'Croatian':'hr','Hungarian':'hu','Indonesian':'id','Icelandic':'is','Italian':'it',
18
+ 'Hebrew':'iw','Japanese':'ja','Javanese':'jw','Khmer':'km','Kannada':'kn','Korean':'ko',
19
+ 'Latin':'la','Latvian':'lv','Malayalam':'ml','Marathi':'mr','Malay':'ms',
20
+ 'Myanmar (Burmese)':'my','Nepali':'ne', 'Dutch':'nl','Norwegian':'no',
21
+ 'Polish':'pl','Portuguese':'pt','Romanian':'ro','Russian':'ru','Sinhala':'si',
22
+ 'Slovak':'sk', 'Albanian':'sq','Serbian':'sr','Sundanese':'su','Swedish':'sv',
23
+ 'Swahili':'sw','Tamil':'ta','Telugu':'te','Thai':'th','Filipino':'tl','Turkish':'tr',
24
+ 'Ukrainian':'uk','Urdu':'ur','Vietnamese':'vi','Chinese (Simplified)':'zh-CN',
25
+ 'Chinese (Mandarin/Taiwan)':'zh-TW',
26
+ 'Chinese (Mandarin)':'zh'}
27
+
28
+ class TLD:
29
+ """ Depending on the top-level domain, gTTS can speak in different accents. """
30
+
31
+ tld = {'English(Australia)':'com.au', 'English (United Kingdom)':'co.uk',
32
+ 'English (United States)':'us', 'English (Canada)':'ca','English (India)':'co.in',
33
+ 'English (Ireland)':'ie','English (South Africa)':'co.za','French (Canada)':'ca',
34
+ 'French (France)':'fr','Portuguese (Brazil)':'com.br','Portuguese (Portugal)':'pt',
35
+ 'Spanish (Mexico)':'com.mx','Spanish (Spain)':'es','Spanish (United States)':'us'}
36
+
37
+ class TTSLayer():
38
+ """ Layer on top of gTTS - providing text to speech for """
39
+
40
+ def __init__(self, text, tld, lang) -> None:
41
+ """ [Constructor takes in text, the top-level domain and the language in which the text is : ] """
42
+ self.text = text
43
+ self.tld = tld
44
+ self.lang = lang
45
+
46
+ def tts(self):
47
+ """ [Converts the text to speech.] """
48
+ tts = gTTS(text=self.text,tld= TLD.tld[self.tld], lang=Languages.lang[self.lang])
49
+ tts.save('tts.mp3')
50
+ with open('tts.mp3') as fp:
51
+ return fp.name
52
+
53
+ langs = Languages()
54
+ top_level_domain = TLD()
55
+
56
+ """ [Utiility Functions] """
57
+
58
+ def T2TConversion(text, dest):
59
+ """ [(Utility Function) : Converts sentence from english to another language ] """
60
+ translator = Translator(to_lang=langs.lang[dest])
61
+ return translator.translate(text)
62
+
63
+ def convert_text(Text,Language, Accent):
64
+ """ [(Utility Function) : Performs Text-To-Speech provided language and accent.] """
65
+ tts = TTSLayer(Text,Accent, Language)
66
+ return tts.tts()
67
+
68
+
69
+ class GRadioInterface:
70
+ """ [Class for managing UI for the application.] """
71
+ def __init__(self, function) -> None:
72
+ """ [Interface for packaging GRadio Application] """
73
+
74
+ # Necessary for interface
75
+ self._function = function
76
+ self._inputs = [
77
+ gr.TextArea(label = 'The Text to be Converted to Audio'),
78
+ gr.Dropdown([key for key,_ in langs.lang.items()], label='Languages Available',),
79
+ gr.Dropdown([key for key,_ in top_level_domain.tld.items()])]
80
+
81
+ self.outputs = gr.Audio()
82
+
83
+ # Necessary for descriptive content
84
+ self.title = 'A Text-To-Speech Converter for Low Resource Languages'
85
+ self.description = 'Support over 50 languages !'
86
+ self.article = 'How does it work ? Just write a sentence (in target language) in the space provided and select the target language and accent and press submit. That is it. Wait and Enjoy.'
87
+
88
+ def start(self):
89
+ """ [Launching the interface in a tabbed manner.] """
90
+
91
+ it_1 = gr.Interface(fn=self._function, inputs=self._inputs,outputs=self.outputs,
92
+ title = self.title,
93
+ description= self.description,
94
+ article= self.article)
95
+
96
+ it_2 = gr.Interface(fn=T2TConversion,
97
+ inputs = [
98
+ gr.Text(label='Write a sentence in English'),
99
+ gr.Dropdown([key for key,_ in langs.lang.items()])],
100
+ outputs= gr.Text(label='The Converted Text'),
101
+ title = 'Translation from English',
102
+ description='Write a sentence in english and convert to other languages for speech synthesis',
103
+ article='What if you do not have a sentence in a particular language? Just write the sentence in english and let us do the magic.')
104
+
105
+ demo = gr.TabbedInterface([it_1,it_2],['Speech Synthesis', 'Sentence Translation'])
106
+ demo.launch(share=True)
107
+
108
+ demo_app = GRadioInterface(function=convert_text)
109
+ demo_app.start()