Picking Pair of Cities (Budapest/Vienna) vs. (Budapest/Krakow)

I'm trying to get a last minute trip (US based) planned together in basically the first half of June this year. I'm trying to decide what people would recommend between a Budapest/Vienna trip vs. Budapest/Krakow.

I've been to Budapest before but loved it. I'm a little hesitant to return to a city that I've already been too but I really liked my time there and I have a work benefit that allows me to stay at a pretty nice place at a low rate.

I've never been to Krakow or Vienna. I vaguely know the superficial differences. I like a mix of sightseeing, good food, and checking out cool bars/cocktail bars at night. Not really a clubbing person per se, more kind of like to suss out the local night life vs. looking for a raging party.

I feel like I should want to see Vienna more based on all the sights to see, cafes, etc. but it seems a little...restrained? Not sure if that's the right word. Maybe less dynamic than the feel I got from Budapest. On the flipside Krakow is much smaller and less to do.

Based on flight prices right now it makes more financial sense to do a round trip so unfortunately I'd have to go from City A-->City B-->Back to City A.

What would people recommend, and also transport between the proposed cities? I know Budapest and Krakow are not as easy to travel between.