!7

sS keyphrase = "computer";
// bigger number means more tolerance
// 1e-40 works pretty well with few false positives
// 1e-30 gets no hits
// 1e-35 doesn't trigger reliably
// Using 1e-38 for now
sS threshold = "1e-38";

// seconds to listen after hot word
// TODO: keep listening when user said something
static int recognitionTimeout = 10;

sS icon = #1009664; //#1009662;

// milliseconds before starting recognition (time for saying "yep" or "ja")
static int delayBeforeRecognition = 300;

sO onKeyPhrase;
sbool exited;

sS python = trim([[
#!/usr/bin/python
from subprocess import Popen, PIPE, STDOUT
import pty
import os
import sys

cmd = 'pocketsphinx_continuous -inmic yes -keyphrase #KEYPHRASE# -kws_threshold #THRESHOLD#'

master, slave = pty.openpty()

p = Popen(cmd, shell=True, stdin=PIPE, stdout=slave, stderr=slave, close_fds=True)
stdout = os.fdopen(master)
while True:
    sys.stdout.write(stdout.readline())
    sys.stdout.flush()
]]);

static File fPython;
static JButton btnStartStop;
sbool paused;

p-subst {
  //onLeftClick(installTrayIcon(icon, "Computer!"), r { jump() });
  onLeftClick(
    installTrayIcon2(resizeImage(loadImage2(icon), 24), "Computer!",
      makePopupMenu(menuItem("Computer!", r { jump() }))),
      r { jump() });
  
  makeBot("Hot Word Detection.");
  unix_killAll("pocketsphinx");
  
  print("Key phrase is: " + keyphrase);
  
  onKeyPhrase = voidfunc(S s) { jump() };
  
  swing {
    showControls(jcenteredline(btnStartStop = jbutton("Stop recognition", f startStop)));
  }
  
  fPython = getProgramFile("pocketsphinx.py");
  makeExecutable(saveTextFile(fPython, python.replace("#THRESHOLD#", threshold).replace("#KEYPHRASE#", bashQuote(keyphrase)));
  
  titleStatus_setBaseTitle(consoleFrame(), "Hot Word Detection");
  infoBox("Say " + quote(keyphrase) + " to activate me!");
  start();
}

svoid start {
  backtickToConsole_lineBuf(f2s(fPython), voidfunc(S line) {
    print("Line read: " + quote(line));
    if (eq(line, keyphrase))
      pcallF(onKeyPhrase, line);
    else if (startsWith(line, "READY."))
      consoleStatus("Ready");
    else if (startsWith(line, "Listening."))
      consoleStatus("Listening");
  });
  paused = true;
  setText(btnStartStop, "Start recognition");
  consoleStatus("Stopped");
}

svoid cleanMeUp {
  if (!exited) { exited = true; unix_killAll("pocketsphinx"); }
}

svoid startStop {
  if (paused) {
    paused = false;
    setText(btnStartStop, "Stop recognition");
    thread { main.start(); }
  } else {
    paused = true;
    setText(btnStartStop, "Start recognition");
    unix_killAll("pocketsphinx");
  }
}

svoid jump {
  fS response = botAnswerWord();
  infoBox(response);
  thread { sleep(delayBeforeRecognition); sendOpt("Chrome Speech.", "start recognition timeout " + recognitionTimeout); }
  thread { sendOpt("Mouth.", response); }
}