Update interim.py

interim.py

@@ -1,6 +1,6 @@
 import streamlit as st
 from typing import TypedDict, Annotated
-from langgraph.graph import StateGraph, START, END
+from langgraph.graph import StateGraph
 from langgraph.checkpoint.memory import MemorySaver
 from langgraph.graph.message import add_messages
 from langchain_openai import ChatOpenAI
@@ -28,32 +28,25 @@ if openai_api_key and tavily_api_key:
     # Initialize LLM and Tools
     llm = ChatOpenAI(model="gpt-4o-mini")
     tool = TavilySearchResults(max_results=2)
-    tools = [tool]
-    llm_with_tools = llm.bind_tools(tools)
+    llm_with_tools = llm.bind_tools([tool])
 
     # Agent Function
     def Agent(state: State):
-        print("Agent received state:", state)
-        # Force tool invocation
+        st.sidebar.write("Agent received state:", state["messages"])
         response = llm_with_tools.invoke(state["messages"])
-        print("Agent Response:", response)
+        st.sidebar.write("Agent Response:", response)
         return {"messages": [response]}
 
     # Memory Checkpoint
     memory = MemorySaver()
 
-    # Graph Definition
+    # Build the Graph
     graph = StateGraph(State)
     tool_node = ToolNode(tools=[tool])
 
     graph.add_node("Agent", Agent)
     graph.add_node("tools", tool_node)
-
-    # Corrected Conditional Edge
-    def always_true(state):
-        return "True"  # Return string "True"
-
-    graph.add_conditional_edges("Agent", always_true, {"True": "tools"})
+    graph.add_conditional_edges("Agent", tools_condition)
     graph.add_edge("tools", "Agent")
     graph.set_entry_point("Agent")
 
@@ -71,46 +64,42 @@ if openai_api_key and tavily_api_key:
 
     if st.button("Execute Workflow"):
         thread = {"configurable": {"thread_id": thread_id}}
-        input_message = {'messages': HumanMessage(content=user_input)}
+        input_message = {"messages": [HumanMessage(content=user_input)]}
 
         st.write("### Execution Outputs")
         outputs = []
         for event in app.stream(input_message, thread, stream_mode="values"):
-            if "messages" in event and event["messages"]:
-                latest_message = event["messages"][-1].pretty_print()
-                outputs.append(latest_message)
-                st.code(latest_message)
+            st.code(event["messages"][-1].content)
+            outputs.append(event["messages"][-1].content)
+            st.sidebar.write("Intermediate State:", event["messages"])
 
         if outputs:
             st.subheader("Intermediate Outputs")
-            for i, output in enumerate(outputs, 1):
-                st.write(f"**Step {i}:**")
+            for idx, output in enumerate(outputs, start=1):
+                st.write(f"**Step {idx}:**")
                 st.code(output)
         else:
             st.warning("No outputs generated. Adjust your input to trigger tools.")
 
-        # Snapshot of Current State
+        # Display Snapshot of State
         st.subheader("Current State Snapshot")
         snapshot = app.get_state(thread)
-        if snapshot.values["messages"]:
-            current_message = snapshot.values["messages"][-1]
-            st.code(current_message.pretty_print())
-
-            # Safe Access to Tool Calls
-            if hasattr(current_message, "tool_calls") and current_message.tool_calls:
-                tool_call_id = current_message.tool_calls[0]["id"]
-                manual_response = st.text_area("Manual Tool Response", "Enter response to continue...")
-                if st.button("Update State"):
-                    new_messages = [
-                        ToolMessage(content=manual_response, tool_call_id=tool_call_id),
-                        AIMessage(content=manual_response),
-                    ]
-                    app.update_state(thread, {"messages": new_messages})
-                    st.success("State updated successfully!")
-                    st.code(app.get_state(thread).values["messages"][-1].pretty_print())
-            else:
-                st.warning("No tool calls available for manual updates.")
+        current_message = snapshot.values["messages"][-1]
+        st.code(current_message.pretty_print())
+
+        # Manual Update Section
+        if hasattr(current_message, "tool_calls") and current_message.tool_calls:
+            tool_call_id = current_message.tool_calls[0]["id"]
+            manual_response = st.text_area("Manual Tool Response", "Enter your response to continue execution...")
+            if st.button("Update State"):
+                new_messages = [
+                    ToolMessage(content=manual_response, tool_call_id=tool_call_id),
+                    AIMessage(content=manual_response),
+                ]
+                app.update_state(thread, {"messages": new_messages})
+                st.success("State updated successfully!")
+                st.code(app.get_state(thread).values["messages"][-1].pretty_print())
         else:
-            st.warning("No state messages available.")
+            st.warning("No tool calls available for manual updates.")
 else:
-    st.error("API keys are missing! Please set `OPENAI_API_KEY` and `TAVILY_API_KEY` in Hugging Face Spaces Secrets.")
+    st.error("API keys are missing! Please set `OPENAI_API_KEY` and `TAVILY_API_KEY` in Hugging Face Spaces Secrets.")