Update tts_engine.py

tts_engine.py

@@ -1,30 +1,48 @@
-from TTS.api import TTS
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import pipeline
+from gtts import gTTS
+import tempfile
+
 
-english_tts = None
-tamil_tts = None
 translator = None
-tokenizer = None
 
 
-def load_models():
+def load_translator():
 
-    global english_tts, tamil_tts, translator, tokenizer
+    global translator
 
-    if english_tts is None:
+    if translator is None:
 
-        english_tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2")
+        translator = pipeline(
+            "translation",
+            model="Helsinki-NLP/opus-mt-en-ta"
+        )
 
-    if tamil_tts is None:
+    return translator
 
-        tamil_tts = TTS("tts_models/ta/cv/vits")
 
-    if translator is None:
+def translate_to_tamil(text):
 
-        tokenizer = AutoTokenizer.from_pretrained(
-            "ai4bharat/indictrans2-en-indic"
-        )
+    translator = load_translator()
+
+    result = translator(text)
+
+    return result[0]["translation_text"]
+
+
+def generate_voice(text, language):
+
+    audio_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
+
+    if language == "Tamil":
+
+        text = translate_to_tamil(text)
+
+        tts = gTTS(text=text, lang="ta")
+
+    else:
+
+        tts = gTTS(text=text, lang="en")
+
+    tts.save(audio_file.name)
 
-        translator = AutoModelForSeq2SeqLM.from_pretrained(
-            "ai4bharat/indictrans2-en-indic"
-        )
+    return audio_file.name