package main

import (
	"fmt"
	"strings"

	"github.com/eminano/markov"
)

func main() {
	// Create a chain for trigrams (3-grams)
	chain, _ := markov.NewNGramChain(3)

	// Parse text to process
	text := strings.NewReader(`
	I am batman. 
	I am groot.
	I am your father`)

	chain.ProcessText(text)

	// generate random text based on input
	output := chain.GenerateRandomText(10)

	// get a random candidate for the prefix
	candidate := chain.GetCandidate("I am")

	// get the probability of a given candidate for a prefix
	probability := chain.CandidateProbability("I am", "batman.")
}