Added tts api

Author: FAReTek1

scratchattach/other/other_apis.py

@@ -6,7 +6,7 @@
 from ..utils import commons
 from ..utils.exceptions import BadRequest
 from ..utils.requests import Requests as requests
-from ..utils.supportedlangs import SUPPORTED_CODES, SUPPORTED_NAMES
+from ..utils.supportedlangs import SUPPORTED_CODES, SUPPORTED_NAMES, tts_lang
 
 
 # --- Front page ---
@@ -163,3 +163,41 @@ def translate(language: str, text: str = "hello"):
         return response_json["result"]
     else:
         raise BadRequest(f"Language '{language}' does not seem to be valid.\nResponse: {response_json}")
+
+
+def text2speech(text: str = "hello", gender: str = "female", language: str = "en-US"):
+    """
+    Sends a request to Scratch's TTS synthesis service.
+    Returns:
+        - The TTS audio (mp3) as bytes
+        - The playback rate (e.g. for giant it would be 0.84)
+    """
+    if gender == "female" or gender == "alto":
+        gender = ("female", 1)
+    elif gender == "male" or gender == "tenor":
+        gender = ("male", 1)
+    elif gender == "squeak":
+        gender = ("female", 1.19)
+    elif gender == "giant":
+        gender = ("male", .84)
+    elif gender == "kitten":
+        gender = ("female", 1.41)
+    else:
+        gender = ("female", 1)
+
+    if language not in SUPPORTED_NAMES:
+        if language.lower() in SUPPORTED_NAMES:
+            language = language.lower()
+
+        elif language.title() in SUPPORTED_CODES:
+            language = SUPPORTED_NAMES[SUPPORTED_CODES.index(language.title())]
+
+    lang = tts_lang(language.title())
+    if lang is None:
+        warnings.warn(f"Language '{language}' is probably not a supported language")
+    else:
+        language = lang["speechSynthLocale"]
+
+    response = requests.get(f"https://synthesis-service.scratch.mit.edu/synth"
+                            f"?locale={language}&gender={gender[0]}&text={text}")
+    return response.content, gender[1]

scratchattach/utils/supportedlangs.py

@@ -1,8 +1,9 @@
 """
-List of supported languages of scratch's translate extension.
+List of supported languages of scratch's translate and text2speech extensions.
 Adapted from https://translate-service.scratch.mit.edu/supported?language=en
 """
 
+# Supported langs for translate
 SUPPORTED_LANGS = {'sq': 'Albanian', 'am': 'Amharic', 'ar': 'Arabic', 'hy': 'Armenian', 'az': 'Azerbaijani',
                    'eu': 'Basque', 'be': 'Belarusian', 'bg': 'Bulgarian', 'ca': 'Catalan',
                    'zh-tw': 'Chinese (Traditional)', 'hr': 'Croatian', 'cs': 'Czech', 'da': 'Danish', 'nl': 'Dutch',
@@ -27,3 +28,141 @@
 # for lang in raw:
 #     SUPPORTED_LANGS[lang["code"]] = lang["name"]
 # print(SUPPORTED_LANGS)
+
+# Language info for tts
+TTS_LANGUAGE_INFO = [
+    {
+        "name": 'Arabic',
+        "locales": ['ar'],
+        "speechSynthLocale": 'arb',
+        "singleGender": True
+    },
+    {
+        "name": 'Chinese (Mandarin)',
+        "locales": ['zh-cn', 'zh-tw'],
+        "speechSynthLocale": 'cmn-CN',
+        "singleGender": True
+    },
+    {
+        "name": 'Danish',
+        "locales": ['da'],
+        "speechSynthLocale": 'da-DK'
+    },
+    {
+        "name": 'Dutch',
+        "locales": ['nl'],
+        "speechSynthLocale": 'nl-NL'
+    },
+    {
+        "name": 'English',
+        "locales": ['en'],
+        "speechSynthLocale": 'en-US'
+    },
+    {
+        "name": 'French',
+        "locales": ['fr'],
+        "speechSynthLocale": 'fr-FR'
+    },
+    {
+        "name": 'German',
+        "locales": ['de'],
+        "speechSynthLocale": 'de-DE'
+    },
+    {
+        "name": 'Hindi',
+        "locales": ['hi'],
+        "speechSynthLocale": 'hi-IN',
+        "singleGender": True
+    },
+    {
+        "name": 'Icelandic',
+        "locales": ['is'],
+        "speechSynthLocale": 'is-IS'
+    },
+    {
+        "name": 'Italian',
+        "locales": ['it'],
+        "speechSynthLocale": 'it-IT'
+    },
+    {
+        "name": 'Japanese',
+        "locales": ['ja', 'ja-hira'],
+        "speechSynthLocale": 'ja-JP'
+    },
+    {
+        "name": 'Korean',
+        "locales": ['ko'],
+        "speechSynthLocale": 'ko-KR',
+        "singleGender": True
+    },
+    {
+        "name": 'Norwegian',
+        "locales": ['nb', 'nn'],
+        "speechSynthLocale": 'nb-NO',
+        "singleGender": True
+    },
+    {
+        "name": 'Polish',
+        "locales": ['pl'],
+        "speechSynthLocale": 'pl-PL'
+    },
+    {
+        "name": 'Portuguese (Brazilian)',
+        "locales": ['pt-br'],
+        "speechSynthLocale": 'pt-BR'
+    },
+    {
+        "name": 'Portuguese (European)',
+        "locales": ['pt'],
+        "speechSynthLocale": 'pt-PT'
+    },
+    {
+        "name": 'Romanian',
+        "locales": ['ro'],
+        "speechSynthLocale": 'ro-RO',
+        "singleGender": True
+    },
+    {
+        "name": 'Russian',
+        "locales": ['ru'],
+        "speechSynthLocale": 'ru-RU'
+    },
+    {
+        "name": 'Spanish (European)',
+        "locales": ['es'],
+        "speechSynthLocale": 'es-ES'
+    },
+    {
+        "name": 'Spanish (Latin American)',
+        "locales": ['es-419'],
+        "speechSynthLocale": 'es-US'
+    },
+    {
+        "name": 'Swedish',
+        "locales": ['sv'],
+        "speechSynthLocale": 'sv-SE',
+        "singleGender": True
+    },
+    {
+        "name": 'Turkish',
+        "locales": ['tr'],
+        "speechSynthLocale": 'tr-TR',
+        "singleGender": True
+    },
+    {
+        "name": 'Welsh',
+        "locales": ['cy'],
+        "speechSynthLocale": 'cy-GB',
+        "singleGender": True
+    }]
+
+
+def tts_lang(attribute: str, by: str = None):
+    for lang in TTS_LANGUAGE_INFO:
+        if by is None:
+            if attribute in lang.values():
+                return lang
+            continue
+
+        if lang.get(by) == attribute:
+            return lang