Update script.js

script.js

@@ -1,49 +1,35 @@
-document.addEventListener("DOMContentLoaded", function () {
-    const sendButton = document.getElementById("send-button");
-    const userInput = document.getElementById("user-input");
-    const chatMessages = document.getElementById("chat-messages");
-
-    sendButton.addEventListener("click", async function () {
-        const message = userInput.value.trim();
-        if (!message) {
-            alert("لطفاً یک پیام وارد کنید!");
-            return;
-        }
-
-        // نمایش پیام کاربر
-        const userMessage = document.createElement("p");
-        userMessage.textContent = `You: ${message}`;
-        chatMessages.appendChild(userMessage);
-
-        // پاک کردن ورودی
-        userInput.value = "";
-
-        try {
-            // ارسال پیام به سرور
-            const response = await fetch("?message=" + encodeURIComponent(message), {
-                method: "GET",
-                headers: { "Content-Type": "application/json" },
-            });
-
-            if (!response.ok) {
-                throw new Error(`HTTP error! status: ${response.status}`);
-            }
-
-            const data = await response.json();
-
-            // نمایش پاسخ مدل
-            const assistantMessage = document.createElement("p");
-            assistantMessage.textContent = `Assistant: ${data.response}`;
-            chatMessages.appendChild(assistantMessage);
-        } catch (error) {
-            console.error("Error fetching response:", error);
-            const errorMessage = document.createElement("p");
-            errorMessage.textContent = "خطایی در دریافت پاسخ از سرور رخ داده است.";
-            errorMessage.style.color = "red";
-            chatMessages.appendChild(errorMessage);
-        }
-
-        // اسکرول به پایین
-        chatMessages.scrollTop = chatMessages.scrollHeight;
-    });
+// تابع ارسال پیام به 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();
+  }
+});