Update app.py

app.py

@@ -11,7 +11,7 @@ import os
 
 # Streamlit UI Header
 st.title("Checkpoints and Breakpoints")
-#st.caption("A detailed demonstration of graph workflow execution with checkpoints and manual intervention.")
+st.caption("Demonstrating workflow execution with checkpoints and tool invocation.")
 
 # Fetch API Keys from Hugging Face Secrets
 openai_api_key = os.getenv("OPENAI_API_KEY")
@@ -33,7 +33,9 @@ if openai_api_key and tavily_api_key:
 
     # Agent Function
     def Agent(state: State):
-        return {"messages": [llm_with_tools.invoke(state["messages"])]}
+        response = llm_with_tools.invoke(state["messages"])
+        print("Agent Response:", response)
+        return {"messages": [response]}
 
     # Memory Checkpoint
     memory = MemorySaver()
@@ -57,7 +59,7 @@ if openai_api_key and tavily_api_key:
 
     # Input Section
     st.subheader("Run the Workflow")
-    user_input = st.text_input("Enter a message to start the graph:", "Hello, I am from U.P.")
+    user_input = st.text_input("Enter a message to start the graph:", "Search for the weather in Uttar Pradesh")
     thread_id = st.text_input("Thread ID", "1")
 
     if st.button("Execute Workflow"):
@@ -71,14 +73,14 @@ if openai_api_key and tavily_api_key:
                 latest_message = event["messages"][-1].pretty_print()
                 outputs.append(latest_message)
                 st.code(latest_message)
-
+        
         if outputs:
             st.subheader("Intermediate Outputs")
             for i, output in enumerate(outputs, 1):
                 st.write(f"**Step {i}:**")
                 st.code(output)
         else:
-            st.warning("No outputs generated.")
+            st.warning("No outputs generated. Adjust your input to trigger tools.")
 
         # Snapshot of Current State
         st.subheader("Current State Snapshot")