feat(translate): hard coded /toEn endpoint

project-ait · Nov 2, 2023 · da8e3ae · da8e3ae
1 parent 7998fcf
commit da8e3ae
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/routers/translate.py b/routers/translate.py
@@ -2,13 +2,23 @@
 
 import deepl
 from fastapi import APIRouter
+from pydantic import BaseModel
 
 router = APIRouter()
 
 API_KEY = os.environ.get("DEEPL_KEY")
 translator = deepl.Translator(API_KEY)
 
 
+class TranslatedModel(BaseModel):
+    text: str
+
+
 @router.post("/toKr")
-async def translate(text: str):
-    return translator.translate_text(text, source_lang="EN", target_lang="KO")
+async def translate(data: TranslatedModel):
+    return translator.translate_text(data.text, source_lang="EN", target_lang="KO")
+
+
+@router.post("/toEn")
+async def translate(data: TranslatedModel):
+    return translator.translate_text(data.text, source_lang="KO", target_lang="EN")