qrefactor to include main index route

app.py

@@ -2,19 +2,28 @@
 """
 HuggingFace Spaces entry point for the Weights & Biases MCP Server.
 
-This script starts the MCP server in HTTP mode, making it accessible
-via Server-Sent Events (SSE) for remote MCP clients.
+This script creates a FastAPI application with a landing page and mounts
+the MCP server from the existing server module.
 """
 
 import os
 import sys
 import logging
 from pathlib import Path
+from contextlib import asynccontextmanager
 
 # Add the src directory to Python path
 sys.path.insert(0, str(Path(__file__).parent / "src"))
 
-from wandb_mcp_server.server import cli
+from fastapi import FastAPI, Request
+from fastapi.responses import HTMLResponse
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
+from fastapi.middleware.cors import CORSMiddleware
+import uvicorn
+
+# Import the existing MCP server instance and utilities
+from wandb_mcp_server.server import mcp
 from wandb_mcp_server.utils import get_rich_logger
 
 # Configure logging for the app
@@ -25,43 +34,87 @@ logging.basicConfig(
 
 logger = get_rich_logger("huggingface-spaces-app")
 
-def main():
-    """Main entry point for HuggingFace Spaces."""
+# Lifecycle manager for the app
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    """Manage application lifecycle."""
+    # Startup
     logger.info("Starting Weights & Biases MCP Server on HuggingFace Spaces")
     
-    # Ensure we're running in HTTP mode for HuggingFace Spaces
-    # Override command line arguments to force HTTP transport
-    original_argv = sys.argv.copy()
-    sys.argv = [
-        sys.argv[0],  # Keep the script name
-        "--transport", "http",
-        "--host", os.environ.get("HOST", "0.0.0.0"),  # Listen on all interfaces for HuggingFace Spaces
-        "--port", str(os.environ.get("PORT", "7860"))  # Use PORT env var or default to 7860
-    ]
-    
     # Check for required environment variables
     wandb_api_key = os.environ.get("WANDB_API_KEY")
     if not wandb_api_key:
-        logger.error("WANDB_API_KEY environment variable is required!")
-        logger.error("Please set your Weights & Biases API key in the Space's environment variables.")
-        logger.error("You can get your API key from: https://wandb.ai/authorize")
-        sys.exit(1)
+        logger.warning("WANDB_API_KEY environment variable is not set!")
+        logger.warning("Please set your Weights & Biases API key in the Space's environment variables.")
+        logger.warning("You can get your API key from: https://wandb.ai/authorize")
+    else:
+        logger.info("WANDB_API_KEY configured: Yes")
     
-    logger.info(f"WANDB_API_KEY configured: {'Yes' if wandb_api_key else 'No'}")
     logger.info(f"Starting HTTP server on port {os.environ.get('PORT', '7860')}")
-    logger.info("MCP endpoint will be available at: /mcp")
+    logger.info("Landing page available at: /")
+    logger.info("MCP endpoint available at: /mcp/sse")
+    
+    yield
+    
+    # Shutdown
+    logger.info("Shutting down Weights & Biases MCP Server")
+
+# Create FastAPI app with lifecycle management
+app = FastAPI(
+    title="Weights & Biases MCP Server",
+    description="Model Context Protocol server for querying W&B data",
+    lifespan=lifespan
+)
+
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Mount static files (create directory if it doesn't exist)
+static_dir = Path(__file__).parent / "static"
+static_dir.mkdir(exist_ok=True)
+app.mount("/static", StaticFiles(directory=str(static_dir)), name="static")
+
+# Set up templates
+templates_dir = Path(__file__).parent / "templates"
+templates = Jinja2Templates(directory=str(templates_dir))
+
+@app.get("/", response_class=HTMLResponse)
+async def index(request: Request):
+    """Serve the landing page."""
+    return templates.TemplateResponse("index.html", {"request": request})
+
+@app.get("/health")
+async def health():
+    """Health check endpoint."""
+    return {"status": "healthy", "service": "wandb-mcp-server"}
+
+# Mount the existing MCP server at /mcp path
+# The MCP server from server.py already has all tools registered
+mcp.run(
+    transport="streamable-http",
+    http_app=app,
+    http_path="/mcp"
+)
+
+def main():
+    """Main entry point for HuggingFace Spaces."""
+    # Run the FastAPI app with Uvicorn
+    port = int(os.environ.get("PORT", "7860"))
+    host = os.environ.get("HOST", "0.0.0.0")
     
-    try:
-        # Call the CLI function which will start the server
-        cli()
-    except KeyboardInterrupt:
-        logger.info("Server stopped by user")
-    except Exception as e:
-        logger.error(f"Server error: {e}", exc_info=True)
-        sys.exit(1)
-    finally:
-        # Restore original argv
-        sys.argv = original_argv
+    uvicorn.run(
+        "app:app",
+        host=host,
+        port=port,
+        log_level="info",
+        reload=False
+    )
 
 if __name__ == "__main__":
-    main()
+    main()