Update server.py

server.py

@@ -3,8 +3,8 @@ from fastapi import FastAPI, Request
 from fastapi.responses import HTMLResponse, FileResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
-from echo_server import mcp as echo_mcp
-from math_server import mcp as math_mcp
+from pokemon.pokemon_server import mcp as pokemon_mcp
+from geogussr_server import mcp as geogussr_mcp
 import os
 
 
@@ -12,8 +12,8 @@ import os
 @contextlib.asynccontextmanager
 async def lifespan(app: FastAPI):
     async with contextlib.AsyncExitStack() as stack:
-        await stack.enter_async_context(echo_mcp.session_manager.run())
-        await stack.enter_async_context(math_mcp.session_manager.run())
+        await stack.enter_async_context(pokemon_mcp.session_manager.run())
+        await stack.enter_async_context(geogussr_mcp.session_manager.run())
         yield
 
 
@@ -39,8 +39,8 @@ async def index(request: Request):
     return templates.TemplateResponse("index.html", {"request": request, "base_url": base_url})
 
 
-app.mount("/echo", echo_mcp.streamable_http_app())
-app.mount("/math", math_mcp.streamable_http_app())
+app.mount("/geoguessr", geogussr_mcp.streamable_http_app())
+app.mount("/Pokemon", pokemon_mcp.streamable_http_app())
 
 PORT = int(os.environ.get("PORT", "10000"))