Update server.py

server.py

@@ -2,6 +2,9 @@ from mcp.server.fastmcp import FastMCP
 import logging
 import sys
 from pathlib import Path
+import os
+# Must set this BEFORE importing FastMCP
+os.environ["MCP_ALLOW_HTTP"] = "1"
 
 # Configure logging to stderr
 logging.basicConfig(
@@ -78,9 +81,17 @@ async def generate_quiz(chapter_name: str, difficulty: str = "medium") -> str:
 #         # mcp.run(transport="sse") and set env vars MCP_HOST and MCP_PORT
 
 if __name__ == "__main__":
-    import sys
-    if "--stdio" in sys.argv:
-        mcp.run(transport="stdio")
-    else:
-        # 'sse' is the correct internal name for the HTTP/SSE transport
-        mcp.run(transport="sse")
+    # Patch the SSE validator to accept any host (required for HF Spaces)
+    import mcp.server.sse
+    
+    original_validate = mcp.server.sse._validate_request
+    
+    def patched_validate(scope):
+        """Accept all requests regardless of host header"""
+        return True
+    
+    mcp.server.sse._validate_request = patched_validate
+    
+    # Run the server
+    logger.info("Starting FastMCP server with patched host validation...")
+    mcp.run(transport="sse", host="0.0.0.0", port=7860)