feat: Hardcoded samples to be migrated - Text Translation Samples (#12531)

* Revised and corrected NMT model Sample and test
* Added Gemini translation sample

Author: Thoughtseize1

translate/samples/snippets/requirements.txt

@@ -1,3 +1,4 @@
 google-cloud-translate==3.16.0
 google-cloud-storage==2.9.0
 google-cloud-automl==2.11.1
+google-cloud-aiplatform[vertexai]

translate/samples/snippets/snippets_test.py

@@ -28,13 +28,13 @@ def test_list_languages(capsys: pytest.LogCaptureFixture) -> None:
     out, _ = capsys.readouterr()
     for language in results:
         print("{name} ({language})".format(**language))
-    assert "Afrikaans" in [r["name"] for r in results]
+    assert "Abkhaz" in results[0]["name"]
 
 
 def test_list_languages_with_target(capsys: pytest.LogCaptureFixture) -> None:
     results = snippets.list_languages_with_target("is")
     out, _ = capsys.readouterr()
-    assert "afríkanska" in [r["name"] for r in results]
+    assert "abkasíska" in results[0]["name"]
 
 
 def test_translate_text(capsys: pytest.LogCaptureFixture) -> None:

translate/samples/snippets/translate_v3_translate_text.py

@@ -14,39 +14,48 @@
 
 # [START translate_v3_translate_text]
 # Imports the Google Cloud Translation library
-from google.cloud import translate
+import os
 
+from google.cloud import translate_v3
 
-# Initialize Translation client
-def translate_text(
-    text: str = "YOUR_TEXT_TO_TRANSLATE", project_id: str = "YOUR_PROJECT_ID"
-) -> translate.TranslationServiceClient:
-    """Translating Text."""
-
-    client = translate.TranslationServiceClient()
-
-    location = "global"
+PROJECT_ID = os.environ.get("GOOGLE_CLOUD_PROJECT")
 
-    parent = f"projects/{project_id}/locations/{location}"
 
-    # Translate text from English to French
-    # Detail on supported types can be found here:
-    # https://cloud.google.com/translate/docs/supported-formats
+# Initialize Translation client
+def translate_text(
+    text: str = "YOUR_TEXT_TO_TRANSLATE",
+    language_code: str = "fr",
+) -> translate_v3.TranslationServiceClient:
+    """Translating Text from English.
+    Args:
+        text: The content to translate.
+        language_code: The language code for the translation.
+            E.g. "fr" for French, "es" for Spanish, etc.
+            Available languages: https://cloud.google.com/translate/docs/languages#neural_machine_translation_model
+    """
+
+    client = translate_v3.TranslationServiceClient()
+    parent = f"projects/{PROJECT_ID}/locations/global"
+    # Translate text from English to chosen language
+    # Supported mime types: # https://cloud.google.com/translate/docs/supported-formats
     response = client.translate_text(
-        request={
-            "parent": parent,
-            "contents": [text],
-            "mime_type": "text/plain",  # mime types: text/plain, text/html
-            "source_language_code": "en-US",
-            "target_language_code": "fr",
-        }
+        contents=[text],
+        target_language_code=language_code,
+        parent=parent,
+        mime_type="text/plain",
+        source_language_code="en-US",
     )
 
     # Display the translation for each input text provided
     for translation in response.translations:
         print(f"Translated text: {translation.translated_text}")
+    # Example response:
+    # Translated text: Bonjour comment vas-tu aujourd'hui?
 
     return response
 
 
 # [END translate_v3_translate_text]
+
+if __name__ == "__main__":
+    translate_text(text="Hello! How are you doing today?", language_code="fr")

translate/samples/snippets/translate_v3_translate_text_test.py

@@ -12,17 +12,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import os
-
 import pytest
 
 import translate_v3_translate_text
 
 
-PROJECT_ID = os.environ["GOOGLE_CLOUD_PROJECT"]
-
-
 def test_translate_text(capsys: pytest.LogCaptureFixture) -> None:
-    response = translate_v3_translate_text.translate_text("Hello World!", PROJECT_ID)
+    response = translate_v3_translate_text.translate_text("Hello World!", "fr")
     out, _ = capsys.readouterr()
     assert "Bonjour le monde" in response.translations[0].translated_text

translate/samples/snippets/translate_with_gemini.py

@@ -0,0 +1,67 @@
+# Copyright 2024 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# [START aiplatform_gemini_translate]
+import os
+
+import vertexai
+from vertexai.generative_models import GenerationResponse, GenerativeModel, Part
+
+PROJECT_ID = os.environ.get("GOOGLE_CLOUD_PROJECT")
+
+
+def translate_text(text: str, target_language_code: str = "fr") -> GenerationResponse:
+    """Translates the given text to the specified target language using the Gemini model.
+    Args:
+        text (str): The text to be translated.
+        target_language_code (str): The language code of the target language. Defaults to "fr" (French).
+            Available language codes: https://cloud.google.com/translate/docs/languages#neural_machine_translation_model
+    Returns:
+        responses: The response from the model containing the translated text.
+    """
+    # Initializes the Vertex AI with the specified project and location
+    vertexai.init(project=PROJECT_ID, location="europe-west2")
+
+    model = GenerativeModel("gemini-1.0-pro")
+
+    # Configuration for the text generation
+    generation_config = {
+        "candidate_count": 1,
+        "max_output_tokens": 50,
+        "temperature": 0.1,
+        "top_k": 1,
+        "top_p": 1.0,
+    }
+
+    # Creates a prompt with the text to be translated and the target language code
+    promt = Part.from_text(
+        f"TEXT_TO_TRANSLATE:{text}. TARGET_LANGUAGE_CODE:{target_language_code}."
+    )
+
+    responses = model.generate_content(
+        contents=[promt],
+        generation_config=generation_config,
+    )
+
+    print(responses.candidates[0].content.text)
+    # Example response:
+    # Bonjour ! Comment allez-vous aujourd'hui ?
+
+    return responses
+
+
+# [END aiplatform_gemini_translate]
+
+if __name__ == "__main__":
+    translate_text(text="Hello! How are you doing today?", target_language_code="fr")

translate/samples/snippets/translate_with_gemini_test.py

@@ -0,0 +1,20 @@
+# Copyright 2024 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import translate_with_gemini
+
+
+def test_translate_text_with_gemini() -> None:
+    response = translate_with_gemini.translate_text("Hello World!", "fr")
+    assert "Bonjour le monde" in response.candidates[0].content.text