Update app.py

app.py

@@ -1,10 +1,9 @@
 import os
 import time
 import requests
-from fastapi import FastAPI, Response
+from fastapi import FastAPI, WebSocket, WebSocketDisconnect
 from fastapi.responses import StreamingResponse
 from datetime import datetime, timedelta
-from starlette.routing import Mount  # 用于挂载
 from fastmcp.server import FastMCP
 from contextlib import asynccontextmanager
 
@@ -61,7 +60,6 @@ async def fetch_github_trending():
 
 # 定义MCP服务器和工具
 mcp_server = FastMCP(name="GithubTrending")
-app.mount("/mcp", mcp_server.http_app())
 
 @mcp_server.tool()
 def get_trending_repos(num: int = 10) -> dict:
@@ -70,15 +68,7 @@ def get_trending_repos(num: int = 10) -> dict:
         fetch_github_trending()
     return {"trending": cached_trending[:num]}
 
-#app.mount("/mcp", mcp_server.streamable_http_app())
-
-async def trending_generator():
-    """SSE 事件生成器"""
-    while True:
-        if not last_updated or (datetime.now() - last_updated) > timedelta(minutes=5):
-            await fetch_github_trending()
-        yield f"data: {cached_trending}\n\n"
-        time.sleep(30)
+app.mount("/mcp", mcp_server.http_app())
 
 @app.get("/trending")
 async def get_trending():
@@ -87,11 +77,21 @@ async def get_trending():
         await fetch_github_trending()
     return {"trending": cached_trending, "last_updated": last_updated.isoformat() if last_updated else None}
 
-@app.get("/trending-sse")
-async def get_trending_sse():
-    """原有SSE端点"""
-    return StreamingResponse(
-        trending_generator(),
-        media_type="text/event-stream",
-        headers={"Cache-Control": "no-cache", "Connection": "keep-alive"}
-    )
+@app.websocket("/trending-ws")
+async def websocket_endpoint(websocket: WebSocket):
+    """WebSocket 端点替代 SSE"""
+    await websocket.accept()
+    try:
+        while True:
+            if not last_updated or (datetime.now() - last_updated) > timedelta(minutes=5):
+                await fetch_github_trending()
+            await websocket.send_json({
+                "trending": cached_trending,
+                "last_updated": last_updated.isoformat() if last_updated else None
+            })
+            await asyncio.sleep(30)  # 使用 asyncio.sleep 而不是 time.sleep
+    except WebSocketDisconnect:
+        print("Client disconnected")
+    except Exception as e:
+        print(f"WebSocket error: {e}")
+        await websocket.close(code=1011)