Update app.py

app.py

@@ -1,7 +1,7 @@
 from fastapi import FastAPI, Request, Query
 import src.Paraphrase as Paraphrase
-import src.Translation as Translation
-import src.Translate as Cltranslate
+import src.Translate as Translate
+import src.Translators as Translators
 from typing import Optional
 from fastapi_mcp import FastApiMCP
 from huggingface_hub import hf_hub_download, list_repo_files
@@ -60,16 +60,16 @@ def paraphrase(text: str, model: str  = MODELS['paraphrase']):
 def listmodels():
     return {"MODELS": MODELS, "EMBEDDING_MODELS": EMBEDDING_MODELS}
 
-# text: str, model: Optional[str] = MODELS['benro']
+# model: Optional[str] = MODELS['benro']
 @app.get("/translate", operation_id="get_translate", description="Translate text", tags=["translate"], summary="Translate text")
 def translate(input_text: str, model_name: str = MODELS['mttcbig'], sl: str = 'en', tl: str = 'ro'):
     if 'BlackKakapo' in model_name:
-        translation = Translation.paraphraseTranslateMethod(text, model_name)
+        translation = Translate.paraphraseTranslateMethod(text, model_name)
     elif 'Helsinki-NLP' in model_name:
-        translation = Cltranslate.HelsinkiNLP_mulroa(model_name, sl, tl, input_text)
+        translation = Translators.HelsinkiNLP_mulroa(model_name, sl, tl, input_text)
     else:
-        translation: str = Translate.gemma_direct(text, model)
-    return {"input": text, "result": translation, "model": model}
+        translation: str = Translate.gemma_direct(text, model_name)
+    return {"input": text, "result": translation, "model": model_name}
 
 # Keep track of installed (src, tgt) pairs
 installed_pairs = set()