From 2296911b5eecd1e3e100de3ba2f48612aa113bc8 Mon Sep 17 00:00:00 2001 From: Dima Timofeev <1127655+CuriousDima@users.noreply.github.com> Date: Mon, 10 Jun 2024 22:24:19 -0700 Subject: [PATCH] add spinner instead of a text --- frontend.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/frontend.py b/frontend.py index b3fe1ba..8ab570a 100644 --- a/frontend.py +++ b/frontend.py @@ -53,10 +53,10 @@ def print_nerdy_details(num_tokens: int): st.error("Please wait 5 seconds before rewriting another text.") st.stop() else: - st.write("Rewriting...", unsafe_allow_html=True) - rewritten_text, total_tokens_used = llm_client.rewrite(input_text) - st.markdown("---") - st.markdown(f"{rewritten_text}") - st.session_state["time"] = time.time() - st.markdown("---") - print_nerdy_details(total_tokens_used) + with st.spinner("Rewriting..."): + rewritten_text, total_tokens_used = llm_client.rewrite(input_text) + st.markdown("---") + st.markdown(f"{rewritten_text}") + st.session_state["time"] = time.time() + st.markdown("---") + print_nerdy_details(total_tokens_used)