Snowball demo

demo_websocket.py

import snowboydecoder
import sys
import signal
import websocket
# Demo code for listening two hotwords at the same time

interrupted = False

ws = websocket.WebSocket()
ws.connect("ws://localhost:42002");

def signal_handler(signal, frame):
    global interrupted
    interrupted = True


def interrupt_callback():
    global interrupted
    return interrupted

if len(sys.argv) != 3:
    print("Error: need to specify 2 model names")
    print("Usage: python demo.py 1st.model 2nd.model")
    sys.exit(-1)

def turn_on():
    print("ON")
    ws.send("/radio/new/xx/1/on")

def turn_off():
    print("OFF")
    ws.send("/radio/new/xx/1/off")

models = sys.argv[1:]

# capture SIGINT signal, e.g., Ctrl+C
signal.signal(signal.SIGINT, signal_handler)

sensitivity = [0.5]*len(models)
detector = snowboydecoder.HotwordDetector(models, sensitivity=sensitivity)
callbacks = [lambda: turn_on(),
             lambda: turn_off()]
print('Listening... Press Ctrl+C to exit')

# main loop
# make sure you have the same numbers of callbacks and models
detector.start(detected_callback=callbacks,
               interrupt_check=interrupt_callback,
               sleep_time=0.03)

detector.terminate()