Update app.py

app.py

@@ -87,23 +87,23 @@ installed_pairs = set()
 
 # 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: list[str] = Query(description="Input list of strings"), sl: str = 'de', tl: str = 'en', model_name: Optional[str] = 'base/deen'):
+def bergamot(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.
     Parameters:
-        input_text (str): The source text to be translated
+        input_text (str | list[str]): The source text to be translated, can be either a string or a list of strings
         sl (str): The source language of the input text
         tl (str): The target language into which the input text is translated
         model_name (str): The selected translation model name
     Returns:
         dict:
             input_text(str): The input text in the source language
-            translated_text(str): The input text translated to the selected target language
-            message_text(str):  A descriptive message summarizing the translation process. Example: "Translated from English to German with ende."
+            translated_text(str): The input text translated into the selected target language
+            message_text(str):  A descriptive message summarizing the translation process. Example: "Translated from English to German with base/ende."
     
     Example:
-        >>> bergamot("Hello world", "en", "de", "ende")
-        {"input_text": "Hello world", "translated_text": "Hallo Welt", "message_text": "Translated from English to German with ende."}
+        >>> bergamot("Hello world", "en", "de", "base/ende")
+        {"input_text": "Hello world", "translated_text": "Hallo Welt", "message_text": "Translated from English to German with base/ende."}
     """
     try:
         import bergamot
@@ -117,9 +117,7 @@ def bergamot(input_text: list[str] = Query(description="Input list of strings"),
         localfolder = f"{subfolder}/{model_name}"
         # List all files in the repo
         all_files = list_repo_files(repo_id, repo_type='model')
-        print(len(all_files), 'installed_pairs', installed_pairs, 'defaultlocalfolder', localfolder)
-        # local_files_only (bool, optional, defaults to False) — If True, avoid downloading the file and return the path to the local cached file if it exists.
-        # dry_run (bool, optional, defaults to False) — If True, perform a dry run without actually downloading the file. Returns a DryRunFileInfo object containing information about what would be downloaded.
+        print('input text type:', type(input_text), len(all_files), 'installed_pairs', installed_pairs, 'defaultlocalfolder', localfolder)
         for branch in branches:
             branch_files = [f for f in all_files if f.startswith(branch)] 
             fullmodel_files = [f for f in branch_files if f.startswith(model_name)]
@@ -128,6 +126,8 @@ def bergamot(input_text: list[str] = Query(description="Input list of strings"),
             print('branch_files', len(branch_files), 'model_files', model_files)
             for file_path in model_files:
                 if localfolder not in installed_pairs:
+                    # local_files_only (bool, optional, defaults to False) — If True, avoid downloading the file and return the path to the local cached file if it exists.
+                    # dry_run (bool, optional, defaults to False) — If True, perform a dry run without actually downloading the file. Returns a DryRunFileInfo object containing information about what would be downloaded.
                     local_path = hf_hub_download(repo_id=repo_id, subfolder=model_name, filename=file_path, local_dir=subfolder)
                     print(f"Downloaded to: {local_path}") # Downloaded to: deen/base/deen/config.yml
                     # localfolder = local_path.rsplit('/', 1)[0]
@@ -139,15 +139,15 @@ def bergamot(input_text: list[str] = Query(description="Input list of strings"),
             print('installed_pairs', installed_pairs, 'localfolder', localfolder, 'dry_run', dryrunerror)
         model = service.modelFromConfigPath(f"{localfolder}/config.yml")
         # model = service.modelFromConfig(localfolder)
-        options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
+        options = bergamot.ResponseOptions(alignment=False, sentenceMappings-False, qualityScores=False, HTML=False)
         rawresponse = service.translate(model, bergamot.VectorString(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(input_text), len(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 {sl} to {tl} with {model_name}."
-    except Exception as error:
-        response, message_text = str(error), str(error)
+    except Exception as generalerror:
+        response, message_text = str(generalerror), f"Error translating from {sl} to {tl} with {model_name}: {generalerror}."
         print(error)
     return {"input": input_text, "translated_text": response, "message_text": message_text}