Update main.py

main.py

@@ -45,10 +45,13 @@ tts = TTS(m, gpu=True)
 tts.to("cuda")
 
 @app.get("/text-to-speech/")
-def text_to_speech(text: str):
-    from gtts import gTTS
-    tts = gTTS(text)
-    audio_file = 'text_to_speech.mp3'
-    tts.save(audio_file)
+def text_to_speech(text,language):
+    if language=="vietnamese":
+        from gtts import gTTS
+        tts = gTTS(text)
+        audio_file = 'text_to_speech.mp3'
+        tts.save(audio_file)
+    else:
+        tts.tts_to_file(text,speaker_wav="input.wav", file_path='text_to_speech.mp3')
     return FileResponse(audio_file, media_type='audio/mpeg')