!7

concept Example {

  S input;

  Triple<S> triple;

  transient Triple<S> guessed;

}

static F1<S, Triple<S>> guesser;

static JLabel lblStatus;

static Map<S, Triple<S>> examples, errors;

p-subst {

  dbIndexing(Example, 'input);

  autoRestart();

  for (S s : keys(gac36k()))

    uniq(Example, input := s);

  print("Making CRUD");

  sexyTable_drag = false;

  final SimpleCRUD<Example> crud = new SimpleCRUD<Example>(Example) {

    void saveComponent(Example c, S field, JComponent comp) {

      if (eqOneOf(field, 'triple, 'guessed))

        cset(c, field, unstructureOrNull(getTextTrim(comp)));

      else

        super.saveComponent(c, field, comp);

    }

  };

  crud.renderer = func(Example e) {

    litorderedmap(Input := e.input, Triple := ai_tripleToString(e.triple), Guessed := ai_tripleToString(e.guessed))

  };

  crud.showMaximized();

  swing {

    tablePopupMenuItemFirst(crud.table, "Accept", voidfunc(int row) {

      for (Example e : crud.selectedConcepts())

        cset(e, triple := e.guessed);

    });

    containerAddFirst(crud.buttons, /*jbutton("Set multi...", r {

      final L<Example> l = crud.selectedConcepts();

      showTextFieldForm("Set multi (" + l(l) + ")", "Verb", first(l).verb_guessed, voidfunc(S s) {

        for (Example e : l) cset(e, verb := s);

      });

    }),*/ jbutton("Show errors", f showErrors));

    containerAddFirst(crud.buttons, jbutton("Clear caches", f clearAllCaches));

  }

  addToFrame(crud.panel, lblStatus = jlabel());

  thread { guessLoop(); }

}

static Map<S, Triple<S>> examples() {

  ret mapFromKeyAndValueField([Example e : list(Example) | e.triple != null], 'input, 'triple);

}

svoid guessLoop {

  repeat with sleep 5 {

    int changes = 0;

    new HashMap<S, Triple<S>> guessedMap;

    Window anim = miniLoadingAnim();

    try {

      examples = examples();

      int nTrain = l(examples)/2;

      Map<S, Triple<S>> trainExamples = keysToMap(examples, selectRandom_fast(keysList(examples), nTrain));

      guesser = new F1<S, Triple<S>>() {

        Triple<S> get(S s) {

          ret ai_tripelize(unquoteIfNotContainingOtherQuotes(s));

        }

      };

      for (Example e) {

        Triple<S> t = (Triple<S>) pcallF(guesser, e.input);

        changes += cset(e, guessed := t);

        guessedMap.put(e.input, t);

      }

      if (changes > 0) change();

      Pair<Double, Map<S, Triple<S>>> p = scoreTripelizer(mapGetter(guessedMap), examples);

      double score = p.a;

      errors = p.b;

      setText(lblStatus, n(examples, "example") + ", trained with " + l(trainExamples) + ", score: " + formatDouble(score*100, 1) + "% ("

      + iround((1-score)*l(examples)) + " missing)");

    } finally {

      disposeWindow(anim);

    }

  }

}

svoid showErrors {

  showTable(map(keys(errors), func(S input) {

    litorderedmap(Input := input, Real := ai_tripleToString(examples.get(input)), Guessed := ai_tripleToString(errors.get(input)))

  }), "Errors");

}