Not sure if I am missing something. But it is the default implementation in macOS. Just make sure "show floating thumbnail" is turned on (CMD+SHIFT+5 then click options).

I can open it, and take screenshots. But when you edit the image I can’t tell what the app is. Am I still in the screenshot application?

I understand, that’s a good question actually.