Update app.py

app.py

@@ -5,6 +5,7 @@ import urllib.request
 import re
 from http.server import BaseHTTPRequestHandler, HTTPServer
 import threading
+import time
 
 async def scrape_site(url: str) -> str:
     req = urllib.request.Request(url, headers={'User-Agent': 'Mozilla/5.0'})
@@ -29,7 +30,6 @@ class SimpleMCPServer:
         try:
             req = json.loads(request)
             method = req.get("method")
-
             if method == "initialize":
                 return json.dumps({
                     "jsonrpc": "2.0",
@@ -40,11 +40,9 @@ class SimpleMCPServer:
                         "serverInfo": {"name": "simple-mcp", "version": "1.0.0"}
                     }
                 })
-
             if method == "tools/list":
                 tools = [{"name": name, "description": info["description"]} for name, info in self.tools.items()]
                 return json.dumps({"jsonrpc": "2.0", "id": req.get("id"), "result": {"tools": tools}})
-
             if method == "tools/call":
                 tool_name = req["params"]["name"]
                 args = req["params"].get("arguments", {})
@@ -55,11 +53,9 @@ class SimpleMCPServer:
                         "id": req.get("id"),
                         "result": {"content": [{"type": "text", "text": result}]}
                     })
-
         except Exception as e:
             return json.dumps({"jsonrpc": "2.0", "error": {"code": -32000, "message": str(e)}})
 
-
 class MCPRequestHandler(BaseHTTPRequestHandler):
     def __init__(self, server_instance, *args, **kwargs):
         self.server_instance = server_instance
@@ -69,14 +65,12 @@ class MCPRequestHandler(BaseHTTPRequestHandler):
         try:
             content_length = int(self.headers['Content-Length'])
             post_data = self.rfile.read(content_length).decode('utf-8')
-
             loop = asyncio.new_event_loop()
             asyncio.set_event_loop(loop)
             try:
                 response = loop.run_until_complete(self.server_instance.handle_request(post_data))
             finally:
                 loop.close()
-
             self.send_response(200)
             self.send_header('Content-Type', 'application/json')
             self.send_header('Access-Control-Allow-Origin', '*')
@@ -98,31 +92,47 @@ class MCPRequestHandler(BaseHTTPRequestHandler):
         self.send_header('Access-Control-Allow-Headers', 'Content-Type')
         self.end_headers()
 
+    def do_GET(self):
+        if self.path == "/sse":
+            self.send_response(200)
+            self.send_header('Content-Type', 'text/event-stream')
+            self.send_header('Cache-Control', 'no-cache')
+            self.send_header('Connection', 'keep-alive')
+            self.send_header('Access-Control-Allow-Origin', '*')
+            self.end_headers()
+            try:
+                for i in range(1000):
+                    data = f"data: {{\"message\": \"ping {i}\"}}\n\n"
+                    self.wfile.write(data.encode('utf-8'))
+                    self.wfile.flush()
+                    time.sleep(1)
+            except BrokenPipeError:
+                pass
+        else:
+            self.send_response(404)
+            self.end_headers()
+
 def run_http_server(server_instance):
     def handler_factory(server_instance):
         def create_handler(*args, **kwargs):
             return MCPRequestHandler(server_instance, *args, **kwargs)
         return create_handler
-
     httpd = HTTPServer(('0.0.0.0', 7860), handler_factory(server_instance))
     print("Simple MCP HTTP Server started on port 7860")
-    print("Available tools: scrape_site, plaintext")
     httpd.serve_forever()
 
 async def main():
     server = SimpleMCPServer()
     server.register_tool("scrape_site", scrape_site, "Get HTML content from a website URL")
     server.register_tool("plaintext", plaintext, "Get plain text content from a website URL (removes HTML tags, scripts, and styles)")
-
     server_thread = threading.Thread(target=run_http_server, args=(server,))
     server_thread.daemon = True
     server_thread.start()
-
     try:
         while True:
             await asyncio.sleep(1)
     except KeyboardInterrupt:
-        print("Server shutting down...")
+        pass
 
 if __name__ == "__main__":
     asyncio.run(main())