Configure MS Translator API region

patkub · Oct 27, 2024 · da7c79f · da7c79f
1 parent e478d0b
commit da7c79f
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -17,6 +17,7 @@ Create an `.env` file with your API keys:
 ```
 GENIUS_ACCESS_TOKEN=...
 MS_TRANSLATOR_KEY=...
+MS_TRANSLATOR_REGION=...
 ```
 - [Genius API](https://docs.genius.com)
 - [Azure Translator Resource](https://learn.microsoft.com/en-us/azure/ai-services/translator/create-translator-resource)

diff --git a/songs.py b/songs.py
@@ -51,7 +51,7 @@
 # Translate lyrics to English using Microsoft Azure AI Translator
 #
 
-lyrics_translator = TranslateLyrics(os.getenv("MS_TRANSLATOR_KEY"))
+lyrics_translator = TranslateLyrics(os.getenv("MS_TRANSLATOR_KEY"), os.getenv("MS_TRANSLATOR_REGION"))
 english_translation = lyrics_translator.translate_lyrics(song_lyrics)
 
 #

diff --git a/translate_lyrics/TranslateLyrics.py b/translate_lyrics/TranslateLyrics.py
@@ -2,12 +2,13 @@
 from requests.exceptions import RequestException
 
 class TranslateLyrics:
-    def __init__(self, translator_key):
+    def __init__(self, translator_key, region):
         """
         Translate song lyrics using Microsoft Azure AI Translator
         @param translator_key: string access token for Azure Translator Resource https://learn.microsoft.com/en-us/azure/ai-services/translator/create-translator-resource
         """
         self.subscription_key = translator_key
+        self.region = region
 
     def translate_lyrics(self, lyrics):
         """
@@ -28,7 +29,7 @@ def translate_lyrics(self, lyrics):
 
         headers = {
             'Ocp-Apim-Subscription-Key': self.subscription_key,
-            'Ocp-Apim-Subscription-Region': 'eastus',
+            'Ocp-Apim-Subscription-Region': self.region,
             'Content-type': 'application/json',
             'X-ClientTraceId': str(uuid.uuid4())
         }

diff --git a/translate_lyrics/test_translate_lyrics.py b/translate_lyrics/test_translate_lyrics.py
@@ -54,7 +54,8 @@ def test_translate_lyrics_sets_translator_key():
     Sets translator key
     """
     translator_key = "fake_translator_key"
-    translate_lyrics = TranslateLyrics(translator_key)
+    region = "eastus"
+    translate_lyrics = TranslateLyrics(translator_key, region)
     assert translate_lyrics.subscription_key == translator_key
 
 
@@ -66,7 +67,8 @@ def test_translate_lyrics(mock_req):
 
     # Prepare: instantiate TranslateLyrics with fake translator key
     translator_key = "fake_translator_key"
-    translate_lyrics_obj = TranslateLyrics(translator_key)
+    region = "eastus"
+    translate_lyrics_obj = TranslateLyrics(translator_key, region)
 
     # mock_requests_post returns a mocked response from "https://api.cognitive.microsofttranslator.com/"
     # call translate_lyrics_obj.translate_lyrics(lyrics)
@@ -84,7 +86,8 @@ def test_translate_lyrics_invalid(mock_req):
 
     # Prepare: instantiate TranslateLyrics with fake translator key
     translator_key = "fake_translator_key"
-    translate_lyrics_obj = TranslateLyrics(translator_key)
+    region = "eastus"
+    translate_lyrics_obj = TranslateLyrics(translator_key, region)
 
     # mock_requests_post returns a mocked response from "https://api.cognitive.microsofttranslator.com/"
     # call translate_lyrics_obj.translate_lyrics(lyrics)