I actually found a workaround for the issue.
If you click the 3 dots next to the Photos | Albums buttons in my second screenshot, then click Browse, it will bring you to the old, familiar photo selection screen. If you choose images from here, it will have the location attached like normal.
Hopefully a full fix can be found, but this is a pretty simple workaround in the meantime