!7

static int perPage = 8;
static L<S> guessPrograms = ll(#1008696);

static SingleComponentPanel scp;
static int startIndex;
static new L guessModules;
static ReliableSingleThread relearnThread = new(f relearn);
static JComboBox cbMode;

concept Sentence {
  S text;
  new SS data;
}

p-substance {
  unswing {
    db(); 
    indexConceptField(Sentence, "text");
    L<S> sentences = toLinesFullTrim(loadSnippet(#1008606));
    sentences = notStartingWith("\"", sentences);
    time { makeUniqueConcepts(Sentence, "text", sentences); }
    for (S id : guessPrograms)
      pcall { guessModules.add(runDependent(id)); }
  }
  showFrame(centerAndSouth(scp = new SingleComponentPanel,
    withTopMargin(20, jcenteredline(
      jbutton("Prev", r { makePanel(max(startIndex-perPage, 0)) }),
      jbutton("Next", f nextPage),
      jbutton("Random", r { makePanel(random(countConcepts(Sentence))/perPage*perPage) }),
      jbutton("All OK", r { clickButtons(findButtons(scp, "OK")); nextPage(); }),
      jbutton("Re-learn", r { relearnThread.trigger() })))));
  setFrameWidth(scp, 1000);
  centerFrame(scp);
  makePanel(0);
}

svoid makePanel(int startIndex) {
  main.startIndex = startIndex;
  makePanel();
}

svoid makePanel() {
  JPanel panel = gridLayoutPanel_trackWidth_evenlySpacedCols(2);
  panel.add(jlabel("Sentence"));
  panel.add(jline(cbMode = jcombobox_copySelection(cbMode, "Subject", "Verb")));
  //onChange(cbMode, f makePanel);
  for (final Sentence s : takeFirst(perPage, dropFirst(startIndex, list(Sentence)))) {
    final JTextField tfText = jtextfield_caretLeft(s.text);
    tfText.setCaret(new DefaultCaret {
      @Override
      public void setSelectionVisible(boolean visible) {
        super.setSelectionVisible(true);
      }
    });
    panel.add(tfText);

    final JTextField tf = jtextfield_caretLeft(dataMapToString(s.data));
    final JTextField tfGuess = jtextfield_caretLeft(dataMapToString(makeGuess(s.text)));
    JButton okBtn = jbutton("OK", r {
      pcall-messagebox {
        cset(s, data := dataMapFromString(getTextTrim(tf)));
        relearnThread.trigger();
      }
    });
    panel.add(vstack(
      centerAndEast(withLabel("Saved:", tf), okBtn),
      centerAndEast(withLabel("Guess:", tfGuess), jbutton("OK")),
    ));
    
    /*onCaret(tfText, r {
      if ((verbMode() ? s.verb : s.action) == null) {
        S a;
        IntRange r = IntRange(tfText.getSelectionStart(), tfText.getSelectionEnd());
        if (verbMode())
          a = ai_makeVerbAction(s.text, ll(r));
        else
          a = ai_makeSubjectAction(s.text, r);
        if (a != null) tf.setText(a);
      }
    });*/
  }
  scp.setComponent(jscroll(panel));
}

svoid relearn {
  for (O guessModule : guessModules) pcall { callMain(guessModule); }
  swing { makePanel(); }
}

svoid nextPage {
  if (startIndex+perPage < countConcepts(Sentence)) makePanel(startIndex+perPage);
}

sbool verbMode() {
  ret cbMode.getSelectedIndex() > 0;
}

static S dataMapToString(SS data) {
  ret join(" | ", mapMapToList(data, func(S a, S b) {
    a + "=" + b }));
}

static SS dataMapFromString(S input) {
  new SS data;
  for (S s : trimAll(splitByJavaToken(input, "|"))) {
    L<S> tok = javaTok(s);
    if (eq(get(tok, 3), "="))
      data.put(tok.get(1), join(subList(tok, 5)));
  }
  ret data;
}

static SS makeGuess(S sentence) {
  new SS data;
  for (O guessModule : guessModules)
    pcall(guessModule, "callGuesser", get(guessModule, "best"), sentence, data);
  ret data;
}