n4a_analytics_lib/st_components.py

# -*- coding:utf-8 -*-

import streamlit as st


def st_pb(method):
    def progress_bar(ref):
        container = st.empty()
        bar = st.progress(0)
        pg_gen = method(ref)
        try:
            while True:
                progress = next(pg_gen)
                bar.progress(progress[0])
                if progress[2]:
                    container.write("✅ Processing... " + progress[1])
                else:
                    container.write("❌️ Errror with..." + progress[1])
        except StopIteration as result:
            return result.value

    return progress_bar