Update main.py

main.py

@@ -1,5 +1,6 @@
 from fastapi import FastAPI, File, UploadFile
 from fastapi.responses import StreamingResponse
+from fastapi.responses import FileResponse, HTMLResponse
 import os
 import io
 
@@ -33,4 +34,21 @@ async def speech_to_text(file: UploadFile = File(...)):
         return respond
 
 
+os.environ["COQUI_TOS_AGREED"] = "1"
+from TTS.api import TTS
+
+model_names = TTS().list_models()
+m = model_names[0]
+print(model_names)
+tts = TTS(m, gpu=True)
+#tts.to("cpu") # no GPU or Amd
+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)
+    return FileResponse(audio_file, media_type='audio/mpeg')