!759

static S scriptID = "#1003305";

!include #1003380 // Chat

static JTextArea thoughtArea;

static Chat chat;

static new L<E> script;

static int idx;

// learned examples - >0 = move, <0 = stop

static PersistentMap<O, Int> examples;

p {

  examples = new PersistentMap("examples");

  printStructure(examples);

  script = parsePoem(scriptID);

  chat = new Chat;

  thoughtArea = new JTextArea;

  JPanel buttons = hgrid(

    jbutton("Stop", "learnStop"),

    jbutton("Move", "learnMove"));

  showSideBySide(programTitle(), chat.swing(),

    withTitle("Thoughts", centerAndSouth(thoughtArea, buttons)));

  chat.requestFocus();

  chat.onEnter_pre = runnable {

    if (eq(probableNextMove(), "move")) {

      ++idx;

      predict;

    }

  };

  onUpdate(chat.chatField, "predict");

  //swingEverySecond(thoughtArea, runnable { predict; });

  predict;

}

svoid predict {

  new StringBuilder buf;

  try {

    buf.append(unparsePoem(subList(script, idx)) + "\n\n");

    S move = probableNextMove();

    buf.append("Probable next move: " + move + "\n\n");

  } catch Throwable e {

    buf.append("\n\n" + getStackTrace(e));

  }

  setText(thoughtArea, buf);

}

sS userInput { ret chat.chatField.getText(); }

sS probableNextMove {

  S input = userInput();

  // stop at end

  if (idx >= l(script)) ret "stop";

  // move if correct input

  E e = get(script, idx);

  if (e.q != null && match(e.q, input)) ret "move";

  // ask learning module

  S cmd = query(makeState());

  if (nempty(cmd)) ret cmd;

  // anything

  ret "stop";

}

sL makeState {

  ret litlist(idx, scriptID, userInput());

}

svoid learnStop { learn(makeState(), -1); }

svoid learnMove { learn(makeState(),  1); }

svoid learn(O state, int value) {

  int count = toInt(examples.get(state)) + value;

  print("Learner: Updating " + structure(state) + " to " + count);

  examples.put(state, count);

  predict;

}

sS query(O state) {

  int count = toInt(examples.get(state));

  print("Learner: Query " + structure(state) + " = " + count);

  ret count > 1 ? "move" : count < 0 ? "stop" : null;

}