Update app.py

app.py

@@ -88,10 +88,17 @@ if st.button("Submit"):
     messages = [HumanMessage(content=user_question)]
     response = react_graph.invoke({"messages": messages})
     
-    # Display results
-    st.subheader("Responses")
+    # Display results step-by-step like the original code
+    st.subheader("Responses:")
     for m in response['messages']:
-        st.write(m.content)
+        if hasattr(m, "content") and m.content:  # Display human and assistant messages
+            st.write("**AI Message:**", m.content)
+        if hasattr(m, "tool_calls") and m.tool_calls:  # Display tool call steps
+            for tool_call in m.tool_calls:
+                st.write(f"**Tool Call:** `{tool_call['name']}`")
+                st.json(tool_call['args'])  # Display tool arguments in JSON
+                if "output" in tool_call:  # Handle tool outputs if available
+                    st.write("**Tool Output:**", tool_call['output'])
 
     st.success("Processing complete!")