Update app.py

app.py

@@ -48,7 +48,8 @@ if prompt := st.chat_input("Subject of the song"):
     with st.chat_message("user"):
         st.markdown(prompt)
     prompt = f"Write a rap in the style of the artist SUM about {prompt}"
-    output = query(prompt)
-    st.session_state.messages.append({"role": "assistant", "content": output})
-    with st.chat_message("assistant"):
-        st.markdown(output)
+    with st.spinner(text=f"Generating lyrics..."):
+        output = query(prompt)
+        st.session_state.messages.append({"role": "assistant", "content": output})
+        with st.chat_message("assistant"):
+            st.markdown(output)