Update app.py

app.py

@@ -370,8 +370,7 @@ if query:
         <div class="spinner"></div>
     </div>
     """, unsafe_allow_html=True)
-    time.sleep(4)
-    think.empty()
+    
 else:
     st.markdown("")
 
@@ -445,6 +444,7 @@ if query:
             ])
             rewritten = response.content.strip()
             st.markdown(f'<div class="chat-message">{rewritten}</div>', unsafe_allow_html=True)
+            think.empty()
             found = True
             break
 
@@ -455,4 +455,5 @@ if query:
             HumanMessage(content=prompt)
         ])
         rewritten = response.content.strip()
-        st.markdown(f'<div class="chat-message">{rewritten}</div>', unsafe_allow_html=True)
+        st.markdown(f'<div class="chat-message">{rewritten}</div>', unsafe_allow_html=True)
+        think.empty()