add spinner instead of a text

CuriousDima · Jun 11, 2024 · 2296911 · 2296911
1 parent be1e8cf
commit 2296911
Showing 1 changed file with 7 additions and 7 deletions.
diff --git 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("<i>Rewriting...</i>", 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)