Update app.py

app.py

@@ -55,23 +55,41 @@ def translate(text: str, model: Optional[str] = MODELS['enro']):
 
 # https://tiberiucristianleon-fastapimt.hf.space/bergamot?input_text=das%20ist%20keine%20gute%20Frau&input_text=das%20ist%20eine%20gute%20Nachricht&sl=de&tl=en&model=bergamot
 @app.get("/bergamot", operation_id="get_bergamot", description="Translate text with Bergamot", tags=["bergamot"], summary="Translate text with Bergamot")
-def bergamot(input_text: list[str] = Query(description="Input list of strings"), sl: str = 'de', tl: str = 'en', model: Optional[str] = 'bergamot'):
+def bergamot(input_text: list[str] = Query(description="Input list of strings"), sl: str = 'de', tl: str = 'en', model_name: Optional[str] = 'deen'):
+    """
+    Translates the input text from the source language to the target language using a specified model.
+    Parameters:
+        input_text (str): The source text to be translated
+        sl (str): The source language of the input text
+        tl (str): The target language into which the input text is translated
+        model_name (str): The selected translation model name
+    Returns:
+        dict:
+            input_text(str): The input text in the source language
+            translated_text(str): The input text translated to the selected target language
+            message_text(str):  A descriptive message summarizing the translation process. Example: "Translated from English to German with ende."
+    
+    Example:
+        >>> bergamot("Hello world", "en", "de", "ende")
+        {"input_text": "Hello world", "translated_text": "Hallo Welt", "message_text": "Translated from English to German with ende."}
+    """
     try:
         import bergamot
         # print(type(input_text))
         # input_text = [input_text] if isinstance(input_text, str) else input_text           
         config = bergamot.ServiceConfig(numWorkers=4)
         service = bergamot.Service(config)
-        models = service.modelFromConfigPath("./deen/bergamot.config.yml")
+        model = service.modelFromConfigPath(f"./{model_name}/bergamot.config.yml")
         options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
-        rawresponse = service.translate(models, bergamot.VectorString(input_text), options)
+        rawresponse = service.translate(model, bergamot.VectorString(input_text), options)
         response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text
         print(type(input_text), len(input_text), len(rawresponse), type(response), response)
         # response = [r.target.text for r in model_response][0] if isinstance(response, bergamot._bergamot.VectorResponse) else next(iter(response)).target.text
         # response is of type bergamot._bergamot.VectorResponse, an iterable of bergamot._bergamot.Response
+        message_text = f"Translated from {sl} to {tl} with {model_name}."
     except Exception as error:
         response = error
-    return {"input": input_text, "result": response, "model": model}
+    return {"input": input_text, "translated_text": response, "message_text": message_text}
 
 # Create an MCP server based on this app
 mcp = FastApiMCP(