Update app.py

app.py

@@ -111,10 +111,9 @@ def cs_body():
                 message_placeholder.markdown(full_response + "▌")
             message_placeholder.markdown(full_response)
         st.session_state.messages.append({"role": "assistant", "content": full_response})
+        st.success(full_response)
 
 
-    st.success(full_response) 
-
     return None
 
 # Run main()