Update app.py

app.py

@@ -1858,8 +1858,8 @@ def handle_prompt(prompt):
                 st.session_state.chat_history.append({"role": "assistant", "content": assistant_response})
                 with st.chat_message("assistant"):
                     st.markdown(assistant_response)
-                st.session_state["handled"] = True
-                return  # Exit to prevent further processing
+            st.session_state["handled"] = True
+            return  # Exit to prevent further processing
 
             # Handle save TrustBuilder command
             if not st.session_state["handled"]: