lollms-webui/static/js/main.js

function update_main(){
  const chatWindow = document.getElementById('chat-window');
  const chatForm = document.getElementById('chat-form');
  const userInput = document.getElementById('user-input');

  chatForm.addEventListener('submit', event => {
      event.preventDefault();
      console.log("Submitting")
  
      // get user input and clear input field
      message = userInput.value;
      userInput.value = '';
      
      // add user message to chat window
      const sendbtn = document.querySelector("#submit-input")
      const waitAnimation = document.querySelector("#wait-animation")
      sendbtn.style.display="none";
      waitAnimation.style.display="block";
      console.log("Sending message to bot")

      let hiddenElement = undefined
      let messageTextElement = undefined

      fetch('/bot', {
          method: 'POST',
          headers: {
              'Content-Type': 'application/json'
          },
          body: JSON.stringify({ message })
      }).then(function(response) {
          const stream = new ReadableStream({
              start(controller) {
                  const reader = response.body.getReader();
                  function push() {
                      reader.read().then(function(result) {
                          if (result.done) {
                              sendbtn.style.display="block";
                              waitAnimation.style.display="none";
                              console.log(result)
                              controller.close();
                              return;
                          }
                          controller.enqueue(result.value);
                          push();
                      })
                  }
                  push();
              }
          });
          const textDecoder = new TextDecoder();
          const readableStreamDefaultReader = stream.getReader();
          let entry_counter = 0
          function readStream() {
              readableStreamDefaultReader.read().then(function(result) {
                  if (result.done) {
                      return;
                  }
  
                  text = textDecoder.decode(result.value);
  
                  // The server will first send a json containing information about the message just sent
                  if(entry_counter==0)
                  {
                    // We parse it and
                    infos = JSON.parse(text)
                    addMessage('User', infos.message, infos.id, 0, can_edit=true);
                    elements = addMessage(infos.sender, '', infos.response_id, 0, can_edit=true);
                    messageTextElement=elements['messageTextElement'];
                    hiddenElement=elements['hiddenElement'];
                    entry_counter ++;
                  }
                  else{
                    entry_counter ++;   
                    // For the other enrtries, these are just the text of the chatbot
                    for (const char of text) {
                          txt = hiddenElement.innerHTML;
                          if (char != '\f') {
                            txt += char
                            hiddenElement.innerHTML = txt
                            messageTextElement.innerHTML = txt
                          }
  
                      // scroll to bottom of chat window
                      chatWindow.scrollTop = chatWindow.scrollHeight;
                    }
                  }
  
                  readStream();
              });
          }
          readStream();
      });
  
  });  
}