fix(translate): update sample translate_text to unify across languages (#13395)

* fix(translate): update sample `translate_text` and unify across languages

* fix(translate): fix call to `translate_text` in CLI

* fix(translate): implement support to list[str] in `translate_text` sample

Author: eapl-gemugami

translate/samples/snippets/snippets.py

@@ -21,6 +21,8 @@
 https://cloud.google.com/translate/docs.
 """
 
+from __future__ import annotations
+
 import argparse
 
 
@@ -116,30 +118,60 @@ def translate_text_with_model(target: str, text: str, model: str = "nmt") -> dic
 
 
 # [START translate_translate_text]
-def translate_text(target: str, text: str) -> dict:
-    """Translates text into the target language.
-
-    Target must be an ISO 639-1 language code.
-    See https://g.co/cloud/translate/v2/translate-reference#supported_languages
+def translate_text(
+    text: str | bytes | list[str] = "¡Hola amigos y amigas!",
+    target_language: str = "en",
+    source_language: str | None = None,
+) -> dict:
+    """Translates a given text into the specified target language.
+
+    Find a list of supported languages and codes here:
+    https://cloud.google.com/translate/docs/languages#nmt
+
+    Args:
+        text: The text to translate. Can be a string, bytes or a list of strings.
+              If bytes, it will be decoded as UTF-8.
+        target_language: The ISO 639 language code to translate the text into
+                         (e.g., 'en' for English, 'es' for Spanish).
+        source_language: Optional. The ISO 639 language code of the input text
+                         (e.g., 'fr' for French). If None, the API will attempt
+                         to detect the source language automatically.
+
+    Returns:
+        A dictionary containing the translation results.
     """
+
     from google.cloud import translate_v2 as translate
 
     translate_client = translate.Client()
 
     if isinstance(text, bytes):
-        text = text.decode("utf-8")
+        text = [text.decode("utf-8")]
 
-    # Text can also be a sequence of strings, in which case this method
-    # will return a sequence of results for each text.
-    result = translate_client.translate(text, target_language=target)
+    if isinstance(text, str):
+        text = [text]
 
-    print("Text: {}".format(result["input"]))
-    print("Translation: {}".format(result["translatedText"]))
-    print("Detected source language: {}".format(result["detectedSourceLanguage"]))
+    # If a string is supplied, a single dictionary will be returned.
+    # In case a list of strings is supplied, this method
+    # will return a list of dictionaries.
 
-    return result
+    # Find more information about translate function here:
+    # https://cloud.google.com/python/docs/reference/translate/latest/google.cloud.translate_v2.client.Client#google_cloud_translate_v2_client_Client_translate
+    results = translate_client.translate(
+        values=text,
+        target_language=target_language,
+        source_language=source_language
+    )
 
+    for result in results:
+        if "detectedSourceLanguage" in result:
+            print(f"Detected source language: {result['detectedSourceLanguage']}")
 
+        print(f"Input text: {result['input']}")
+        print(f"Translated text: {result['translatedText']}")
+        print()
+
+    return results
 # [END translate_translate_text]
 
 
@@ -178,4 +210,4 @@ def translate_text(target: str, text: str) -> dict:
     elif args.command == "list-languages-with-target":
         list_languages_with_target(args.target)
     elif args.command == "translate-text":
-        translate_text(args.target, args.text)
+        translate_text(text=args.text, target_language=args.target)

translate/samples/snippets/snippets_test.py

@@ -38,13 +38,13 @@ def test_list_languages_with_target(capsys: pytest.LogCaptureFixture) -> None:
 
 
 def test_translate_text(capsys: pytest.LogCaptureFixture) -> None:
-    result = snippets.translate_text("is", "Hello world")
+    result = snippets.translate_text(text="Hello world", target_language="is")
     out, _ = capsys.readouterr()
-    assert "Halló heimur" in result["translatedText"]
+    assert "Halló heimur" in result[0]["translatedText"]
 
 
 def test_translate_utf8(capsys: pytest.LogCaptureFixture) -> None:
     text = "파인애플 13개"
-    result = snippets.translate_text("en", text)
+    result = snippets.translate_text(text=text, target_language="en")
     out, _ = capsys.readouterr()
-    assert "13 pineapples" in result["translatedText"]
+    assert "13 pineapples" in result[0]["translatedText"]