Update script.js

script.js

@@ -1,39 +1,18 @@
-document.getElementById('send-button').addEventListener('click', function() {
-    let userInput = document.getElementById('user-input').value.trim();
-    if (userInput) {
-        displayMessage(userInput, 'user');
-        sendMessage(userInput);
-        document.getElementById('user-input').value = '';
-    }
-});
+document.addEventListener('DOMContentLoaded', () => {
+    var socket = io();
 
-function sendMessage(message) {
-    fetch('https://huggingface.co/spaces/Mosberg-Control/CodeNinjaOpenChat', {
-        method: 'POST',
-        headers: {
-            'Content-Type': 'application/json',
-        },
-        body: JSON.stringify({ message: message })
-    })
-    .then(response => {
-        if (!response.ok) {
-            throw new Error(`HTTP error! status: ${response.status}`);
-        }
-        return response.json();
-    })
-    .then(data => {
-        displayMessage(data.response, 'bot');
-    })
-    .catch((error) => {
-        console.error('Error:', error);
-        displayMessage(`Sorry, there was an error processing your request. Error: ${error.message}`, 'error');
-    });
-}
+    document.querySelector('#send-button').onclick = () => {
+        const message = document.querySelector('#user-input').value;
+        const selectedModel = document.querySelector('#model-selector').value;
+        socket.emit('send_message', {'message': message, 'model': selectedModel});
+        document.querySelector('#user-input').value = '';
+        return false;
+    };
 
-function displayMessage(message, sender) {
-    let chatBox = document.getElementById('chat-box');
-    let newMessage = document.createElement('div');
-    newMessage.classList.add('message', sender);
-    newMessage.textContent = message;
-    chatBox.appendChild(newMessage);
-}
+    socket.on('receive_message', data => {
+        const node = document.createElement("div");
+        const textnode = document.createTextNode(data.message);
+        node.appendChild(textnode);
+        document.querySelector('#chat-box').appendChild(node);
+    });
+});