import math

from handler import EndpointHandler
from scipy.io.wavfile import write

# init handler
my_handler = EndpointHandler(path=".")

generation_params = {
    "duration": 12,
}

# prepare sample payload
payload = {
    "prompt": "rock, rock and rock",
    "generation_params": generation_params,
    "audio_window": 2,
    "chunk_size": 4,
}

# test the handler
test = my_handler(payload)

write("test.wav", 32000, test["generated_audio"])