Update app.py

app.py

@@ -62,13 +62,14 @@ def listmodels():
 # 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'):
+    message = f'Translated from {sl} to {tl} with {model_name}'
     if 'BlackKakapo' in model_name:
         translation = src.Translate.paraphraseTranslateMethod(input_text, model_name)
     elif 'Helsinki-NLP' in model_name:
-        translation = src.Translate.Translators(model_name, sl, tl, input_text).HelsinkiNLP_mulroa()
+        translation, message = src.Translate.Translators(model_name, sl, tl, input_text).HelsinkiNLP_mulroa()
     else:
         translation: str = src.Translate.gemma_direct(input_text, model_name)
-    return {"input": input_text, "result": translation, "model": model_name}
+    return {"input_text": input_text, "translation": translation, "model_name": model_name, "message": message}
 
 # Keep track of installed (src, tgt) pairs
 installed_pairs = set()