from pydub import AudioSegment
from io import BytesIO
import random

class VoiceManager:

    def __init__(self):
        self.yes = AudioSegment.from_ogg("/home/luca/git_repos/telegram_amicobot/data/vocal/si.ogg")
        self.no = AudioSegment.from_ogg("/home/luca/git_repos/telegram_amicobot/data/vocal/no.ogg")

    def oracle(self, vocal_message):
        vm = AudioSegment.from_file(BytesIO(vocal_message), format="ogg")

        random_number = random.choice([0, 1])

        if random_number==0:
            combined = vm + self.no
        else:
            combined = vm + self.yes

        return combined.export(format="ogg", codec="libopus").read()