Module output inside a Panel

[presedo93]

Hi!

I want to launch some pytorch_lightning scripts using textual. But I'm facing a big difficulty and I haven't found a way to solve it... when I launch the train method inside a Widget, the pytorch_lightning module starts to print all of its info over the TUI (which doesn't look cool). I would like to render all those prints inside a rich Panel so I could have a kind of Console Widget...

-----------------------------------------------------------
|                                                         |
| Start training...                                       |
|                                                         |
|    Progress bar...                                      |
|                                                         |
-----------------------------------------------------------

Is there any way this can be done? The farthest I've gotten is by contextlib.redirect_stdout(f) but I need to wait for the whole training process to finish so the interface is frozen in the meanwhile... I've been also thinking on some way of using the self.console.screen() method of the Widget but didn't find a solution yet...

thanks!!