Update app.py

app.py

@@ -155,8 +155,8 @@ if prompt:
             [f"<|{msg['role']}|>{msg['content']}<|end|>" for msg in st.session_state["messages"]]
         )
 
+    with st.spinner("Dubs is thinking... Woof Woof! 🐾"):    
         with st.chat_message("assistant", avatar=DUBS_PATH):
-            with st.spinner("Dubs is thinking... Woof Woof! 🐾"):
                 full_response = ""
                 placeholder = st.empty()
             response = stream_response(chat_history, HF_API_KEY)
@@ -169,3 +169,5 @@ if prompt:
         st.session_state["messages"].append({"role": "assistant", "content": full_response})
         # 6) Persist updated chat history
         save_chat_history(st.session_state["session_name"], st.session_state["messages"])
+
+