Update app.py

app.py

@@ -56,16 +56,15 @@ def translate(text: str, model: Optional[str] = MODELS['enro']):
 
 @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: Union[str, List[str]] = Query(default=None, description="One or more input strings"), sl: str = 'de', tl: str = 'en', model: Optional[str] = 'bergamot'):
+def bergamot(input_text: Union[str, list[str]] = Query(default=None, description="One or more input strings"), sl: str = 'de', tl: str = 'en', model: Optional[str] = 'bergamot'):
     try:
         import bergamot
-        # if isinstance(input_text, str):
-        #     input_text = [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")
         options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
-        response = service.translate(models, bergamot.VectorString([input_text]), options)
+        response = service.translate(models, bergamot.VectorString(input_text), options)
         print(type(response), 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
         # response = [r.target.text for r in model_response][0] if isinstance(response, bergamot._bergamot.VectorResponse) else next(iter(response)).target.text