!752

static L<E> poem;

static int index;

static new Map<S, S> wordMap;

p {

  poem = parsePoem(first(loadPoems("#1003074")).lines);

  printStructure(poem);

  testWith(parsePoem(first(loadPoems("#1003076")).lines));

  reset();

  makeBot("Chemistry Bot");

}

static void testWith(L<E> test) {

  for (int i = 0; i+1 < l(test); i += 2) {

    S q = test.get(i).q, a = test.get(i+1).a;

    S answer = callStaticAnswerMethod(mc(), q);

    if (!linesEq(a, answer))

      fail("Test failed: " + quote(answer) + " vs " + quote(a));

    print("OK: " + answer);

  }

  print("Test OK!");

}

static bool linesEq(S a, S b) {

  ret eqic(simplifyLine(a), simplifyLine(b));

}

static S simplifyLine(S a) {

  ret join(" ", codeTokens(nlTok2(a)));

}

static void reset() {

  index = 0;

  wordMap.clear();

}

answer {

  exceptionToUser {

    if (index >= l(poem)) ret "Sorry, out of script. ^^";

    learnLine(assertNotNull(poem.get(index).q), s);

    ++index;

    if (index >= l(poem)) ret "Sorry, out of script. ^^";

    S a = poem.get(index).a;

    if (nempty(a)) {

      ++index;

      ret translateLine(a);

    }

  }

}

static void learnLine(S in, S out) {

  L<S> t1 = nlTok2(dropPunctuation2(in));

  L<S> t2 = nlTok2(dropPunctuation2(out));

  if (l(t1) != l(t2)) fail("Please say something like this: " + in);

  for (int i = 1; i < l(t1); i += 2) {

    S w1 = t1.get(i), w2 = t2.get(i);

    if (!eqic(w1, w2))

      // just overwrite - be flexible!

      wordMap.put(w1.toLowerCase(), w2.toLowerCase());

  }

}

static S translateLine(S s) {

  L<S> tok = nlTok2(s);

  for (int i = 1; i < l(tok); i += 2) {

    S w = wordMap.get(tok.get(i).toLowerCase());

    if (nempty(w))

      tok.set(i, w);

  }

  ret join(tok);

}