Update app.py

app.py

@@ -3,7 +3,6 @@ import src.Paraphrase as Paraphrase
 import src.Translate as Translate
 from typing import Optional
 from fastapi_mcp import FastApiMCP
-from typing import Union
 
 app = FastAPI()
 # app = FastAPI(docs_url="/docs")
@@ -56,7 +55,6 @@ 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: str, 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: Optional[str] = 'bergamot'):
     try:
         import bergamot
@@ -66,9 +64,9 @@ def bergamot(input_text: list[str] = Query(description="Input list of strings"),
         service = bergamot.Service(config)
         models = service.modelFromConfigPath("./deen/bergamot.config.yml")
         options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
-        response = service.translate(models, bergamot.VectorString(input_text), options)
-        print(type(input_text), len(response), [r.target.text for r in response], next(iter(response)).target.text)
-        response = [r.target.text for r in response] if len(response) > 1 else next(iter(response)).target.text
+        rawresponse = service.translate(models, bergamot.VectorString(input_text), options)
+        response = [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), 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
     except Exception as error:
@@ -83,7 +81,7 @@ mcp = FastApiMCP(
     describe_all_responses=True,
     describe_full_response_schema=True,
     include_operations=["get_translate", "get_paraphrase"],
-    include_tags=["paraphrase", "translate"]
+    include_tags=["paraphrase", "translate", "bergamot"]
 )
 # Mount the MCP server directly to the FASTAPI app
 mcp.mount()