update

app.py

@@ -1,6 +1,5 @@
 from mcp.server.fastmcp import FastMCP
 from gradio_client import Client
-from fastapi import FastAPI
 import uvicorn
 import logging
 
@@ -8,9 +7,6 @@ import logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Initialize FastAPI app as a fallback
-app = FastAPI()
-
 # Initialize FastMCP server
 try:
     mcp = FastMCP("gradio-spaces")
@@ -80,25 +76,13 @@ async def run_dia_tts(prompt: str, space_id: str = "ysharma/Dia-1.6B") -> str:
     )
     return result
 
-# Determine which FastAPI app to use
+# Use mcp.sse_app as the FastAPI application
 try:
-    fastapi_app = mcp.app  # Try to use mcp.app
-    logger.info("Using mcp.app as the FastAPI application")
+    fastapi_app = mcp.sse_app  # Use SSE app for MCP server
+    logger.info("Using mcp.sse_app as the FastAPI application")
 except AttributeError:
-    logger.warning("mcp.app not found; attempting to use mcp.server or fallback FastAPI app")
-    try:
-        fastapi_app = mcp.server  # Check if mcp.server is the FastAPI app
-        logger.info("Using mcp.server as the FastAPI application")
-    except AttributeError:
-        logger.warning("mcp.server not found; using fallback FastAPI app")
-        fastapi_app = app
-        # Attempt to attach MCP routes to fallback app
-        try:
-            mcp.run(transport="http", app=fastapi_app)
-            logger.info("Attached MCP routes to fallback FastAPI app")
-        except Exception as e:
-            logger.error("Failed to attach MCP routes to fallback app: %s", str(e))
-            raise
+    logger.error("mcp.sse_app not found; FastMCP configuration may be incorrect")
+    raise
 
 if __name__ == "__main__":
     uvicorn.run(fastapi_app, host="0.0.0.0", port=7860)