Update app.py

app.py

@@ -41,38 +41,32 @@ html = """
     <button onclick="sendMessage()">Send</button>
 
     <script>
-        let ws = new WebSocket("wss://chris4k-a-i-statemachine.hf.space/ws");
-
-        ws.onmessage = (event) => {
-            let chatbox = document.getElementById("chatbox");
-            chatbox.innerHTML += `<p>${event.data}</p>`;
-            chatbox.scrollTop = chatbox.scrollHeight;
-        };
-
-        function sendMessage() {
-        console.log("sending .... ")
-            let input = document.getElementById("inputbox");
-            let message = input.value.trim();
-            if (message) {
-                ws.send(message);
-                input.value = "";
-            }
-        }
-
-
-
-      ws.onopen = function() {
-        console.log("WebSocket connected!");
-        ws.send("Hello AI!"); // Send message to backend once the connection is open.
-      };
-    
-      ws.onmessage = function(event) {
-        console.log("Message from server: ", event.data); // Logs the AI/system message.
-      };
-    
-      ws.onerror = function(error) {
-        console.error("WebSocket Error: ", error);
-      };    
+let ws = new WebSocket("wss://chris4k-a-i-statemachine.hf.space/ws");
+
+ws.onopen = function() {
+    console.log("WebSocket connected!");
+    ws.send("Hello AI!"); // Send initial message
+};
+
+ws.onmessage = (event) => {
+    console.log("Message from server: ", event.data);
+    let chatbox = document.getElementById("chatbox");
+    chatbox.innerHTML += `<p>${event.data}</p>`;
+    chatbox.scrollTop = chatbox.scrollHeight;
+};
+
+ws.onerror = function(error) {
+    console.error("WebSocket Error: ", error);
+};
+
+function sendMessage() {
+    let input = document.getElementById("inputbox");
+    let message = input.value.trim();
+    if (message) {
+        ws.send(message);
+        input.value = "";
+    }
+}   
  
         
     </script>