Need help choosing my next phone

I currently have a Galaxy S9 and am wanting to buy a new phone since my current one has several issues.

I’m not after the top of the range phones just want a mid-range at this point. My only requirements would be:
– Have a headphone jack
– Flat screen (just over the curves at this point)
– Water resistant

My first option was the A71 but that phone isn’t water resistant and then I looked at the S10 Lite but that phone doesn’t have a headphone jack. Ultimately I would just like a S10 without the curved edges so if anyone knows a phone that fits the build please let me know. I’m also open to other brands (besides Apple) of phones not just Samsung.