Update main.py

main.py

@@ -31,14 +31,21 @@ HTML = """
     const output = document.getElementById("output");
 
     const ws = new WebSocket("wss://daniilalpha-answerer-api.hf.space/answer");  
-    ws.onopen = () => console.log("socket open!");
-    ws.onclose = (e) => console.log(`socket closed with code ${e.code}!`);
-    ws.onmessage = (e) => output.innerText = e.data;
+    ws.onmessage = (e) => answer(e.data);
 
     function ask(event) {
+      if(ws.readyState != 1) {
+        answer("websocket is not connected!");
+        return;
+      }
+
       ws.send(prompt.value);
       event.preventDefault();
     }
+
+    function answer(value) {
+      output.innerHTML = value;
+    }
   </script>
 </body>