Update server.py

server.py

@@ -1,6 +1,7 @@
 from mcp.server.fastmcp import FastMCP
 import logging
 import os
+from pathlib import Path
 
 # Initialize the server
 mcp = FastMCP("Math-Education-Server")
@@ -57,6 +58,22 @@ async def generate_quiz(chapter_name: str, difficulty: str = "medium") -> str:
     )
 
 if __name__ == "__main__":
-    # FastMCP handles the --transport sse argument automatically 
-    # if you use mcp.run()
-    mcp.run()
+    # For Docker/HF Spaces deployment, we need to run with HTTP transport
+    # The server will listen on the port specified by environment variable or default to 7860
+    import sys
+    
+    # Check if we're running in deployment mode (Docker/HF Spaces)
+    port = int(os.getenv("PORT", "7860"))
+    host = os.getenv("HOST", "0.0.0.0")
+    
+    logger.info(f"Starting MCP Server on {host}:{port}")
+    
+    # For local development, use stdio
+    # For deployment, use SSE transport
+    if "--stdio" in sys.argv:
+        logger.info("Running in stdio mode for local development")
+        mcp.run(transport="stdio")
+    else:
+        logger.info(f"Running in SSE mode for deployment on {host}:{port}")
+        # Run with SSE transport for HTTP-based communication
+        mcp.run(transport="sse", host=host, port=port)