Update app.py

app.py

@@ -54,21 +54,15 @@ 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(text: str, model: Optional[str] = 'bergamot'):
+def bergamot(text: str, sl: str = 'de', tl: str = 'en', model: Optional[str] = 'bergamot'):
     try:
         import bergamot
         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([
-            "Das ist ein deutscher Text", 
-            "Das ist der zweite Satz.",
-            text,
-        ]), options)
-        responselist = [r.target.text for r in response]
+        options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
+        response = service.translate(models, bergamot.VectorString(text), options)
+        responselist = [r.target.text for r in response] if isinstance(response, list) else responselist
     except Exception as error:
         response = error
     return {"input": text, "result": responselist, "model": str(model)}