from handler import EndpointHandler
import base64

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

# prepare sample payload
text_payload = {"text": "I am quite excited how this will turn out"}
audio_payload = {"audio": base64.b64encode(
    open("/home/arthur/data/musicdb/split_demucses/474499231_456751864 --__-- Breakbot - Programme_mp3_drums/chunk_11.wav", 'rb').read())}


# test the handler
# text_pred = my_handler(text_payload)
audio_pred = my_handler(audio_payload)

# show results
# print("text_pred", text_pred)
print("audio_pred", audio_pred)