Update app.py

app.py

@@ -5,6 +5,7 @@ from typing import Optional
 from fastapi_mcp import FastApiMCP
 from huggingface_hub import hf_hub_download, list_repo_files
 from sentence_transformers import SentenceTransformer
+import bergamot
 
 app = FastAPI()
 # app = FastAPI(docs_url="/docs")
@@ -83,7 +84,6 @@ def translate(input_text: str, model_name: str = MODELS['mttcbig'], sl: str = 'e
     return {"input_text": input_text, "translation": translation, "model_name": model_name, "message": message}
 
 class Bergamot:
-    import bergamot
     config = bergamot.ServiceConfig(numWorkers=4)
     service = bergamot.Service(config)
         
@@ -124,17 +124,15 @@ class Bergamot:
         except Exception as downloaderror:
             response, message_text = str(downloaderror), f"Error downloading {self.model_name}: {downloaderror}."
             print(downloaderror)
+    
     def translate(self):
         self.downloadbergamotfiles()
         try:
             model = self.service.modelFromConfigPath(f"{self.localfolder}/config.yml")
-            # model = self.service.modelFromConfig(self.localfolder)
             options = bergamot.ResponseOptions(alignment=False, sentenceMappings=False, qualityScores=False, HTML=False)
             rawresponse = self.service.translate(model, bergamot.VectorString(self.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(self.input_text), len(self.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 {self.sl} to {self.tl} with {self.model_name}."
         except Exception as translateerror:
             response, message_text = str(translateerror), f"Error translating from {self.sl} to {self.tl} with {self.model_name}: {translateerror}."
@@ -142,7 +140,7 @@ class Bergamot:
         return {"input": self.input_text, "translated_text": response, "message_text": message_text}
 
 # 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")
+@app.get("/bergamott", operation_id="get_bergamott", description="Translate text with Bergamot", tags=["bergamott"], summary="Translate text with Bergamot")
 def bergamott(input_text: list[str] = Query(description="Input string or list of strings"), sl: str = 'de', tl: str = 'en', model_name: Optional[str] = 'base/deen'):
     """
     Translates the input text from the source language to the target language using a specified model.
@@ -169,7 +167,7 @@ def bergamott(input_text: list[str] = Query(description="Input string or list of
         print(mainerror)
 
 # 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("/bergamots", operation_id="get_bergamot", description="Translate text with Bergamot", tags=["bergamot"], summary="Translate text with Bergamot")
+@app.get("/bergamots", operation_id="get_bergamot", description="Translate text with Bergamot", tags=["bergamots"], summary="Translate text with Bergamot")
 def bergamots(input_text: list[str] = Query(description="Input string or list of strings"), sl: str = 'de', tl: str = 'en', model_name: Optional[str] = 'base/deen'):
     """
     Translates the input text from the source language to the target language using a specified model.
@@ -256,7 +254,7 @@ mcp = FastApiMCP(
     describe_all_responses=True,
     describe_full_response_schema=True,
     include_operations=["get_translate", "get_paraphrase"],
-    include_tags=["paraphrase", "translate", "bergamot"]
+    include_tags=["paraphrase", "translate", "bergamott"]
 )
 # Mount the MCP server directly to the FASTAPI app
 mcp.mount()