Update main.py

main.py

@@ -34,28 +34,27 @@ async def translate(text,language):
 
 
 
-#text_to_speech bock
+#text_to_speech bock===========================================================================
 os.environ["COQUI_TOS_AGREED"] = "1"
-#from TTS.api import TTS
+from TTS.api import TTS
 
-#model_names = TTS().list_models()
-#m = model_names[0]
-#print(model_names)
-#global xtts
-#xtts = TTS(m, gpu=True)
+model_names = TTS().list_models()
+m = model_names[0]
+global xtts
+xtts = TTS(m, gpu=True)
 #tts.to("cpu") # no GPU or Amd
-#xtts.to("cuda")
+xtts.to("cuda")
 
 @app.get("/text-to-speech/")
 def text_to_speech(text,language):
-    #global xtts
+    global xtts
     audio_file = 'text_to_speech.wav'
-    #if language=="vietnamese":
-    from gtts import gTTS
-    tts = gTTS(text)
-    tts.save(audio_file)
-    #else:
-        #xtts.tts_to_file(text,speaker_wav="input.wav",language=keys[vals.index(language)],file_path=audio_file)
+    if language=="vietnamese":
+        from gtts import gTTS
+        tts = gTTS(text)
+        tts.save(audio_file)
+    else:
+        xtts.tts_to_file(text,speaker_wav="input.wav",language=keys[vals.index(language)],file_path=audio_file)
     return FileResponse(audio_file, media_type='audio/mpeg')