Update app.py

app.py

@@ -2019,13 +2019,14 @@ def handle_prompt(prompt):
                                     st.markdown(formatted_text, unsafe_allow_html=False)
         
                             update_message_usage(st.session_state.get("wix_user_id", "default_user"))
+                            st.session_state.chat_history.append({"role": "assistant", "content": formatted_text})
+                            copy_to_clipboard(cleaned_text)
         
                         except Exception as e:
                             logging.error(f"Error generating response: {e}")
                             st.error("An error occurred while generating the response. Please try again.")
         
-                        st.session_state.chat_history.append({"role": "assistant", "content": formatted_text})
-                        copy_to_clipboard(cleaned_text)
+                        
                         st.session_state["handled"] = True  # Mark as handled