Update server.js

server.js

@@ -7,8 +7,7 @@ import { CallToolRequestSchema, ListToolsRequestSchema } from "@modelcontextprot
 
 const app = express();
 app.use(cors());
-// Middleware to parse JSON for the /messages endpoint
-app.use('/messages', express.json());
+// ❌ express.json() ব্যবহার করা যাবে না, কারণ MCP SDK নিজেই র ডেটা স্ট্রিম হ্যান্ডেল করে।
 
 const PORT = process.env.PORT || 7860;
 
@@ -30,7 +29,8 @@ async function initBrowser() {
             '--no-sandbox',
             '--disable-setuid-sandbox',
             '--disable-dev-shm-usage',
-            '--disable-gpu'
+            '--disable-gpu',
+            '--window-size=1280,800'
         ],
         executablePath: process.env.PUPPETEER_EXECUTABLE_PATH || '/usr/bin/google-chrome-stable'
     });
@@ -97,7 +97,7 @@ mcpServer.setRequestHandler(CallToolRequestSchema, async (request) => {
             }
             case "get_page_content": {
                 const text = await page.evaluate(() => document.body.innerText);
-                return { content: [{ type: "text", text: text.substring(0, 5000) }] }; // Limit characters
+                return { content: [{ type: "text", text: text.substring(0, 5000) }] }; 
             }
             case "click_element": {
                 const { selector } = request.params.arguments;
@@ -110,27 +110,59 @@ mcpServer.setRequestHandler(CallToolRequestSchema, async (request) => {
                 return { content: [{ type: "text", text: `Typed "${text}" into ${selector}` }] };
             }
             default:
-                throw new Error("Tool not found");
+                throw new Error(`Tool ${request.params.name} not found`);
         }
     } catch (error) {
         return { content: [{ type: "text", text: `Error executing tool: ${error.message}` }], isError: true };
     }
 });
 
+// Store active SSE connections (Professional Session Management)
+const transports = new Map();
+
 // SSE Endpoint for MCP Connection
-let transport;
 app.get('/sse', async (req, res) => {
-    transport = new SSEServerTransport("/messages", res);
-    await mcpServer.connect(transport);
-    console.log("New MCP Client connected via SSE");
+    try {
+        // Generate a unique session ID for each connection
+        const sessionId = Math.random().toString(36).substring(2, 15);
+        
+        // Build dynamic absolute URL for Hugging Face Proxy routing
+        const protocol = req.headers['x-forwarded-proto'] || req.protocol;
+        const host = req.headers['x-forwarded-host'] || req.get('host');
+        const messagesEndpoint = `${protocol}://${host}/messages?sessionId=${sessionId}`;
+        
+        const transport = new SSEServerTransport(messagesEndpoint, res);
+        transports.set(sessionId, transport);
+        
+        await mcpServer.connect(transport);
+        console.log(`New MCP Client connected via SSE [Session: ${sessionId}]`);
+        
+        // Cleanup on disconnect
+        req.on('close', () => {
+            console.log(`Client disconnected [Session: ${sessionId}]`);
+            transports.delete(sessionId);
+        });
+
+    } catch (error) {
+        console.error("Error establishing SSE connection:", error);
+        res.status(500).send("Internal Server Error");
+    }
 });
 
-// Endpoint to receive messages from the client
+// Endpoint to receive POST messages from the client
 app.post('/messages', async (req, res) => {
+    const sessionId = req.query.sessionId;
+    const transport = transports.get(sessionId);
+    
     if (transport) {
-        await transport.handlePostMessage(req, res);
+        try {
+            await transport.handlePostMessage(req, res);
+        } catch (error) {
+            console.error(`Error handling POST message for session ${sessionId}:`, error);
+        }
     } else {
-        res.status(400).send("SSE connection not established");
+        console.warn(`Received message for unknown session: ${sessionId}`);
+        res.status(404).send("Session not found or expired");
     }
 });