Update app.py

app.py

@@ -3,6 +3,7 @@ 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")
@@ -54,7 +55,8 @@ def translate(text: str, model: Optional[str] = MODELS['enro']):
     return {"input": text, "result": translation, "model": model}
 
 @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: str, sl: str = 'de', tl: str = 'en', model: Optional[str] = 'bergamot'):
+def bergamot(input_text: Union[str, list[str]], sl: str = 'de', tl: str = 'en', model: Optional[str] = 'bergamot'):
     try:
         import bergamot
         config = bergamot.ServiceConfig(numWorkers=4)