Update main.py

main.py

@@ -25,15 +25,16 @@ HTML = """
     <input type="submit" value="SEND" />
   </form>
 
-  <p id="output" placeholder="Output"></p>
+  <p id="output"></p>
   <script>
     const prompt = document.getElementById("prompt");
     const output = document.getElementById("output");
 
     const ws = new WebSocket("wss://huggingface.co/spaces/DaniilAlpha/answerer-api:8000/answer");  
-    ws.onmessage = function (event) {
-      output.innerText = event.data;
-    }
+    ws.onopen = () => console.log("socket open!");
+    ws.onclose = () => console.log("socket closed!");
+    ws.onerror = (e) => console.error(`error: ${e}`);
+    ws.onmessage = (event) => output.innerText = event.data;
 
     function ask(event) {
       ws.send(prompt.value);