streamlit rich interaction

[franzhaas]

I am very happy with rich output in the console application, jupyterlab use case. However when using streamlit i had the following issue.:

When using the progress bar, the progress was shown on the console i started streamlit from, not the web-browser.
When 2 users wanted to do something which tried to use a progress bar, this failed.

Solution.:

class RichProgressWrapper():
    """This is to allow to have multiple instances of converters active at every given time, and to have some progress information similar to what rich provides...
    """
    progress = collections.defaultdict(lambda :0)
    workload = collections.defaultdict(lambda :None)

    def __init__(self, *args):
        self.expectednum = 0
        pass

    def __enter__(self):
        return self

    def __exit__(self, *args):
        pass

    def add_task(self, name, total):
        ident = threading.current_thread().ident
        self.expectednum = total
        self.workload[ident] = total

    def update(self, _, completed):
        self.progress[threading.current_thread().ident] = completed/self.expectednum

richProgressBarUser.Progress = RichProgressWrapper

And at the point where the progress bar shall be shown.:

    thread = threading.Thread(target=get_data_worker, args=args)
    thread.start()
    progressbar = st.progress(0, text="")
    while True:
        thread.join(timeout=3)
        text = ""
        if RichProgressWrapper.workload[thread.ident]:
            passed = time.time()-start_time
            speed = (RichProgressWrapper.progress[thread.ident]*RichProgressWrapper.workload[thread.ident])/(passed*1e6)
            text = f"Working on {RichProgressWrapper.workload[thread.ident]/1e6:.1f} MB input... {passed:.3f}s have passed... {(RichProgressWrapper.progress[thread.ident]*100.0):.1f}% competed... running at {speed:.3f} MB/s "
        progressbar.progress(RichProgressWrapper.progress[thread.ident], text=text)
        if not thread.is_alive():
            progressbar.empty()
            break

which provides a progressbar which does not throw an exception if multiple connections want to use the progress bar at the same time.

Is there any interest in streamlit support?

Thanks in advance,
Franz