import azure.cognitiveservices.speech as speechsdk


class TtsClientWrapper:
    def __init__(self):
        self.speech_key = "DzMBKYw2oVnL7brsyLZYlpvOqhiCDshlzRqQNtDOrwJA6KL5DOdcJQQJ99BCAC3pKaRXJ3w3AAAYACOGDWcJ"
        self.speech_region = "eastasia"
        self.voice_map = {
            0: "en-GB-OllieMultilingualNeural",
            1: "en-US-BrandonMultilingualNeural",
            2: "en-GB-AdaMultilingualNeural",
            3: "en-US-AmandaMultilingualNeural"
        }

    def getTtsClient(self, code):
        speech_config = speechsdk.SpeechConfig(subscription=self.speech_key, region=self.speech_region)
        speech_config.speech_synthesis_voice_name = self.voice_map[code]
        synthesizer = speechsdk.SpeechSynthesizer(speech_config=speech_config, audio_config=None)
        return synthesizer