Update tools/translate_text.py

tools/translate_text.py

@@ -1,13 +1,26 @@
+from typing import TypedDict
 import requests
 
-def translate_text(text: str, target_lang: str) -> str:
-url = "https://libretranslate.com/translate"
-payload = {
-"q": text,
-"source": "auto",
-"target": target_lang,
-"format": "text"
-}
-resp = requests.post(url, data=payload)
-resp.raise_for_status()
-return resp.json()["translatedText"]
+class InputSchema(TypedDict):
+    text: str
+    source_lang: str
+    target_lang: str
+
+class OutputSchema(TypedDict):
+    translated_text: str
+
+def translate_text(inputs: InputSchema) -> OutputSchema:
+    url = "https://libretranslate.de/translate"
+    payload = {
+        "q": inputs["text"],
+        "source": inputs["source_lang"],
+        "target": inputs["target_lang"],
+        "format": "text"
+    }
+    headers = {"Content-Type": "application/json"}
+
+    response = requests.post(url, json=payload, headers=headers)
+    response.raise_for_status()
+    translated = response.json()["translatedText"]
+
+    return {"translated_text": translated}