Update app.py

app.py

@@ -1750,30 +1750,39 @@ module "{module_name}" {{
 # ============================================================================
 
 async def main():
-    """Main function for running the MCP server"""
-    
-    # Initialize the clean MCP server
-    terraform_server = CleanTerraformMCPServer()
-    
-    # Log startup information
     logger.info("🚀 Clean Terraform MCP Server starting...")
     logger.info("🔧 Available tools: generate_terraform_config, validate_terraform_config, get_deployment_workflow, convert_to_module, format_terraform_code, generate_terraform_docs")
     logger.info("📦 No external dependencies - pure Terraform tooling")
-    
-    # Run the server with stdio transport for HF Spaces/MCP clients
-    async with stdio_server() as (read_stream, write_stream):
-        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={}
+
+    terraform_server = CleanTerraformMCPServer()
+
+    logger.debug("Attempting to initialize stdio_server context...")
+    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={}
+                    )
                 )
             )
-        )
+            logger.debug("MCP server.run() completed (should not happen in normal operation unless server exits).")
+    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
 
 if __name__ == "__main__":
-    asyncio.run(main())
+    try:
+        logger.debug("Script execution started via __main__.")
+        asyncio.run(main())
+        logger.debug("Script finished execution (normal exit).")
+    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