package gt.umg.pedrito;

import de.daslaboratorium.machinelearning.classifier.bayes.BayesClassifier;
import java.util.Arrays;
import java.util.Scanner;

public class Pedrito {

    public static void main(String args[]){
        BayesClassifier bNet = new BayesClassifier<String, String>(); // Cerebro
        System.out.println("Soy pedrito el bot predictivo\n");

        String entrada = "";
        do{
            String etiqueta = "";
            System.out.println("Ingrese una frase\n");
            entrada = readConsole();
            System.out.println("Ingrese una etiqueta\n");
            etiqueta = readConsole();
            System.out.println("Frase ingresada " + entrada +
                    " Etiqueta " + etiqueta + "\n");
            //1- Dividir frase en palabras
            String[] palabras = entrada.split("\\s");

            //2- Ingresarlo a red bayesiana
            bNet.learn(etiqueta, Arrays.asList(palabras));
        }while(!entrada.equalsIgnoreCase("predecir"));

        System.out.println("Llego el momento de la inferencia\n");

        String fraseAPredecir = "";
        do{
            System.out.println("Ingrese frase a predecir \n");
            fraseAPredecir = readConsole();

            System.out.println("Mi inferencia es: " + bNet.classify(
        			Arrays.asList(fraseAPredecir.split("\\s"))));

        }while(!fraseAPredecir.equalsIgnoreCase("Salir"));

        System.out.println("Bye");
    }

    public static String readConsole(){
        Scanner scanner = new Scanner(System.in);
    	return scanner.nextLine();
    }

}