Update app.py

app.py

@@ -281,10 +281,6 @@ def handle_chat(prompt, user_input, teapot_ai):
         st.markdown(user_input)
         st.session_state.messages.append({"role": "user", "content": user_input})
         
-    with st.chat_message("assistant"):
-        st.markdown(response)
-        st.session_state.messages.append({"role": "assistant", "content": response})
-    
     results = brave_search(user_input)
     
     documents = [desc.replace('<strong>','').replace('</strong>','') for _, desc, _ in results]
@@ -300,6 +296,11 @@ def handle_chat(prompt, user_input, teapot_ai):
     context = "\n".join(documents)
     prompt = "You are Teapot, an open-source AI assistant optimized for low-end devices, providing short, accurate responses without hallucinating while excelling at information extraction and text summarization."
     response = query_teapot(prompt, context, user_input, teapot_ai)
+
+    with st.chat_message("assistant"):
+        st.markdown(response)
+        st.session_state.messages.append({"role": "assistant", "content": response})
+    
     
     return response