Update apis/chat_api.py

apis/chat_api.py

@@ -43,6 +43,7 @@ class ChatAPIApp:
             default="Hello",
             description="(str) `Text for translate`",
         )
+   
 
     def chat_completions(self, item: ChatCompletionsPostItem):
         translator = Translator()
@@ -51,22 +52,40 @@ class ChatAPIApp:
         from_lang = 'en'
         to_lang = 'en'
         for lang_item in available_langs:
-          if item.from_language == lang_item['code']:
-              from_lang = item.from_language
           if item.to_language == lang_item['code']:
               to_lang = item.to_language
+              break
               
           
-        translated = translator.translate(item.input_text, dest=to_lang).text
+        translated = translator.translate(item.input_text, dest=to_lang)
         item_response = {
-            "from_language": item.from_language,
-            "to_language": item.to_language,
+            "from_language": translated.src,
+            "to_language": translated.dest,
             "text": item.input_text,
-            "translate": translated
+            "translate": translated.text
         }
         json_compatible_item_data = jsonable_encoder(item_response)
         return JSONResponse(content=json_compatible_item_data)
 
+        
+    class DetectLanguagePostItem(BaseModel)
+        input_text: str = Field(
+            default="Hello",
+            description="(str) `Text for translate`",
+        )
+
+    def detect_language(self, item: DetectLanguagePostItem)
+        translator = Translator()
+        detected = translator.detect(item.input_text)
+
+        item_response = {
+            "lang": detected.lang,
+            "confidence": detected.confidence,
+        }
+        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(
@@ -79,6 +98,11 @@ class ChatAPIApp:
                 summary="translate text",
             )(self.chat_completions)
 
+            self.app.post(
+                prefix + "/detect",
+                summary="detect language",
+            )(self.detect_language)
+
 
 class ArgParser(argparse.ArgumentParser):
     def __init__(self, *args, **kwargs):