update

app.py

@@ -5,6 +5,7 @@ import os
 import sys
 import asyncio
 import logging
+import uvicorn # Correctly imported
 
 # Configure basic logging to see server activity
 logging.basicConfig(level=logging.INFO, stream=sys.stdout)
@@ -16,11 +17,15 @@ logger = logging.getLogger(__name__)
 HOST_PORT = int(os.getenv("PORT", 7860))
 HOST_ADDRESS = os.getenv("HOST", "0.0.0.0") # Bind to all interfaces
 
-# Pass host and port directly to the FastMCP constructor for HTTP transport
+# Initialize FastMCP. The host/port settings passed here will be used by
+# mcp.streamable_http_app() to configure its internal server if not mounted to FastAPI.
+# However, when passed to uvicorn.run(), uvicorn's host/port will take precedence.
+# It's good practice to keep them consistent or omit from here if uvicorn dictates.
+# For simplicity, we'll keep them here as they are also part of MCP server's self-description.
 mcp = FastMCP(
     name="GradioSpacesTools",
-    host=HOST_ADDRESS, # Configure host here
-    port=HOST_PORT    # Configure port here
+    host=HOST_ADDRESS,
+    port=HOST_PORT
 )
 
 clients = {}
@@ -94,17 +99,18 @@ async def run_dia_tts(prompt: str, space_id: str = "ysharma/Dia-1.6B") -> str:
         logger.error(f"Error performing TTS: {e}")
         return f"Error performing TTS: {e}"
 
-async def main():
-    logger.info(f"Attempting to start GradioSpacesTools MCP server on http://{HOST_ADDRESS}:{HOST_PORT}")
-    try:
-        # CORRECT FIX: Use 'streamable-http' transport AND mcp.run_async()
-        # This function runs within an already existing asyncio loop (started by asyncio.run(main()))
-        await mcp.run_async(transport='streamable-http')
-    except Exception as e:
-        logger.exception(f"Failed to start MCP server: {e}")
-    logger.info("GradioSpacesTools MCP server stopped.")
-
 if __name__ == "__main__":
     sys.stdout = os.fdopen(sys.stdout.fileno(), 'w', 1)
-    # This is the single entry point to the asyncio event loop for the application
-    asyncio.run(main())
+    logger.info(f"Starting GradioSpacesTools MCP server via Uvicorn on http://{HOST_ADDRESS}:{HOST_PORT}")
+    
+    # The correct way to get the FastMCP ASGI application for streamable-http
+    # as indicated by the article and FastMCP's internal structure.
+    # Uvicorn will then run this ASGI application.
+    uvicorn.run(
+        mcp.streamable_http_app(), # <-- Use this method! It returns the ASGI app.
+        host=HOST_ADDRESS,
+        port=HOST_PORT,
+        log_level="info", # Set logging level for Uvicorn itself
+        loop="asyncio"    # Ensure Uvicorn uses the asyncio event loop
+    )
+    logger.info("GradioSpacesTools MCP server stopped.")