How to get formatted text?

Hello!
I am currently stuck on trying to get the formatted text. Is there any easy way to do that?

I tried a lot and came up with this, but it is not perfect . After calling the get_formatted_text function twice, I get a I/O operation on closed file error.

from prompt_toolkit import print_formatted_text, HTML
from io import StringIO
import sys

# set isatty to True so create_output does not create a PlainTextOutput
class TtyStringIO(StringIO):
    def isatty(self) -> bool:
        return True


def get_formatted_text(text):
    with TtyStringIO() as str_io:
        sys.stdout = str_io
        print_formatted_text(HTML(text))
        sys.stdout = sys.__stdout__
        val = str_io.getvalue()
    return val

# This Works as expected
print(get_formatted_text("<b>Hello World</b>"))
# This fails with a ValueError: I/O operation on closed file
print(get_formatted_text("Hello World"))

Start by reproducing the closed-file error with the provided get_formatted_text example, then inspect print_formatted_text and its output handling. Done means repeated calls return the formatted text without raising an I/O operation on closed file error.