Finished and improved on pull request #768 'Creating the "visualization" api for FLAML'

[ghost]

Why are these changes needed?

This is a task assigned to me by @qingyun-wu as a practice pull request. The code is partially adopted from pull request #768 by chadhardin:main.

Related issue number

Checks

• I've used pre-commit to lint the changes in this PR, or I've made sure lint with flake8 output is two 0s.
• I've included any doc changes needed for https://microsoft.github.io/FLAML/. See https://microsoft.github.io/FLAML/docs/Contribute#documentation to build and test documentation locally.
• I've added tests (if relevant) corresponding to the changes introduced in this PR.
• I've made sure all auto checks have passed.