const { EmbedBuilder } = require('discord.js');
var unirest = require('unirest');
const jsonfile = require('jsonfile')

// Start session defaults:
var apiUrl = `http://${process.env.INTERNAL_IP}:8008/api/chat/`;

var model = 'gpt4all';
var temperature = 0.1;
var topK = 50;
var topP = 0.95;
var maxLength = 256;
var contextWindow = 512;
var repeatLastN = 64;
var repeatPenalty = 1.3;
var initPrompt = 'Below is an instruction that describes a task. Write a response that appropriately completes the request. The response must be accurate, concise and evidence-based whenever possible. A complete answer is always ended by [end of text].';
var nThreads = 7;
// End session defaults


// Set model list we have downloaded
let modelList = ["7B", "7B-native", "gpt4all"]

module.exports = {
  name: "create-session",
  description: "create a new session chat",
  private: true,
  options: [{
    "name": "model",
    "description": `The model you want to run, choose from the following: ${modelList.join(", ")} | Char case matters`,
    "required": false,
    "type": 3
  },
  {
    "name": "init-prompt",
    "description": "A prompt you want to init the chat with, a default is used if not provided.",
    "required": false,
    "type": 3
  },
  {
    "name": "temperature",
    "description": "The higher the temperature, the more random the model output. A default 0.1 is used if not provided.",
    "required": false,
    "type": 3
  },
  {
    "name": "repeat-penalty",
    "description": "The weight of the penalty to avoid repeating the last last_n tokens. 1.3 is used if not provided.",
    "required": false,
    "type": 3
  }
  ],



  run: async (client, interaction) => {
    const file = './cache/' + interaction.user.id

    let options = interaction.options._hoistedOptions;
    let varsToCheck = [
      { name: "model", value: null },
      { name: "temperature", value: null },
      { name: "init-prompt", value: null },
      { name: "repeat-penalty", value: null }
    ];

    for (let i = 0; i < options.length; i++) {
      let option = options[i];
      for (let j = 0; j < varsToCheck.length; j++) {
        let varToCheck = varsToCheck[j];
        if (option.name === varToCheck.name) {
          varToCheck.value = option.value;
          break;
        }
      }
    }
    // Now you can access the values of each variable you are interested in:
    let userInputModel = varsToCheck.find(v => v.name === "model")?.value;
    let userInputTemperature = varsToCheck.find(v => v.name === "temperature")?.value;
    let userInputInitPrompt = varsToCheck.find(v => v.name === "init-prompt")?.value;
    let userInputRepeatPenalty = varsToCheck.find(v => v.name === "repeat-penalty")?.value;

    // Init Prompt Setting
    if (userInputInitPrompt === null) {
      console.log("-- No init-prompt provided, using default --");
    } else {
      initPrompt = userInputInitPrompt;
    }

    // Modal Setting
    if (userInputModel === null) {
      console.log("-- No model provided, using default --")
    } else {
      if (modelList.includes(userInputModel)) {
        model = userInputModel;
      } else {
        let modelListStr = modelList.join(", ");
        return interaction.followUp(`You may only use one of the following: ${modelListStr}`);
      }
    }

    // temperature setting
    if (userInputTemperature === null) {
      console.log("-- No temperature provided, using default --")
    } else {
      const parsedTemperature = parseFloat(userInputTemperature);
      if (parsedTemperature >= 0.1 && parsedTemperature <= 2) {
        // temperature is within range
        temperature = parsedTemperature;
      } else {
        // temperature is outside of range
        return interaction.followUp(`Temperature must be between 0.1 and 2`);
      }
    }

    // repeat setting
    if (userInputRepeatPenalty === null) {
      console.log("-- No RepeatPenalty provided, using default --")
    } else {
      const parsedRepeatPenalty = parseFloat(userInputRepeatPenalty);
      if (parsedRepeatPenalty >= 0.1 && parsedRepeatPenalty <= 2) {
        // RepeatPenalty is within range
        repeatPenalty = parsedRepeatPenalty;
      } else {
        // RepeatPenalty is outside of range
        return interaction.followUp(`Repeat Penalty must be between 0.1 and 2`);
      }
    }


    var req = unirest('POST', apiUrl + '?model=' + model + '&temperature=' + temperature + '&top_k=' + topK + '&top_p=' + topP + '&max_length=' + maxLength + '&context_window=' + contextWindow + '&repeat_last_n=' + repeatLastN + '&repeat_penalty=' + repeatPenalty + '&init_prompt=' + initPrompt + '&n_threads=' + nThreads)
      .headers({
        'accept': 'application/json'
      })
      .send('')
      .end(function (response) {
        console.log(response.body);


        const obj = { id: response.body }

        jsonfile.writeFile(file, obj, function (err) {
          if (err) console.error(err)
        })

        const embed = new EmbedBuilder()
          .setColor("#FF0000")
          .setTitle("New Chat Session Started!")
          .setDescription(`New Chat Session ID: \n` + response.body)
          .setTimestamp()
          .setFooter({ text: `Requested by ${interaction.user.tag}`, iconURL: `${interaction.user.displayAvatarURL()}` });
        interaction.followUp({ embeds: [embed] });


      });
  }
};