Spaces:

SEUyishu
/

Materials_discovery

Sleeping

App Files Files Community

SEUyishu commited on Dec 4, 2025

Commit

d0dfe88

verified ·

1 Parent(s): f7b4d0f

Upload 3 files

Browse files

Files changed (3) hide show

Dockerfile +1 -2
entrypoint.py +55 -7
mcp_server.py +16 -7

Dockerfile CHANGED Viewed

@@ -12,8 +12,7 @@ ENV PYTHONDONTWRITEBYTECODE=1 \
     PIP_DISABLE_PIP_VERSION_CHECK=1 \
     GNOME_DATA_DIR=/app/gnome_data \
     GNOME_MODEL_DIR=/app/models \
-    HF_HOME=/app/huggingface \
-    MCP_TRANSPORT_SECURITY_DISABLE_HOST_VALIDATION=1
 # Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \

     PIP_DISABLE_PIP_VERSION_CHECK=1 \
     GNOME_DATA_DIR=/app/gnome_data \
     GNOME_MODEL_DIR=/app/models \
+    HF_HOME=/app/huggingface
 # Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \

entrypoint.py CHANGED Viewed

@@ -10,10 +10,6 @@ import os
 import sys
 import logging
-# Disable MCP host validation for HuggingFace Spaces
-# This allows connections from *.hf.space domains
-os.environ["MCP_TRANSPORT_SECURITY_DISABLE_HOST_VALIDATION"] = "1"
 # Add current directory to path
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
@@ -111,17 +107,69 @@ def main():
     import uvicorn
     from starlette.applications import Starlette
     from starlette.routing import Mount, Route
-    from starlette.responses import JSONResponse
     async def health_check(request):
         """Health check endpoint."""
         return JSONResponse({"status": "healthy", "service": "GNoME MCP Server"})
-    # Create Starlette app
     app = Starlette(
         routes=[
             Route("/health", health_check),
-            Mount("/", app=create_sse_app()),
         ]
     )

 import sys
 import logging
 # Add current directory to path
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
     import uvicorn
     from starlette.applications import Starlette
     from starlette.routing import Mount, Route
+    from starlette.responses import JSONResponse, HTMLResponse
+    async def homepage(request):
+        """Homepage with service info."""
+        html_content = """
+        <!DOCTYPE html>
+        <html>
+        <head>
+            <title>GNoME Materials Discovery MCP Server</title>
+            <style>
+                body { font-family: Arial, sans-serif; max-width: 800px; margin: 50px auto; padding: 20px; }
+                h1 { color: #2c3e50; }
+                .status { color: #27ae60; font-weight: bold; }
+                .endpoint { background: #f5f5f5; padding: 10px; border-radius: 5px; margin: 10px 0; }
+                code { background: #e8e8e8; padding: 2px 6px; border-radius: 3px; }
+            </style>
+        </head>
+        <body>
+            <h1>🔬 GNoME Materials Discovery MCP Server</h1>
+            <p class="status">✅ Service is running</p>
+            <h2>Endpoints</h2>
+            <div class="endpoint">
+                <strong>SSE Endpoint:</strong> <code>/sse</code><br>
+                <small>Connect your MCP client (Cursor, Claude Desktop) to this endpoint</small>
+            </div>
+            <div class="endpoint">
+                <strong>Health Check:</strong> <code>/health</code>
+            </div>
+            <h2>Connection Example</h2>
+            <p>Add to your Cursor MCP settings:</p>
+            <pre><code>{
+  "mcpServers": {
+    "gnome-materials": {
+      "url": "https://YOUR-SPACE.hf.space/sse"
+    }
+  }
+}</code></pre>
+            <h2>Available Tools</h2>
+            <ul>
+                <li>query_materials - Search GNoME materials database</li>
+                <li>get_dataset_statistics - Get dataset overview</li>
+                <li>calculate_decomposition_energy - Calculate stability</li>
+                <li>get_phase_diagram - Build phase diagrams</li>
+                <li>calculate_air_stability - Analyze air stability</li>
+                <li>get_structure - Load crystal structures</li>
+                <li>And more...</li>
+            </ul>
+        </body>
+        </html>
+        """
+        return HTMLResponse(content=html_content)
     async def health_check(request):
         """Health check endpoint."""
         return JSONResponse({"status": "healthy", "service": "GNoME MCP Server"})
+    # Create Starlette app with proper routing
+    # SSE app is mounted at /sse path, root serves info page
     app = Starlette(
         routes=[
+            Route("/", homepage),
             Route("/health", health_check),
+            Mount("/sse", app=create_sse_app()),
         ]
     )

mcp_server.py CHANGED Viewed

@@ -25,11 +25,8 @@ from typing import Optional, List, Dict, Any
 from contextlib import asynccontextmanager
 from collections.abc import AsyncIterator
-# Disable host validation for HuggingFace Spaces deployment BEFORE importing MCP
-# This is needed because HuggingFace uses custom domain names like *.hf.space
-os.environ["MCP_TRANSPORT_SECURITY_DISABLE_HOST_VALIDATION"] = "1"
 from mcp.server.fastmcp import FastMCP
 # Import local modules
 from data_utils import DataManager, get_data_manager
@@ -87,8 +84,16 @@ async def app_lifespan(server: FastMCP) -> AsyncIterator[Dict[str, Any]]:
     data_manager.close()
-# Create FastMCP server
-mcp = FastMCP("GNoME Materials Discovery")
 # ============================================================================
@@ -915,7 +920,11 @@ async def check_data_status() -> Dict[str, Any]:
 # ============================================================================
 def create_sse_app():
-    """Create SSE app for mounting in Starlette."""
     return mcp.sse_app()

 from contextlib import asynccontextmanager
 from collections.abc import AsyncIterator
 from mcp.server.fastmcp import FastMCP
+from mcp.server.transport_security import TransportSecuritySettings
 # Import local modules
 from data_utils import DataManager, get_data_manager
     data_manager.close()
+# Create FastMCP server with DNS rebinding protection disabled for HuggingFace Spaces
+# This allows connections from *.hf.space domains
+mcp = FastMCP(
+    "GNoME Materials Discovery",
+    host="0.0.0.0",
+    port=7860,
+    transport_security=TransportSecuritySettings(
+        enable_dns_rebinding_protection=False
+    )
+)
 # ============================================================================
 # ============================================================================
 def create_sse_app():
+    """Create SSE app for mounting in Starlette.
+    Note: DNS rebinding protection is disabled in FastMCP initialization
+    to allow connections from HuggingFace Spaces domains.
+    """
     return mcp.sse_app()