Update apis/chat_api.py

apis/chat_api.py

@@ -2,6 +2,8 @@ import argparse
 import uvicorn
 import sys
 import json
+import string
+import random
 
 
 from fastapi import FastAPI
@@ -13,6 +15,7 @@ from utils.logger import logger
 from networks.message_streamer import MessageStreamer
 from messagers.message_composer import MessageComposer
 from googletrans import Translator
+from gtts import gTTS
 
 
 class ChatAPIApp:
@@ -71,7 +74,7 @@ class ChatAPIApp:
     class DetectLanguagePostItem(BaseModel):
         input_text: str = Field(
             default="Hello",
-            description="(str) `Text for translate`",
+            description="(str) `Text for detection`",
         )
 
     def detect_language(self, item: DetectLanguagePostItem):
@@ -84,8 +87,31 @@ class ChatAPIApp:
         }
         json_compatible_item_data = jsonable_encoder(item_response)
         return JSONResponse(content=json_compatible_item_data)
+        
+    class TTSPostItem(BaseModel):
+        input_text: str = Field(
+            default="Hello",
+            description="(str) `Text for TTS`",
+        )
+        from_language: str = Field(
+            default="en",
+            description="(str) `TTS language`",
+        )
+        
+    def text_to_speech(self, item: TTSPostItem):
+        audioobj = gTTS(text = item.input_text, 
+                    lang = item.from_language, 
+                    slow = False)
+        fileName = ''.join(random.SystemRandom().choice(string.ascii_uppercase + string.digits) for _ in range(10));
+        fileName = fileName + ".mp3";
+        audioobj.save(fileName)
 
-
+        item_response = {
+            "src": fileName
+        }
+        json_compatible_item_data = jsonable_encoder(item_response)
+        return JSONResponse(content=json_compatible_item_data)
+        
     def setup_routes(self):
         for prefix in ["", "/v1"]:
             self.app.get(
@@ -103,6 +129,11 @@ class ChatAPIApp:
                 summary="detect language",
             )(self.detect_language)
 
+            self.app.post(
+                prefix + "/tts",
+                summary="text to speech",
+            )(self.text_to_speech)
+
 
 class ArgParser(argparse.ArgumentParser):
     def __init__(self, *args, **kwargs):