Update apis/chat_api.py

apis/chat_api.py

@@ -3,6 +3,7 @@ import uvicorn
 import sys
 import json
 
+
 from fastapi import FastAPI
 from fastapi.encoders import jsonable_encoder
 from fastapi.responses import JSONResponse
@@ -11,6 +12,7 @@ from sse_starlette.sse import EventSourceResponse
 from utils.logger import logger
 from networks.message_streamer import MessageStreamer
 from messagers.message_composer import MessageComposer
+from googletrans import Translator
 
 
 class ChatAPIApp:
@@ -43,11 +45,20 @@ class ChatAPIApp:
         )
 
     def chat_completions(self, item: ChatCompletionsPostItem):
+        translator = Translator()
+        f = open('apis/lang_name.json', "r")
+        available_langs = json.loads(f.read())
+        from_lang = "en"
+        for lang_item in available_langs:
+          if item.to_language == lang_item['code']:
+              from_lang = item.to_language
+              break
+        
         item_response = {
             "from_language": item.from_language,
             "to_language": item.to_language,
             "text": item.text,
-            "translate": ""
+            "translate": translator.translate(item.text, dest=from_lang)
         }
         json_compatible_item_data = jsonable_encoder(item_response)
         return JSONResponse(content=json_compatible_item_data)