src/story_engine.js

import { SISTER_LABEL, GIRL_LABEL, BAR_LABEL, DJ_LABEL, SHYGUY_LABEL } from "./constants.js";
import { ConversationLLM } from "./conversation_llm.js";

export function nameToLabel(name) {
  switch (name) {
    case "Jessica's sister":
      return SISTER_LABEL;
    case "bartender":
      return BAR_LABEL;
    case "DJ":
      return DJ_LABEL;
    case "Jessica":
      return GIRL_LABEL;
    case "Shyguy":
      return SHYGUY_LABEL;
    default:
      return "";
  }
}

class Bar {
  constructor() {
    this.name = "bartender";
    this.situation_prompt = "This conversation happens at the bar.";
    this.personality = "This is the bartender. He always offers triple vodka in his first answer. Then he is helpful, but he says that he has a crush on Jessica's sister";
    // this.past_conversation = "";
    this.imgpath = "assets/assets/barman.jpeg";
    this.output_format_prompt = "";
    this.functionDescriptions = [
      {
        key: "analyzeConsumption",
        description:
          "Analyze the conversation to determine how many beers were consumed. Also if heavy alcohol was consumed",
        parameters: {
          num_beers: {
            type: "number",
            description: "Total number of beers mentioned as being consumed in the conversation",
          },
          heavy_alcohol: {
            type: "boolean",
            description: "If heavy alcohol was consumed, True",
          },
        },
      },
    ];
    this.functionPrompt = "";
  }

  getSystemPrompt() {
    return `${this.personality}.`;
  }
}

class DJ {
  constructor() {
    this.name = "DJ";
    this.situation_prompt = "This is a conversation with the DJ.";
    this.personality =
      "This is the DJ. He is funny. He offers drugs to the shyguy. He can only play songs from the options [Let it be, Call me maybe, Shape of you]. If asked to play a song, he will play it and not propose any other song.";
    // this.past_conversation = "";
    this.imgpath = "assets/assets/dj.jpeg";
    this.output_format_prompt = "";
    this.functionDescriptions = [
      {
        key: "chooseSongToPlay",
        description: "Analyze the conversation to choose which song to play",
        parameters: {
          song: {
            type: "string",
            description: "The song to play",
          },
        },
      },
    ];
    this.functionPrompt =
      "Based on the conversation, choose a song from options [Let it be, Call me maybe, Shape of you]";
  }

  getSystemPrompt() {
    return `${this.personality}`;
  }
}

class Sister {
  constructor() {
    // Initialize Sister properties
    this.name = "Jessica's sister";
    this.mood = 1;
    this.situation_prompt = "This is a conversation with the sister of Jessica.";
    this.personality =
      "This is the sister of Jessica. She is a deeply religious Christian. Her first answer is rude. Her second answer is about christianity. First answer is helpful. If asked about the favourite song of the girl, she says happily that it is 'Call me maybe'.";
    // this.past_conversation = "";
    this.imgpath = "assets/assets/sister.jpeg";
    this.functionDescriptions = [
      {
        key: "analyzeMood",
        description: "Mood of the sister and gameover",
        parameters: {
          mood: {
            type: "number",
            description: "The mood of the sister on the scale of 1 to 10",
          },
          game_over: {
            type: "boolean",
            description: "Whether the game is over",
          },
        },
      },
    ];
    this.functionPrompt =
      "Analyze the conversation to determine the mood of the sister on the scale 1 to 10, based of how nice the conversation was. Also determine if the game is over. The game is over if shyguy was mean to the sister.";
  }

  getSystemPrompt() {
    return `${this.personality}. Her mood is ${this.mood} on the level 1 to 10. If the mood is low, she will be rude. If the mood is high, she will be helpful.`;
  }
}

class Girl {
  constructor(shyguy) {
    this.name = "Jessica";
    this.situation_prompt = "This is a conversation with the Jessica. She is the girl that shyguy likes.";
    this.personality = "This is Jessica. She is shy but nice. She likes the song 'Call me maybe'.";
    this.imgpath = "assets/assets/jessica.jpeg";
    this.output_format_prompt = "";
    this.shyguy = shyguy;
    this.functionDescriptions = [
      {
        key: "analyzeLiking",
        description: "Analyze if Jessica is happy with the conversation or not",
        parameters: {
          likes_shyguy: {
            type: "boolean",
            description: "If Jessica is happy with the conversation, True",
          },
        },
      },
    ];
    this.functionPrompt = "Analyze if Jessica is happy with the conversation or not";
  }

  getSystemPrompt() {
    if (this.shyguy.song_playing === "Call me maybe") {
      return `${this.personality}. She is very happy with the song playing. The first thing she says is that she really likes the music. Therefore she is nice and she likes shyguy. However, if he talks about algorithms, she does not like it and she becomes mean. Also if he talks with a lot of hesitation, she does not like it and she becomes mean.`;
    } else {
      return `${this.personality}. She does not like the song that shyguy is playing. The first thing she says is that the song is terrible. Then she is mean all the time.`;
    }
  }
}

class Wingman {
  constructor() {
    this.name = "wingman";
    this.situation_prompt = "This conversation happens with your wingman at the party.";
    this.personality = "This is your wingman. He is experienced with dating and gives practical advice. He's supportive but direct, and always encourages confidence without being aggressive.";
    this.imgpath = "assets/assets/wingman.jpeg";
    this.output_format_prompt = "";
    this.functionDescriptions = [
      {
        key: "analyzeAdvice",
        description: "Analyze the quality and impact of the conversation with the wingman",
        parameters: {
          confidence_boost: {
            type: "number",
            description: "How much confidence boost received on scale 1-5",
          },
          good_advice: {
            type: "boolean",
            description: "If practical, actionable advice was given, True",
          },
        },
      },
    ];
    this.functionPrompt = "Analyze how helpful the wingman's advice was and how much it boosted confidence";
  }

  getSystemPrompt() {
    return `${this.personality}.`;
  }
}



export class StoryEngine {
  constructor(shyguy) {
    // Initialize story engine properties with provided shyguy instance
    this.shyguy = shyguy;
    this.sister = new Sister();
    this.bar = new Bar();
    this.dj = new DJ();
    this.girl = new Girl(shyguy);
  }

  async onEncounter(entity) {
    switch (entity) {
      case SISTER_LABEL:
        return this.generalInteraction(this.sister);
      case BAR_LABEL:
        return this.generalInteraction(this.bar);
      case GIRL_LABEL:
        return this.generalInteraction(this.girl);
      case DJ_LABEL:
        return this.generalInteraction(this.dj);
      default:
        console.log("Unknown entity encountered");
    }
  }

  async generalInteraction(targetEntity) {
    await this.shyguy.learnLesson(targetEntity.name);
    console.log("Lesson for " + targetEntity.name + this.shyguy.lessons_learned);
    const conversation_llm = new ConversationLLM(
      "Shyguy",
      targetEntity.name,
      this.shyguy.getSystemPrompt(),
      targetEntity.getSystemPrompt(),
      targetEntity.situation_prompt,
      targetEntity.output_format_prompt,
      targetEntity.functionDescriptions,
      targetEntity.functionPrompt
    );
    const conversation_output = await conversation_llm.generateConversation(6);
    const conversation = conversation_output.conversation;

    // Append the conversation to shyguy's history
    this.shyguy.conversation_history += `\nConversation with ${targetEntity.name}:\n${conversation}\n`;

    let gameOver = this.decideGameOver(conversation_output.analysis.parameters.game_over);
    let gameSuccesful = this.decideGameSuccesful(conversation_output.analysis.parameters.likes_shyguy);
    if (targetEntity.name === "Jessica" && !gameSuccesful) {
      gameOver = true;
    }
    console.log("gameOver: " + gameOver);
    console.log("gameSuccesful: " + gameSuccesful);
    console.log(conversation_output);

    this.updateStates(conversation_output.analysis, targetEntity.name);
    console.log("shyguy num_beers: " + this.shyguy.num_beers);

    return {
      conversation: conversation,
      char1imgpath: this.shyguy.imgpath,
      char2imgpath: targetEntity.imgpath,
      gameOver: gameOver,
      gameSuccesful: gameSuccesful,
    };
  }

  decideGameOver(gameOverParameter) {
    let gameOver = false;
    if (gameOverParameter === "none") {
      gameOver = false;
    } else if (gameOverParameter === true) {
      gameOver = true;
    } else {
      gameOver = false;
    }
    return gameOver;
  }

  decideGameSuccesful(likesShyguy) {
    let gameSuccesful = false;
    if (likesShyguy) {
      gameSuccesful = true;
    } else {
      gameSuccesful = false;
    }
    return gameSuccesful;
  }

  updateStates(conversation_analysis, targetName) {
    if (targetName === "Jessica's sister") {
      if (conversation_analysis.parameters.mood !== "none") {
        this.sister.mood = conversation_analysis.parameters.mood;
      }
    } else if (targetName === "bartender") {
      if (conversation_analysis.parameters.num_beers !== "none") {
        this.shyguy.num_beers += Number(conversation_analysis.parameters.num_beers);
        this.shyguy.courage += 2 * Number(conversation_analysis.parameters.num_beers);
        this.shyguy.num_beers += 3 * Number(conversation_analysis.parameters.heavy_alcohol === true);
        console.log("Shyguy num_beers inside updateStates: " + this.shyguy.num_beers);
        this.shyguy.courage += 3 * Number(conversation_analysis.parameters.heavy_alcohol);
      }
    } else if (targetName === "DJ") {
      if (conversation_analysis.parameters.song !== "none") {
        this.shyguy.song_playing = conversation_analysis.parameters.song;
      }
    }
  }
}