Naomi Osaka admits she's enjoyed the challenge of getting back to her best after advancing to the semi-finals