// NYUSH F23 - gohai

​

// Note: this uses a build of an unreleased,

// experimental version of ml5.js - functions

// etc might change (gohai - 20230918)

​

let lm;

let words = [];

let curWord = 0;

let curX = 0;

let curY = 20;

​

function preload() {

  lm = ml5.languageModel('TinyStories-15M');

}

​

function setup() {

  createCanvas(400, 400);

  frameRate(15);

  background(0);

  select('#generate').mouseClicked(generateText);

}

​

function draw() {

  fill(0, 2);

  rect(0, 0, width, height);

​

  if (curWord < words.length) {

    let wordWidth = textWidth(words[curWord]);

    if (curX+wordWidth > width) {

      curX = 0;

      curY += textAscent() + textDescent();

    }

    fill(255);

    text(words[curWord] + ' ', curX, curY);

    curX += wordWidth;

    curWord++;

  }

}

​

function generateText() {

  let prompt = select('#prompt').value();

  console.log('Prompt is "' + prompt + '"');

​

  let options = {

    temperature: 0.9

  };

  lm.generate(prompt, options, gotText);

}

​

function gotText(result) {

  console.log('Model returned "' + result.text + '"');

  // lm.words contains the output broken up in words

  words = result.words;

  curWord = 0;

}

​

​