let trained = false;

​

function setup() {

  createCanvas(400, 400);

  background(220);

}

​

function mousePressed() {

  if (keyIsDown(65)) {

    textAlign(CENTER);

    textSize(51);

    text("a", mouseX, mouseY);

    trainingData.push({ letter: "a", x: mouseX, y: mouseY });

  }

  if (keyIsDown(66)) {

    textAlign(CENTER);

    textSize(51);

    text("b", mouseX, mouseY);

    trainingData.push({ letter: "b", x: mouseX, y: mouseY });

  }

​

  if (trained) {

    // Get the predicted letter based on the output

    const predictedLetter = "a";

​

    textAlign(CENTER);

    textSize(51);

    text(predictedLetter, mouseX, mouseY);

  }

}

​

function keyPressed() {

  if (key == "t") {

    trained = true;

  }

}

​