spinner

app.py

@@ -47,20 +47,21 @@ with st.form("form"):
     submitted = st.form_submit_button(label="Generate Recipe!")
 
 if submitted:
-    dish = (dish or DEFAULT_DISH).strip()
-    co = cohere.Client(api_key=os.environ['CO_API_KEY'])
-    prompt = f"Give a recipe for {dish}."
-    if options:
-        prompt += f" Make sure it is {', '.join(options)}."
-    logger.info(f"Sending prompt {repr(prompt)}")
-    start = time.time()
-    gens = command(co, prompt)
-    recipe = gens[0].text
-    num_tokens = len(gens[0].token_likelihoods)
-    logger.info(f"Generated recipe with {num_tokens} tokens in {time.time()-start:.1f}s: {repr(recipe)}")
-    st.markdown(f"## Recipe for {' '.join(options)} {dish}")
-    st.write(recipe)
-    st.markdown("## Delicious! 10/10 with rice")
+    with st.spinner("Cooking up something delicious ... please wait a few seconds ... "):
+        dish = (dish or DEFAULT_DISH).strip()
+        co = cohere.Client(api_key=os.environ['CO_API_KEY'])
+        prompt = f"Give a recipe for {dish}."
+        if options:
+            prompt += f" Make sure it is {', '.join(options)}."
+        logger.info(f"Sending prompt {repr(prompt)}")
+        start = time.time()
+        gens = command(co, prompt)
+        recipe = gens[0].text
+        num_tokens = len(gens[0].token_likelihoods)
+        logger.info(f"Generated recipe with {num_tokens} tokens in {time.time()-start:.1f}s: {repr(recipe)}")
+        st.markdown(f"## Recipe for {' '.join(options)} {dish}")
+        st.write(recipe)
+        st.markdown("## Delicious! 10/10 with rice")
 
 # Draw chat history.
 with st.expander("About", expanded=False):