edu.cmu.sv.system_action.JsonExecutor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of yoda Show documentation
A library that allows rapid prototyping of dialog systems (language understanding, discourse modelling, dialog management, language generation).
There is a newer version: 0.7.0
Show newest version
package edu.cmu.sv.system_action;

import edu.cmu.sv.dialog_state_tracking.Turn;
import edu.cmu.sv.natural_language_generation.NaturalLanguageGenerator;
import edu.cmu.sv.semantics.SemanticsModel;
import edu.cmu.sv.system_action.dialog_act.DialogAct;
import edu.cmu.sv.system_action.non_dialog_task.NonDialogTask;
import edu.cmu.sv.yoda_environment.YodaEnvironment;
import org.apache.commons.lang3.tuple.ImmutablePair;
import org.json.simple.JSONObject;

import java.util.Calendar;
import java.util.Map;

/**
 * Created by David Cohen on 12/21/14.
 *
 * To execute actions, write a JSON representation of the action to stdout and flush
 *
 */
public class JsonExecutor implements Executor {
    YodaEnvironment yodaEnvironment;

    public JsonExecutor(YodaEnvironment yodaEnvironment) {
        this.yodaEnvironment = yodaEnvironment;
    }

    @Override
    public void execute(SystemAction systemAction){
        if (systemAction instanceof DialogAct){
            SemanticsModel model = ((DialogAct) systemAction).getNlgCommand();

            NaturalLanguageGenerator.getLogger().info("nlg request made:"+model);
            Map.Entry chosenUtterance =
                    yodaEnvironment.nlg.generateBestForSemantics(model);
            chosenUtterance.getValue().filterOutLeafSlot("chunk-start");
            chosenUtterance.getValue().filterOutLeafSlot("chunk-end");
            NaturalLanguageGenerator.getLogger().info("chosen utterance:" + chosenUtterance);

            JSONObject outputContent = SemanticsModel.parseJSON("{\"messageType\":\"tts\", \"content\":\""+chosenUtterance.getKey()+"\"}");
            yodaEnvironment.out.sendOutput(outputContent.toJSONString());
            Turn systemTurn = new Turn("system", chosenUtterance.getValue(), model, null, null);
            Calendar calendar = Calendar.getInstance();
            yodaEnvironment.DstInputQueue.add(new ImmutablePair<>(systemTurn, calendar.getTimeInMillis()));

        } else if (systemAction instanceof NonDialogTask){

            JSONObject taskSemantics = SemanticsModel.parseJSON("{\"dialogAct\":\""+systemAction.getClass().getSimpleName()+"\"}");
            taskSemantics.put("verb", SemanticsModel.parseJSON(((NonDialogTask) systemAction).getTaskSpec().toJSONString()));

            SemanticsModel taskSemanticModel = new SemanticsModel(taskSemantics);
            ((NonDialogTask) systemAction).execute(yodaEnvironment);

            JSONObject outputContent = SemanticsModel.parseJSON(
                    "{\"messageType\":\""+systemAction.getClass().getSimpleName()+"\", " +
                            "\"content\":"+((NonDialogTask) systemAction).getTaskSpec().toJSONString()+"}");
            yodaEnvironment.out.sendOutput(outputContent.toJSONString());

            Turn systemTurn = new Turn("system", taskSemanticModel.deepCopy(), taskSemanticModel, null, null);

            Calendar calendar = Calendar.getInstance();
            yodaEnvironment.DstInputQueue.add(new ImmutablePair<>(systemTurn, calendar.getTimeInMillis()));

        } else {
            throw new Error("Can not execute this type of action: "+systemAction);
        }
    }

    @Override
    public void executeUntracked(SystemAction systemAction) {

        if (systemAction instanceof DialogAct){
            SemanticsModel model = ((DialogAct) systemAction).getNlgCommand();
            NaturalLanguageGenerator.getLogger().info("nlg request made:"+model);
            Map.Entry chosenUtterance =
                    yodaEnvironment.nlg.generateBestForSemantics(model);
            chosenUtterance.getValue().filterOutLeafSlot("chunk-start");
            chosenUtterance.getValue().filterOutLeafSlot("chunk-end");
            NaturalLanguageGenerator.getLogger().info("chosen utterance:" + chosenUtterance);
            JSONObject outputContent = SemanticsModel.parseJSON("{\"messageType\":\"tts\", \"content\":\""+chosenUtterance.getKey()+"\"}");
            yodaEnvironment.out.sendOutput(outputContent.toJSONString());

        } else if (systemAction instanceof NonDialogTask){
            JSONObject taskSemantics = SemanticsModel.parseJSON("{\"dialogAct\":\""+systemAction.getClass().getSimpleName()+"\"}");
            taskSemantics.put("verb", SemanticsModel.parseJSON(((NonDialogTask) systemAction).getTaskSpec().toJSONString()));
            ((NonDialogTask) systemAction).execute(yodaEnvironment);
            JSONObject outputContent = SemanticsModel.parseJSON(
                    "{\"messageType\":\""+systemAction.getClass().getSimpleName()+"\", " +
                            "\"content\":"+((NonDialogTask) systemAction).getTaskSpec().toJSONString()+"}");
            yodaEnvironment.out.sendOutput(outputContent.toJSONString());

        } else {
            throw new Error("Can not execute this type of action: "+systemAction);
        }
    }
}