let video;

let flipVideo;

​

// Storing the label

let label = "waiting...";

​

// The classifier

let classifier;

let modelURL = 'https://teachablemachine.withgoogle.com/models/m_WBM9L4/';

​

// STEP 1: Load the model!

function preload() {

  classifier = ml5.imageClassifier(modelURL + 'model.json');

}

​

// Snake Game Variables

let snake;

let rez = 20;

let food;

let w;

let h;

​

function setup() {

  createCanvas(640, 480);

  // Create the video

  video = createCapture(VIDEO);

  video.size(640, 480);

  video.hide();

  // Mirro the video since we trained it that way!

  flipVideo = ml5.flipImage(video);

​

  // STEP 2: Start classifying

  classifyVideo();

​

  // Snake Game

  w = floor(width / rez);

  h = floor(height / rez);

  frameRate(5);

  snake = new Snake();

  foodLocation();

}

​

// STEP 2 classify!

function classifyVideo() {

  // Flip the video!

  flipVideo = ml5.flipImage(video);

  classifier.classify(flipVideo, gotResults);

}

​

// Snake Game

function foodLocation() {

  let x = floor(random(w));

  let y = floor(random(h));

  food = createVector(x, y);

}

​

// Control the game based on the label

function controlSnake() {

  if (label === "Class 1") {

    snake.setDir(-1, 0);

  } else if (label === "Class 2") {

    snake.setDir(1, 0);

  } else if (label === "Class 3") {

    snake.setDir(0, 1);

  } else if (label === "Class 4") {

    snake.setDir(0, -1);

  }

}

​

function draw() {

  background(255);

​

  // Draw the video?

  image(flipVideo, 0, 0);

  textSize(32);

  fill(0);

  text(label, 10, 50);

  textSize(12);

  text("Use fingers to control: one for left, two for right, three for up, four for down", 10,80)

​

  // Draw the game

  scale(rez);

  if (snake.eat(food)) {

    foodLocation();

  }

  snake.update();

  snake.show();

​

  if (snake.endGame()) {

    print("END GAME");

    background(255, 0, 0);

    noLoop();

  }

​

  noStroke();

  fill(255, 0, 0);

  rect(food.x, food.y, 1, 1);

}

​

// STEP 3: Get the classification!

function gotResults(error, results) {

  if (error) {

    console.error(error);

    return;

  }

  label = results[0].label;

  // Control the snake and classify again!

  controlSnake();

  classifyVideo();

}