package main
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
func optionTweetMarkov(states []State) {
|
|
|
|
fmt.Print("entry the first word: ")
|
|
newcommand := bufio.NewReader(os.Stdin)
|
|
firstWord, _ := newcommand.ReadString('\n')
|
|
firstWord = strings.TrimSpace(firstWord)
|
|
fmt.Print("first word: ")
|
|
c.Purple(firstWord)
|
|
|
|
c.Red("how many words you want on the text?")
|
|
newcommand = bufio.NewReader(os.Stdin)
|
|
answer, _ := newcommand.ReadString('\n')
|
|
answer = strings.TrimSpace(answer)
|
|
fmt.Print("Number of words on text to generate: ")
|
|
c.Purple(answer)
|
|
count, err := strconv.Atoi(answer)
|
|
if err != nil {
|
|
fmt.Println("incorrect entry, need a positive number")
|
|
}
|
|
|
|
c.Red("how many sentences you want to generate?")
|
|
newcommand = bufio.NewReader(os.Stdin)
|
|
answer, _ = newcommand.ReadString('\n')
|
|
answer = strings.TrimSpace(answer)
|
|
fmt.Print("Number of sentences to generate: ")
|
|
c.Purple(answer)
|
|
sentences, err := strconv.Atoi(answer)
|
|
if err != nil {
|
|
fmt.Println("incorrect entry, need a positive number")
|
|
}
|
|
|
|
fmt.Println("generating text")
|
|
for i := 0; i < sentences; i++ {
|
|
generatedText := markov.generateText(states, firstWord, count)
|
|
c.Green(generatedText)
|
|
}
|
|
}
|