# from io import BytesIO
from typing import Optional

# import librosa
# import soundfile as sf
from streamlit_TTS import auto_play, text_to_audio

from .config import pipe_tts

# SAMPLING_RATE = 16_000


class T2A:

    # def __get_duration(self, raw: bytes) -> float:
    #     chunk = BytesIO(raw)
    #     audio, sample_rate = librosa.load(chunk, sr=SAMPLING_RATE)
    #     duration = librosa.get_duration(y=audio, sr=sample_rate)
    #     return duration

    def autoplay(self, input_text: Optional[str] = None, lang: str = "en") -> None:

        text = input_text
        output_model = pipe_tts(input_text)
        
        if text is not None:
            if isinstance(self.text, str):
                audio = text_to_audio(text, language=lang)
                auto_play(audio)
            else: 
                text = f"Text you provide is {type(self.text)} accepted only string type"
                audio = text_to_audio(text, language=lang)
                auto_play(audio)
        else:
            auto_play("Please check the input text you have provided, it has a value of None")

    # def get_audio(self) -> tuple[bytes, int, float]:
    #     try:
    #         synth = self.output_model["audio"][0]

    #         print(f"synth : {synth}")

    #         with BytesIO() as buffer:
    #             sf.write(buffer, synth, SAMPLING_RATE, format='wav')
    #             output = buffer.getvalue()  # bytes

    #         print(f"type : {type(output)}")

    #         duration = self.__get_duration(output)

    #         print(f"duration : {duration}")

    #         return output, SAMPLING_RATE, duration
    #     except Exception as e:
    #         logging.error(e)