`print_formatted_text()` messes up terminal output

I've been developing a console application that prints logs while simultaneously prompting the user for input, similar to patch-stdout example.

When I switched to print_formatted_text() instead of print() things messed up. I expect the prompt to stay the last line and for all prints to go above it.

I've been able to reproduce the issue with the patch-stdout.py example by changing print()s to print_formatted_text().

When running it with print() (without any modifications):

$ python patch-stdout.py
i=1
i=2
i=3
i=4
Say something:

Say something: moves below the prints.

When replacing print() with print_formatted_text():

$ python patch-stdout.py
i=1
Say something: i=2
i=3
i=4
i=5
i=6

Say something: is on the 2nd line, and i=2 is appended to it.

I expected the code to behave exactly the same.
How can I use print_formatted_text() without it breaking my terminal output?

Thanks!

Start with examples/prompts/patch-stdout.py and reproduce the behavior by replacing print() with print_formatted_text(). Compare how each function writes while the prompt is active, then verify that formatted output appears above the prompt and leaves the prompt as the last line.