Update app.py

app.py

@@ -2233,14 +2233,15 @@ def handle_prompt(prompt):
                     with response_placeholder:
                         with st.chat_message("assistant"):
                             st.write(combined_text)
+                    st.session_state.chat_history.append({"role": "assistant", "content": combined_text})
+                    copy_to_clipboard(combined_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": combined_text})
-                copy_to_clipboard(combined_text)
+                
                 st.session_state["handled"] = True  # Mark as handled
 
 # Call the function to handle the prompt