Update server.js

server.js

@@ -1,28 +1,19 @@
-const express = require('express');
-const http = require('http');
-const socketIo = require('socket.io');
+const WebSocket = require('ws');
+const PORT = 7860;
 
-const app = express();
-const server = http.createServer(app);
-const io = socketIo(server);
+const wss = new WebSocket.Server({ port: PORT });
 
-let visitorCount = 0;
+wss.on('connection', (ws) => {
+  console.log('Client connected');
 
-app.get('/', (req, res) => {
-    res.json({ message: 'WebSocket server is running', visitorCount });
-});
-
-io.on('connection', (socket) => {
-    visitorCount++;
-    io.emit('visitorCount', visitorCount);
+  ws.on('message', (message) => {
+    console.log(`Received: ${message}`);
+    ws.send(`Echo: ${message}`);
+  });
 
-    socket.on('disconnect', () => {
-        visitorCount--;
-        io.emit('visitorCount', visitorCount);
-    });
+  ws.on('close', () => {
+    console.log('Client disconnected');
+  });
 });
 
-const PORT = process.env.PORT || 7860; // Default to 7860 for Hugging Face
-server.listen(PORT, () => {
-    console.log(`Server running on http://localhost:${PORT}`);
-});
+console.log(`WebSocket server is running on ws://localhost:${PORT}`);