Update app.py

app.py

@@ -1,22 +1,13 @@
-# app.py
-
-from fastapi import FastAPI
-import requests
+from fastapi import FastAPI, HTTPException
+from libretranslate import LibreTranslate
 
 app = FastAPI()
+translator = LibreTranslate()
 
-LIBRETRANSLATE_URL = "http://libretranslate:5000/translate"
-
-@app.get("/")
-def root():
-    return {"message": "Hello World"}
-response = requests.post(LIBRETRANSLATE_URL, json={"q": "text", "source": "auto", "target": "hi"})
-print(response)
-@app.get("/translate")
-def translate_text(text: str):
+@app.get("/translate/")
+async def translate_text(text: str, source_lang: str = "auto", target_lang: str = "en"):
     try:
-        response = requests.post(LIBRETRANSLATE_URL, json={"q": text, "source": "auto", "target": "en"})
-        translation = response.json()["translatedText"]
+        translation = translator.translate(text, source_lang=source_lang, target_lang=target_lang)
         return {"translation": translation}
     except Exception as e:
-        return {"error": str(e)}
+        raise HTTPException(status_code=500, detail=str(e))