Screenshots are actually taken using JavaScript nowadays. So, it will just capture the page rendered using HTML. It will not be able to capture the URL.
You can extend take screenshot method and get the URL of the page by using
driver.getURL()
Once, you receive the URL, you can simply add it to your screenshot image manually.
Let us know your views.