!7

srecord Line (
  S type, // 'user, 'suggestion, 'bot, 'eval, 'evalResult, 'metaComment
  S text
) {}

concept Interaction {
  new L<Line> lines;
  bool current;
  
  void addLine(Line line) {
    lines.add(line);
    change();
  }
}

sS lineToString(Line line) {
  ret lpad(10, firstToUpper(replaceCompleteValue(line.type, 'metaComment, 'comment))) + ": " + line.text;
}

module TrainInteraction {
  bool watchEvals;
  
  transient S imageID = #1101205;
  transient JTextArea taInteraction;
  transient S log;
  transient JTextField tfInput, tfToSay, tfMeta;
  transient JLabel lblOutput;
  transient Interaction current;

  start {
    db();
    current = uniq_sync(Interaction, current := true);
    log = lines(map lineToString(current.lines));
    
    dm_vmBus_onMessage('javaEval_starting, voidfunc(S code, bool realEval) {
      if (watchEvals)
        addLine(Line('eval, code));
    });
    
    for (S msg : ll('javaEval_OK, 'javaEval_error))
      dm_vmBus_onMessage(msg, voidfunc(S code, O result) {
        if (watchEvals)
          addLine(Line('evalResult, shorten(str(result), 1024)));
      });
  }

  visualize {
    onEnter(tfInput = jTextField(), r {
      addLine(Line('user, gtt(tfInput)))
    });
    onEnter(tfToSay = jTextField(), r {
      addLine(Line('suggestion, gtt(tfToSay)))
    });
    onEnter(tfMeta = jTextField(), r {
      addLine(Line('metaComment, gtt(tfMeta)))
    });
    taInteraction = dm_fieldTextArea('log);
    ret centerAndSouth(
      centerAndSouthWithMargins(
        westAndCenterWithMargin(jimage(imageID), taInteraction),
        northAndCenterWithMargin(
          fontSize(20, lblOutput = jCenteredBoldLabel(" ")),
          makeForm3(
            "Your input:", focusOnShow(fontSize(16, tfInput)),
            "Make me say:", fontSize(16, tfToSay),
            "Make a meta-comment:", fontSize(16, tfMeta))
          )),
      westAndCenterWithMargins(
        dm_fieldCheckBox('watchEvals),
        rightAlignedButtons("Start Over", rThread {
          if (empty(current.lines) || confirmOKCancel("Sure?")) {
            cset(current, lines := new L);
            setField(log := "");
          }
        },
        "Save Interaction", rThread {
          dm_call(dm_makeOrShowModule("#1020418/TrainedInteractions"), 'addInteraction, current);
          cset(current, current := false);
          current = uniq_sync(Interaction, current := true);
          setField(log := "");
        })));
  }
  
  Line addLine(Line line) {
    current.addLine(line);
    setField(log := log + lineToString(line) + "\n");
    ret line;
  }
}