more debug

app.py

@@ -1064,6 +1064,7 @@ class CleanTerraformMCPServer:
         
         @self.server.list_tools()
         async def handle_list_tools() -> list[types.Tool]:
+            logger.debug("Handling ListToolsRequest") # Add debug
             """List all available Terraform tools"""
             return [
                 types.Tool(
@@ -1233,6 +1234,7 @@ class CleanTerraformMCPServer:
         
         @self.server.call_tool()
         async def handle_call_tool(name: str, arguments: dict) -> list[types.TextContent]:
+            logger.debug("Handling CallToolRequest for tool: {name} ") # Add debug here
             """Handle tool calls"""
             try:
                 if name == "generate_terraform_config":
@@ -1768,29 +1770,47 @@ async def main():
     try:
         async with stdio_server() as (read_stream, write_stream):
             logger.debug("stdio_server context entered successfully. Attempting to run MCP server...")
-            await terraform_server.server.run(
-                read_stream,
-                write_stream,
-                InitializationOptions(
-                    server_name="terraform-mcp-server",
-                    server_version="1.0.0",
-                    capabilities=terraform_server.server.get_capabilities(
-                        notification_options=NotificationOptions(),
-                        experimental_capabilities={}
+            server_task = asyncio.create_task(
+                terraform_server.server.run(
+                    read_stream,
+                    write_stream,
+                    InitializationOptions(
+                        server_name="terraform-mcp-server",
+                        server_version="1.0.0",
+                        capabilities=terraform_server.server.get_capabilities(
+                            notification_options=NotificationOptions(),
+                            experimental_capabilities={}
+                        )
                     )
                 )
             )
-            logger.debug("MCP server.run() completed (should not happen in normal operation unless server exits).")
+            # This is the crucial part: keep the main event loop running.
+            # The server_task is now running in the background.
+            # We need to explicitly await it or another task that keeps the event loop alive.
+            # asyncio.Future() creates a Future that never completes, effectively blocking main.
+            # Or you can await the server_task if it's meant to run indefinitely.
+            # Given your log "MCP server.run() completed", it suggests it exits.
+            # So, we need something *after* it that blocks.
+
+            # Option 1: Await the server task, assuming it will block indefinitely (but your logs contradict this)
+            # await server_task
+
+            # Option 2: Keep the event loop running indefinitely if server_task completes
+            logger.debug("MCP server task created. Keeping event loop alive...")
+            await asyncio.Future() # This creates a future that never resolves, keeping the event loop running
+
+            logger.debug("This line should theoretically not be reached if asyncio.Future() keeps the loop alive.")
+
     except Exception as e:
         logger.critical(f"Unhandled exception during stdio_server or server.run: {e}", exc_info=True)
-        # Re-raise or exit with a non-zero code if you want to explicitly signal failure
-        raise # Re-raise to ensure HF Spaces sees the error
+        raise
 
 if __name__ == "__main__":
     try:
         logger.debug("Script execution started via __main__.")
         asyncio.run(main())
-        logger.debug("Script finished execution (normal exit).")
+        logger.debug("Script finished execution (normal exit).") # This might not be reached with asyncio.Future()
+    except asyncio.CancelledError:
+        logger.info("Application received cancellation signal. Exiting gracefully.")
     except Exception as e:
         logger.critical(f"Fatal unhandled exception in main execution block: {e}", exc_info=True)
-        # It's possible the exception here is caught by HF Spaces environment directly