"""
Created By: ishwor subedi
Date: 2024-07-31
"""
from src.components.speech_to_text import SpeechToText
from src.components.text_to_speech_gtts import TextToSpeech


class SpeechTranscriptionPipeline:
    def __init__(self):
        self.speech_to_text_ = SpeechToText()
        self.text_to_speech_ = TextToSpeech()

    def text_to_speech(self, text: str, lang: str, tld: str) -> str:
        """
        Convert text to speech.
        :param text: The text to convert to speech.
        :param lang: The language in which to convert the text.
        :return: The speech representation of the text.
        """
        speech = self.text_to_speech_.conversion(text, lang, tld)
        return speech

    def speech_to_text(self, audio, lang: str) -> str:
        """
        Convert speech to text.
        :param audio: The audio data to convert to text.
        :param lang: The language in which the audio is spoken.
        :return: The text representation of the audio.
        """
        transcript_with_timestamp, transcript = self.speech_to_text_.transcribe_audio(audio=audio, language=lang)
        return transcript