Update app.py

app.py

@@ -63,18 +63,16 @@ if prompt := st.chat_input("Ask me a complex math question!"):
     with st.chat_message("assistant"):
         with st.spinner("Thinking..."):
             try:
-               # Change this line:
-# response = agent_executor.invoke({"messages": [("user", prompt)]})
-
-# To this:
-response = agent_executor.invoke({
-    "messages": [system_message, ("user", prompt)]
-})
-
-                # Get the final response from the message list
+                # ALL lines below 'try' must be indented 4 spaces further
+                # We pass the system_message rules directly here
+                response = agent_executor.invoke({
+                    "messages": [system_message, ("user", prompt)]
+                })
+                
                 final_answer = response["messages"][-1].content
                 
                 st.markdown(final_answer)
                 st.session_state.messages.append({"role": "assistant", "content": final_answer})
+            
             except Exception as e:
                 st.error(f"An error occurred: {e}")