!7

static Map<S, Int> attractors = litorderedmap(
  "caffeine", 3,
  "coffee", 2,
  "cube", 1,
  "cubes", 1,
  "hello", 1,
  "some", 1,
  "please", 1);

sS text = "Helo I'd like som caffien-free cofee pleas";

p {
  L<S> tok = javaTok(text);
  for (int i = 1; i < l(tok); i += 2) {
    S t = tok.get(i);
    if (!startsWithLetter(t)) continue;
    tok.set(i, correctWord(t));
  }
  print(join(tok));
}

sS correctWord(S word) {
  word = toLower(word);
  if (attractors.containsKey(word)) ret word;
  S best = null;
  int bestScore = 1000;
  for (S attractor : keys(attractors)) {
    int limit = attractors.get(attractor);
    int diff = leven_limited(attractor, word, min(limit+1, bestScore));
    if (diff <= limit && diff < bestScore) {
      best = attractor;
      bestScore = diff;
    }
  }
  ret or(best, word);
}