// تابع ارسال پیام به Streamlit
function sendMessage() {
  var userInput = document.getElementById("user-input").value;
  document.getElementById("user-input").value = ""; // پاک کردن ورودی
  // اضافه کردن پیام کاربر به لیست پیام‌ها
  var chatMessages = document.getElementById("chat-messages");
  var newMessage = document.createElement("div");
  newMessage.classList.add("message", "user-message");
  newMessage.innerText = userInput;
  chatMessages.appendChild(newMessage);
  // ارسال پیام به Streamlit
  Streamlit.setComponentValue("user-input", userInput);
}

// دریافت پاسخ از Streamlit و نمایش اون
Streamlit.on('update', function() {
  var response = Streamlit.getComponentValue("response");
  if (response) {
    // اضافه کردن پیام به لیست پیام‌ها
    var chatMessages = document.getElementById("chat-messages");
    var newMessage = document.createElement("div");
    newMessage.classList.add("message", "assistant-message");
    newMessage.innerText = response;
    chatMessages.appendChild(newMessage);
  }
});

// ارسال پیام با فشردن Enter
var userInput = document.getElementById("user-input");
userInput.addEventListener("keyup", function(event) {
  if (event.keyCode === 13) {
    event.preventDefault();
    document.getElementById("send-button").click();
  }
});